lib/arel/middleware/railtie.rb in arel_toolkit-0.3.0 vs lib/arel/middleware/railtie.rb in arel_toolkit-0.4.0

@@ -1,6 +1,16 @@
 module Arel
   module Middleware
+    if defined? Rails::Railtie
+      class Railtie < Rails::Railtie
+        initializer 'arel.middleware.insert' do
+          ActiveSupport.on_load :active_record do
+            Arel::Middleware::Railtie.insert_postgresql
+          end
+        end
+      end
+    end
+
     class Railtie
       def self.insert_postgresql
         ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.prepend(
           Arel::Middleware::PostgreSQLAdapter,
         )