Support disallowing permanent connection checkouts with Rails 7.2

lib/immigrant.rb

@@ -51,13 +51,25 @@ def key_validator
       end
 
       def tables
-        @tables ||= ActiveRecord::Base.connection.tables
+        @tables ||= with_connection(&:tables)
       end
 
       def current_foreign_keys
-        tables.map{ |table|
-          ActiveRecord::Base.connection.foreign_keys(table)
-        }.flatten
+        with_connection do |connection|
+          tables.map{ |table|
+            connection.foreign_keys(table)
+          }.flatten
+        end
+      end
+
+      def with_connection
+        if ActiveRecord::Base.respond_to?(:with_connection)
+          ActiveRecord::Base.with_connection do |connection|
+            yield connection
+          end
+        else
+          yield ActiveRecord::Base.connection
+        end
       end
 
       def model_classes

lib/immigrant/key_validator.rb

@@ -26,7 +26,11 @@ def columns_for(table_name)
     end
 
     def connection
-      @connection ||= ActiveRecord::Base.connection
+      @connection ||= if ActiveRecord::Base.respond_to?(:lease_connection)
+        ActiveRecord::Base.lease_connection
+      else
+        ActiveRecord::Base.connection
+      end
     end
   end
 end