lib/fresh_connection/railtie.rb in fresh_connection-0.1.8 vs lib/fresh_connection/railtie.rb in fresh_connection-0.2.0

@@ -1,10 +1,13 @@
+require 'rails'
+
 module FreshConnection
   class Railtie < Rails::Railtie
     initializer "fresh_connection.configure_rails_initialization" do |app|
-      app.config.middleware.swap(
-        ActiveRecord::ConnectionAdapters::ConnectionManagement,
-        FreshConnection::Rack::ConnectionManagement
-      )
+      Initializer.swap_rack(app)
+
+      ActiveSupport.on_load(:active_record) do
+        Initializer.extend_active_record
+      end
     end
   end
 end