departure.rb in departure-6.2.0

- old
+ new

@@ -1,18 +1,21 @@
 require 'active_record'
 require 'active_support/all'
 
+require 'active_record/connection_adapters/for_alter'
+
 require 'departure/version'
 require 'departure/log_sanitizers/password_sanitizer'
 require 'departure/runner'
 require 'departure/cli_generator'
 require 'departure/logger'
 require 'departure/null_logger'
 require 'departure/logger_factory'
 require 'departure/configuration'
 require 'departure/errors'
 require 'departure/command'
+require 'departure/connection_base'
 
 require 'departure/railtie' if defined?(Rails)
 
 # We need the OS not to buffer the IO to see pt-osc's output while migrating
 $stdout.sync = true
@@ -57,10 +60,10 @@
       # Make all connections in the connection pool to use PerconaAdapter
       # instead of the current adapter.
       def reconnect_with_percona
         connection_config = ActiveRecord::Base
           .connection_config.merge(adapter: 'percona')
-        ActiveRecord::Base.establish_connection(connection_config)
+        Departure::ConnectionBase.establish_connection(connection_config)
       end
     end
   end
 end