diff --git a/lib/data_migrate/database_tasks.rb b/lib/data_migrate/database_tasks.rb index dee2c47..59738d3 100644 --- a/lib/data_migrate/database_tasks.rb +++ b/lib/data_migrate/database_tasks.rb @@ -44,6 +44,22 @@ def migration_connection # :nodoc: ensure migration_class.connection_handler.establish_connection(original_db_config) end + + def with_temporary_pool(db_config) + original_db_config = migration_class.connection_db_config + pool = migration_class.establish_connection(db_config) + yield pool + ensure + migration_class.establish_connection(original_db_config) + end + + def migration_class # :nodoc: + ActiveRecord::Base + end + + def migration_connection # :nodoc: + migration_class.connection + end end def db_configs_with_versions