Rails main fix

Author: zendesk-edytaroz

lib/migration_tools/tasks.rb

@@ -28,11 +28,16 @@ def migrations_paths
     end
 
     def migrator(target_version = nil)
-      if ActiveRecord::VERSION::MAJOR >= 7 && ActiveRecord::VERSION::MINOR >= 1
+      if ActiveRecord::VERSION::MAJOR == 7 && ActiveRecord::VERSION::MINOR == 1
         migrate_up(ActiveRecord::MigrationContext.new(
           migrations_paths,
           ActiveRecord::Base.connection.schema_migration
         ).migrations, target_version)
+      elsif ActiveRecord.gem_version >= Gem::Version.new("7.2")
+        migrate_up(ActiveRecord::MigrationContext.new(
+          migrations_paths,
+          ActiveRecord::Base.connection_pool.schema_migration
+        ).migrations, target_version)
       else
         migrate_up(ActiveRecord::MigrationContext.new(
           migrations_paths,
@@ -42,11 +47,16 @@ def migrator(target_version = nil)
     end
 
     def migrate_up(migrations, target_version)
-      if ActiveRecord::VERSION::MAJOR >= 7 && ActiveRecord::VERSION::MINOR >= 1
+      if ActiveRecord::VERSION::MAJOR == 7 && ActiveRecord::VERSION::MINOR == 1
         ActiveRecord::Migrator.new(:up, migrations,
           ActiveRecord::Base.connection.schema_migration,
           ActiveRecord::Base.connection.internal_metadata,
           target_version)
+      elsif ActiveRecord.gem_version >= Gem::Version.new("7.2")
+        ActiveRecord::Migrator.new(:up, migrations,
+          ActiveRecord::Base.connection_pool.schema_migration,
+          ActiveRecord::Base.connection_pool.internal_metadata,
+          target_version)
       else
         ActiveRecord::Migrator.new(:up, migrations, ActiveRecord::SchemaMigration, target_version)
       end