lib/fresh_connection/railtie.rb in fresh_connection-2.1.0 vs lib/fresh_connection/railtie.rb in fresh_connection-2.1.1

@@ -1,12 +1,19 @@
 require 'fresh_connection/rack/connection_management'
 
 module FreshConnection
   class Railtie < Rails::Railtie
     initializer "fresh_connection.configure_rails_initialization" do |app|
-      app.config.app_middleware.insert_before(
-        ActiveRecord::ConnectionAdapters::ConnectionManagement,
-        FreshConnection::Rack::ConnectionManagement
-      )
+      if defined?(ActiveRecord::ConnectionAdapters::ConnectionManagement)
+        app.config.app_middleware.insert_before(
+          ActiveRecord::ConnectionAdapters::ConnectionManagement,
+          FreshConnection::Rack::ConnectionManagement
+        )
+      else
+        app.config.app_middleware.insert_before(
+          ActionDispatch::Reloader,
+          FreshConnection::Rack::ConnectionManagement
+        )
+      end
     end
   end
 end