data_migrate 9.2.0 → 10.0.0.rc1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 98382ecfcd0ad635d3e53bb746f4b2a7a402c2b2a78d312e864bbe6f288b65b1
-  data.tar.gz: 315d89e1465e7f5a804d4e70d67963949608632b91b7276ea23319f2183789ba
+  metadata.gz: d212f0e6fa0c9b767cbdf17854c7c52d640a9c8611d0b10ed6c0460de68985cd
+  data.tar.gz: ac338ef447f1942d64eb5cf6c0fe520f71089968c04fb03fb829a1318534b4eb
 SHA512:
-  metadata.gz: '04098142838de3bb59be85adc46a9a4aed734668a154378488c74ea2a1982793a45450364f089edb479add87cab1b47e801916553c01ad003bbe71c0ba06b855'
-  data.tar.gz: 4654fdfb138a53c5cc4ff59df1d63bc66ccd0f66d292f79c0d37ccbdb43c184144d5122718e9027ea643f1e0786d79844aee44d70feb5a1c455a59670286e5ef
+  metadata.gz: 01ae940433e4cd51a2f421297fb3ffb72372503971ee906164129c035c89aedb6c18d5003cd083930c24245f16728c38c93aed94fb8d9b2f7d0288895362a3d4
+  data.tar.gz: 9d94ff40ed274256247960c0dd1325645b638ac2198d29b3bce2c3c11f842b4b0a2b381bd73fcb17af308ab366cdca16c7f99c9e2fa778da7ab6103352e50ba1

data/.github/workflows/build.yml

@@ -11,13 +11,14 @@ jobs:
       matrix:
         os: [ "ubuntu-20.04" ]
         ruby:
+          - '2.7'
           - '3.0'
           - '3.1'
           - '3.2'
         gemfile:
+          - gemfiles/rails_6.0.gemfile
           - gemfiles/rails_6.1.gemfile
           - gemfiles/rails_7.0.gemfile
-          - gemfiles/rails_7.1.gemfile
     runs-on: ubuntu-latest
     env:
       BUNDLE_GEMFILE: ${{ matrix.gemfile }}
@@ -31,4 +32,4 @@ jobs:
           ruby-version: ${{ matrix.ruby }}
           bundler-cache: true
       - name: Run tests
-        run: bundle exec rspec
+        run: bundle exec rspec

data/.github/workflows/gempush.yml

@@ -1,8 +1,10 @@
-name: Release Gem
+name: Ruby Gem
 
 on:
-  release:
-    types: [released]
+  push:
+    branches:
+      - main
+      - chaunce-multiple_connection_support
 
 jobs:
   build:
@@ -11,10 +13,10 @@ jobs:
 
     steps:
       - uses: actions/checkout@master
-      - name: Set up Ruby 3.2
-        uses: ruby/setup-ruby@v1
+      - name: Set up Ruby 2.6
+        uses: actions/setup-ruby@v1
         with:
-          ruby-version: '3.2'
+          version: 2.6.x
 
       - name: Publish to RubyGems
         run: |

data/.gitignore

@@ -1,12 +1,11 @@
 /.rvmrc
 *.gem
+*.lock
 gemfiles/.bundle
 spec/db/test.db
 spec/db/other_test.db
 spec/db/data_schema.rb
 .vscode/
 .DS_Store
-.ruby-gemset
-.ruby-version
 .idea/
-vendor/
+vendor/

data/.ruby-version

@@ -0,0 +1 @@
+2.7.5

data/.travis.yml

@@ -0,0 +1,14 @@
+language: ruby
+rvm:
+  - 2.7
+  - 3.0
+  - 3.2
+script: bundle exec rspec
+gemfile:
+  - gemfiles/rails_6.0.gemfile
+  - gemfiles/rails_6.1.gemfile
+  - gemfiles/rails_7.0.gemfile
+jobs:
+  exclude:
+    - rvm: 3.0
+    - rvm: 3.2

data/Appraisals

@@ -1,11 +1,11 @@
+appraise 'rails-6.0' do
+  gem 'rails', '~> 6.0.0'
+end
+
 appraise 'rails-6.1' do
   gem 'rails', '~> 6.1.0'
 end
 
 appraise 'rails-7.0' do
-  gem 'rails', '~> 7.0.0'
-end
-
-appraise 'rails-7.1' do
-  gem 'rails', '7.1.0'
+  gem 'rails', '~> 7.0'
 end

data/Changelog.md

@@ -1,44 +1,13 @@
 # Changelog
 
-## 9.2.0
-- Support Rails 7.1 https://github.com/ilyakatz/data-migrate/pull/278
-- Build and test against 7.1.0.rc1 https://github.com/ilyakatz/data-migrate/pull/286
-
-## 9.1.0
-
-- Fix a bug that caused `schema_sha1` in `ar_internal_metadata` to be reset to the `data_schema.rb` file. (#272)
-- Remove the need for empty data_schema files for non-primary databases. (#273)
-
-## [YANKED] 10.0.3.rc
-
-- Remove all travis references [leoarnold](https//:github.com/leoarnold)
-- Changing to rc because of ongoing discussion how to properly handle multiple database environments
-
-## [YANKED] 10.0.2
-
-Change "rails" dependencies to "railties"
-
-## [YANKED] 10.0.1
-
-- Bug fix for Rails 6 config [chaunce](https//:github.com/chaunce)
-- Railties bug fix by [opti](https://github.com/opti)
-
-## [YANKED] 10.0.0
-
-Releasing 10.0.0
-
-!!! Breaking changes !!!
-
-- This version introduces a breaking change which may lead to undesired
-behavior in multi-database environments. See https://github.com/ilyakatz/data-migrate/issues/181
-
-## [YANKED] 10.0.0.rc1
+## 10.0.0.rc1
 
 - Changes by [chaunce](https//:github.com/chaunce)
 - Multiple databases support
 - Refactor to clean things up
 - Deprecate rails 5.2 support for real
 
+
 ## 9.0.0
 
 Ruby 3.2 support [mehanoid](https://github.com/mehanoid)

data/README.md

@@ -2,7 +2,7 @@
 
 - [![Version](http://img.shields.io/gem/v/data_migrate.svg?style=flat-square)](https://rubygems.org/gems/data_migrate)
 - [![License](http://img.shields.io/badge/license-MIT-brightgreen.svg?style=flat-square)](http://opensource.org/licenses/MIT)
-- ![.github/workflows/build.yml](https://github.com/ilyakatz/data-migrate/actions/workflows/.github/workflows/build.yml/badge.svg)
+- [![Travis](https://img.shields.io/travis/ilyakatz/data-migrate.svg)](https://travis-ci.org/ilyakatz/data-migrate)
 - [![Reviewed by Hound](https://img.shields.io/badge/Reviewed_by-Hound-8E64B0.svg)](https://houndci.com)
 
 Run data migrations alongside schema migrations.
@@ -12,7 +12,7 @@ migrations, except they should be reserved for data migrations. For
 instance, if you realize you need to titleize all your titles, this
 is the place to do it.
 
-![directory tree example](./screenshot.png)
+![Travis](./screenshot.png)
 
 ## Why should I use this?
 
@@ -36,7 +36,7 @@ table to track all migrations.
 
 ## Rails Support
 
-Support Rails 6.1 through 7.1
+Support Rails 6.0 through 7.0
 
 
 #### v1
@@ -96,9 +96,7 @@ Tasks work as they would with the 'vanilla' db version. The 'with_data' addition
 
 With 'up' and 'down', you can specify the option 'BOTH', which defaults to false. Using true, will migrate both the data and schema (in the desired direction) if they both match the version provided. Again, going up, schema is given precedence. Down its data.
 
-When using `rake db:migrate:with_data` migrations will be run in ascending order by their version. For example, if you have a data migration with version 20230410000000 and a schema migration with version 20230415000000, expect the data migration to run first.
-
-`rake db:migrate:status:with_data` provides an additional column to indicate which type of migration.
+`rake db:migrate:status:with_data` provides and additional column to indicate which type of migration.
 
 ### Configuration
 
@@ -164,9 +162,9 @@ Run tests for a specific version of Rails
 
 ```
 bundle exec appraisal install
+bundle exec appraisal rails-6.0 rspec
 bundle exec appraisal rails-6.1 rspec
 bundle exec appraisal rails-7.0 rspec
-bundle exec appraisal rails-7.1 rspec
 ```
 
 ## Thanks

data/data_migrate.gemspec

@@ -19,7 +19,7 @@ Gem::Specification.new do |s|
     activerecord
     railties
   ].each do |rails_gem|
-    s.add_dependency(rails_gem, '>= 6.1')
+    s.add_dependency(rails_gem, '>= 6.0')
   end
 
   s.add_development_dependency "appraisal"

data/gemfiles/{rails_7.1.gemfile → rails_6.0.gemfile}

@@ -2,7 +2,7 @@
 
 source "http://rubygems.org"
 
+gem "rails", "~> 6.0.0"
 gem "sqlite3", "~> 1.4"
-gem "rails", "7.1.0"
 
 gemspec path: "../"

data/gemfiles/rails_6.1.gemfile

@@ -2,7 +2,7 @@
 
 source "http://rubygems.org"
 
-gem "sqlite3", "~> 1.4"
 gem "rails", "~> 6.1.0"
+gem "sqlite3", "~> 1.4"
 
 gemspec path: "../"

data/lib/data_migrate/data_migrator.rb

@@ -5,20 +5,32 @@ require "data_migrate/config"
 
 module DataMigrate
   class DataMigrator < ActiveRecord::Migrator
+    def self.migrations_paths
+      [DataMigrate.config.data_migrations_path]
+    end
+
+    def self.assure_data_schema_table
+      DataMigrate::DataSchemaMigration.create_table
+    end
+
+    def initialize(direction, migrations, target_version = nil)
+      @direction         = direction
+      @target_version    = target_version
+      @migrated_versions = nil
+      @migrations        = migrations
+
+      validate(@migrations)
+
+      DataMigrate::DataSchemaMigration.create_table
+      ActiveRecord::InternalMetadata.create_table
+    end
+
     def load_migrated
       @migrated_versions =
-        DataMigrate::RailsHelper.data_schema_migration.normalized_versions.map(&:to_i).sort
+        DataMigrate::DataSchemaMigration.normalized_versions.map(&:to_i).sort
     end
 
     class << self
-      def migrations_paths
-        [DataMigrate.config.data_migrations_path]
-      end
-
-      def create_data_schema_table
-        DataMigrate::RailsHelper.data_schema_migration.create_table
-      end
-
       def current_version
         DataMigrate::MigrationContext.new(migrations_paths).current_version
       end
@@ -67,10 +79,10 @@ module DataMigrate
     def record_version_state_after_migrating(version)
       if down?
         migrated.delete(version)
-        DataMigrate::RailsHelper.data_schema_delete_version(version.to_s)
+        DataMigrate::DataSchemaMigration.where(version: version.to_s).delete_all
       else
         migrated << version
-        DataMigrate::RailsHelper.data_schema_migration.create_version(version.to_s)
+        DataMigrate::DataSchemaMigration.create!(version: version.to_s)
       end
     end
   end

data/lib/data_migrate/data_schema.rb

@@ -9,7 +9,7 @@ module DataMigrate
     #   ActiveRecord::ConnectionAdapters::SchemaStatements
     #     #assume_migrated_upto_version
     def define(info)
-      DataMigrate::DataMigrator.create_data_schema_table
+      DataMigrate::DataMigrator.assure_data_schema_table
 
       return if info[:version].blank?
 
@@ -57,7 +57,7 @@ module DataMigrate
     end
 
     def table_name
-      DataMigrate::RailsHelper.data_schema_migration.table_name
+      DataMigrate::DataSchemaMigration.table_name
     end
   end
 end

data/lib/data_migrate/data_schema_migration.rb

@@ -1,29 +1,12 @@
 module DataMigrate
-  class DataSchemaMigration < ActiveRecord::SchemaMigration
-    # In Rails 7.1+, ActiveRecord::SchemaMigration methods are instance methods
-    # So we only load the appropriate methods depending on Rails version.
-    if DataMigrate::RailsHelper.rails_version_equal_to_or_higher_than_7_1
-      def table_name
-        ActiveRecord::Base.table_name_prefix + 'data_migrations' + ActiveRecord::Base.table_name_suffix
-      end
-
-      def primary_key
-        "version"
-      end
-    else
-      class << self
-        def table_name
-          ActiveRecord::Base.table_name_prefix + 'data_migrations' + ActiveRecord::Base.table_name_suffix
-        end
-
-        def primary_key
-          "version"
-        end
+  class DataSchemaMigration
+    class << self
+      delegate :table_name, :primary_key, :create_table, :normalized_versions, :create, :create!, :table_exists?, :exists?, :where, to: :instance
 
-        def create_version(version)
-          # Note that SchemaMigration.create_version in Rails 7.1 does not
-          # raise an error if validations fail but we retain this behaviour for now.
-          create!(version: version)
+      def instance
+        @instance ||= Class.new(::ActiveRecord::SchemaMigration) do
+          define_singleton_method(:table_name) { ActiveRecord::Base.table_name_prefix + 'data_migrations' + ActiveRecord::Base.table_name_suffix }
+          define_singleton_method(:primary_key) { "version" }
         end
       end
     end

data/lib/data_migrate/database_tasks.rb

@@ -9,15 +9,10 @@ module DataMigrate
     extend ActiveRecord::Tasks::DatabaseTasks
 
     class << self
-      def schema_file(_format = nil)
-        File.join(db_dir, "data_schema.rb")
-      end
-
       def schema_file_type(_format = nil)
         "data_schema.rb"
       end
 
-      # This method is removed in Rails 7.0
       def dump_filename(spec_name, format = ActiveRecord::Base.schema_format)
         filename = if spec_name == "primary"
           schema_file_type(format)
@@ -67,26 +62,10 @@ module DataMigrate
           )
         end
       end
-
-      def schema_dump_path(db_config, format = ActiveRecord.schema_format)
-        return ENV["DATA_SCHEMA"] if ENV["DATA_SCHEMA"]
-
-        # We only require a schema.rb file for the primary database
-        return unless db_config.primary?
-
-        File.join(File.dirname(ActiveRecord::Tasks::DatabaseTasks.schema_dump_path(db_config, format)), schema_file_type)
-      end
-
-      # Override this method from `ActiveRecord::Tasks::DatabaseTasks`
-      # to ensure that the sha saved in ar_internal_metadata table
-      # is from the original schema.rb file
-      def schema_sha1(file)
-        ActiveRecord::Tasks::DatabaseTasks.schema_dump_path(ActiveRecord::Base.configurations.configs_for(env_name: ActiveRecord::Tasks::DatabaseTasks.env, name: "primary"))
-      end
     end
 
     def self.forward(step = 1)
-      DataMigrate::DataMigrator.create_data_schema_table
+      DataMigrate::DataMigrator.assure_data_schema_table
       migrations = pending_migrations.reverse.pop(step).reverse
       migrations.each do | pending_migration |
         if pending_migration[:kind] == :data
@@ -101,10 +80,8 @@ module DataMigrate
 
     def self.pending_data_migrations
       data_migrations = DataMigrate::DataMigrator.migrations(data_migrations_path)
-      data_migrator = DataMigrate::RailsHelper.data_migrator(:up, data_migrations)
-      sort_migrations(
-        data_migrator.pending_migrations.map { |m| { version: m.version, name: m.name, kind: :data } }
-        )
+      sort_migrations(DataMigrate::DataMigrator.new(:up, data_migrations ).
+        pending_migrations.map {|m| { version: m.version, name: m.name, kind: :data }})
     end
 
     def self.pending_schema_migrations
@@ -112,8 +89,8 @@ module DataMigrate
     end
 
     def self.past_migrations(sort = nil)
-      data_versions = DataMigrate::RailsHelper.data_schema_migration.table_exists? ? DataMigrate::RailsHelper.data_schema_migration.normalized_versions : []
-      schema_versions = DataMigrate::RailsHelper.schema_migration.normalized_versions
+      data_versions = DataMigrate::DataSchemaMigration.table_exists? ? DataMigrate::DataSchemaMigration.normalized_versions : []
+      schema_versions = ActiveRecord::SchemaMigration.normalized_versions
       migrations = data_versions.map { |v| { version: v.to_i, kind: :data } } + schema_versions.map { |v| { version: v.to_i, kind: :schema } }
 
       sort&.downcase == "asc" ? sort_migrations(migrations) : sort_migrations(migrations).reverse

data/lib/data_migrate/legacy_migrator.rb

@@ -0,0 +1,22 @@
+module DataMigrate
+  class LegacyMigrator
+    def initialize(migrations_paths = "db/data")
+      @migrations_paths = migrations_paths || "db/data"
+    end
+
+    def migrate
+      dates =
+        DataMigrate::DataMigrator.migrations(@migrations_paths).collect(&:version)
+      legacy = ActiveRecord::SchemaMigration.where(version: dates)
+      legacy.each do |v|
+        begin
+          version = v.version
+          puts "Creating #{version} in data schema"
+          DataMigrate::DataSchemaMigration.create(version: version)
+        rescue ActiveRecord::RecordNotUnique
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/data_migrate/migration_context.rb

@@ -11,8 +11,7 @@ module DataMigrate
                               migrations
                             end
 
-      data_migrator = DataMigrate::RailsHelper.data_migrator(:up, selected_migrations, DataMigrate::RailsHelper.schema_migration, DataMigrate::RailsHelper.internal_metadata, target_version)
-      data_migrator.migrate
+      DataMigrator.new(:up, selected_migrations, target_version).migrate
     end
 
     def down(target_version = nil)
@@ -23,13 +22,11 @@ module DataMigrate
           migrations
         end
 
-      data_migrator = DataMigrate::RailsHelper.data_migrator(:down, selected_migrations, DataMigrate::RailsHelper.schema_migration, DataMigrate::RailsHelper.internal_metadata, target_version)
-      data_migrator.migrate
+      DataMigrator.new(:down, selected_migrations, target_version).migrate
     end
 
     def run(direction, target_version)
-      data_migrator = DataMigrate::RailsHelper.data_migrator(direction, migrations, DataMigrate::RailsHelper.schema_migration, DataMigrate::RailsHelper.internal_metadata, target_version)
-      data_migrator.run
+      DataMigrator.new(direction, migrations, target_version).run
     end
 
     def current_version
@@ -43,12 +40,12 @@ module DataMigrate
     end
 
     def migrations_status
-      db_list = DataMigrate::RailsHelper.data_schema_migration.normalized_versions
+      db_list = DataSchemaMigration.normalized_versions
 
       file_list = migration_files.map do |file|
         version, name, scope = parse_migration_filename(file)
         raise ActiveRecord::IllegalMigrationNameError.new(file) unless version
-        version = DataMigrate::RailsHelper.schema_migration.normalize_migration_number(version)
+        version = ActiveRecord::SchemaMigration.normalize_migration_number(version)
         status = db_list.delete(version) ? "up" : "down"
         [status, version, (name + scope).humanize]
       end.compact
@@ -63,15 +60,15 @@ module DataMigrate
     private
 
     def get_all_versions
-      if DataMigrate::RailsHelper.data_schema_migration.table_exists?
-        DataMigrate::RailsHelper.data_schema_migration.normalized_versions.map(&:to_i)
+      if DataMigrate::DataSchemaMigration.table_exists?
+        DataSchemaMigration.normalized_versions.map(&:to_i)
       else
         []
       end
     end
 
     def move(direction, steps)
-      migrator = DataMigrate::RailsHelper.data_migrator(direction, migrations)
+      migrator = DataMigrator.new(direction, migrations)
 
       if current_version != 0 && !migrator.current_migration
         raise ActiveRecord::UnknownMigrationVersionError.new(current_version)

data/lib/data_migrate/schema_dumper.rb

@@ -26,7 +26,7 @@ module DataMigrate
 
     def initialize(connection)
       @connection = connection
-      all_versions =  DataMigrate::RailsHelper.data_schema_migration.normalized_versions
+      all_versions =  DataSchemaMigration.normalized_versions
 
       @version = begin
                     all_versions.max

data/lib/data_migrate/schema_migration.rb

@@ -7,14 +7,14 @@ module DataMigrate
     def self.pending_schema_migrations
       all_migrations = DataMigrate::MigrationContext.new(migrations_paths).migrations
       sort_migrations(
-        ActiveRecord::Migrator.new(:up, all_migrations, DataMigrate::RailsHelper.schema_migration, DataMigrate::RailsHelper.internal_metadata).
+        ActiveRecord::Migrator.new(:up, all_migrations, ActiveRecord::Base.connection.schema_migration).
         pending_migrations.
         map {|m| { version: m.version, kind: :schema }}
       )
     end
 
     def self.run(direction, migration_paths, version)
-      ActiveRecord::MigrationContext.new(migration_paths, DataMigrate::RailsHelper.schema_migration).run(direction, version)
+      ActiveRecord::MigrationContext.new(migration_paths, ActiveRecord::Base.connection.schema_migration).run(direction, version)
     end
 
     def self.sort_migrations(set1, set2 = nil)
@@ -24,10 +24,9 @@ module DataMigrate
 
     def self.migrations_paths
       spec_name = DataMigrate.config.spec_name
-      # The positional argument true is to include pre-release versions, such as 7.1.0.alpha
-      if spec_name && Gem::Dependency.new("railties", ">= 7.0").match?("railties", Gem.loaded_specs["railties"].version, true)
+      if spec_name && Gem::Dependency.new("rails", "~> 7.0").match?("rails", Gem.loaded_specs["rails"].version)
         ActiveRecord::Base.configurations.configs_for(env_name: Rails.env, name: spec_name).migrations_paths
-      elsif spec_name && Gem::Dependency.new("railties", "~> 6.1").match?("railties", Gem.loaded_specs["railties"].version)
+      elsif spec_name && Gem::Dependency.new("rails", "~> 6.0").match?("rails", Gem.loaded_specs["rails"].version)
         ActiveRecord::Base.configurations.configs_for(env_name: Rails.env, spec_name: spec_name).migrations_paths
       else
         Rails.application.config.paths["db/migrate"].to_a

data/lib/data_migrate/status_service.rb

@@ -24,11 +24,11 @@ module DataMigrate
     private
 
     def table_name
-      DataMigrate::RailsHelper.data_schema_migration.table_name
+      DataMigrate::DataSchemaMigration.table_name
     end
 
     def output(stream)
-      unless DataMigrate::RailsHelper.data_schema_migration.table_exists?
+      unless DataMigrate::DataSchemaMigration.table_exists?
         stream.puts "Data migrations table does not exist yet."
         return
       end
@@ -44,9 +44,9 @@ module DataMigrate
     end
 
     def database_name
-      if Gem::Dependency.new("railties", "~> 7.0").match?("railties", Gem.loaded_specs["railties"].version)
+      if Gem::Dependency.new("rails", "~> 7.0").match?("rails", Gem.loaded_specs["rails"].version)
         ActiveRecord::Base.connection_db_config.configuration_hash[:database]
-      elsif Gem::Dependency.new("railties", "~> 6.1").match?("railties", Gem.loaded_specs["railties"].version)
+      elsif Gem::Dependency.new("rails", "~> 6.0").match?("rails", Gem.loaded_specs["rails"].version)
         ActiveRecord::Base.connection_config[:database]
       end
     end

data/lib/data_migrate/tasks/data_migrate_tasks.rb

@@ -9,9 +9,9 @@ module DataMigrate
         @migrations_paths ||= DataMigrate.config.data_migrations_path
       end
 
-      def dump
+      def dump(db_config)
         if dump_schema_after_migration?
-          filename = DataMigrate::DatabaseTasks.schema_file
+          filename = DataMigrate::DatabaseTasks.dump_filename(spec_name(db_config), ActiveRecord::Base.schema_format)
           ActiveRecord::Base.establish_connection(DataMigrate.config.db_configuration) if DataMigrate.config.db_configuration
           File.open(filename, "w:utf-8") do |file|
             DataMigrate::SchemaDumper.dump(ActiveRecord::Base.connection, file)
@@ -22,7 +22,7 @@ module DataMigrate
       def migrate
         target_version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
 
-        DataMigrate::DataMigrator.create_data_schema_table
+        DataMigrate::DataMigrator.assure_data_schema_table
         DataMigrate::MigrationContext.new(migrations_paths).migrate(target_version)
       end
 
@@ -44,15 +44,16 @@ module DataMigrate
         end
       end
 
-      def status
-        DataMigrate::StatusService.dump
+      def status(db_config)
+        puts "\ndatabase: #{spec_name(db_config)}\n\n"
+        DataMigrate::StatusService.dump(ActiveRecord::Base.connection)
       end
 
-      def status_with_schema
+      def status_with_schema(db_config)
         db_list_data = ActiveRecord::Base.connection.select_values(
-          "SELECT version FROM #{DataMigrate::RailsHelper.data_schema_migration.table_name}"
+          "SELECT version FROM #{DataMigrate::DataSchemaMigration.table_name}"
         )
-        db_list_schema = DataMigrate::RailsHelper.schema_migration_versions
+        db_list_schema = ActiveRecord::SchemaMigration.all.pluck(:version)
         file_list = []
 
         Dir.foreach(File.join(Rails.root, migrations_paths)) do |file|
@@ -76,7 +77,7 @@ module DataMigrate
         file_list.sort!{|a,b| "#{a[1]}_#{a[3] == 'data' ? 1 : 0}" <=> "#{b[1]}_#{b[3] == 'data' ? 1 : 0}" }
 
         # output
-        puts "\ndatabase: #{database_name}\n\n"
+        puts "\ndatabase: #{spec_name(db_config)}\n\n"
         puts "#{"Status".center(8)} #{"Type".center(8)}  #{"Migration ID".ljust(14)} Migration Name"
         puts "-" * 60
         file_list.each do |file|
@@ -93,11 +94,11 @@ module DataMigrate
 
       private
 
-      def database_name
-        if Gem::Dependency.new("railties", "~> 7.0").match?("railties", Gem.loaded_specs["railties"].version)
-          ActiveRecord::Base.connection_db_config.database
-        elsif Gem::Dependency.new("railties", "~> 6.1").match?("railties", Gem.loaded_specs["railties"].version)
-          ActiveRecord::Base.connection_config[:database]
+      def spec_name(db_config)
+        if Gem::Dependency.new("rails", "~> 7.0").match?("rails", Gem.loaded_specs["rails"].version)
+          db_config.name
+        elsif Gem::Dependency.new("rails", "~> 6.0").match?("rails", Gem.loaded_specs["rails"].version)
+          db_config.spec_name
         end
       end
     end

data/lib/data_migrate/version.rb

@@ -1,3 +1,3 @@
 module DataMigrate
-  VERSION = "9.2.0".freeze
+  VERSION = "10.0.0.rc1".freeze
 end