dekiru-data_migration 0.2.0 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b2aedd5e4be5d28fb840f19ea8599d4398345b4ae861410719d550aa4342a55b
-  data.tar.gz: a36038e89f49f415b58556f190a1199882b58628df39671f6ca56dfc5ed57885
+  metadata.gz: c5428a8cf5e2a02ff63e8b5a7d3dd2703cf471f60a7bf1dfdd289418a93f1bca
+  data.tar.gz: d0b1afb0f2d5bad6112849c8f046e90c309a19eddde473bde1cf179e8e2a395a
 SHA512:
-  metadata.gz: d116ce6bb08682d5d23c3942b06e8f27dc898a4593d6520efb89c0d5dfa319ad317dec6adc2ef8816dd60fb39ce7f5f058762e0e48cdfc3e716601b3f54e039f
-  data.tar.gz: 5c6d29334eb32a2debd4d44e2c445431a5d6632de181a8cc05403dcb613264e66d1bef53755d3cfaaa1486d0d16c5bb8d5cb049250e97046229e5b269c43ea6d
+  metadata.gz: 30e87b497f402579cac1d212f0cc9e12ba4bf8bcc84166de33dbbf9619f9da734c5d468d115801db0b326cffc7dfe09e4ca21b1b7e2c818ecf4848e18bf7c894
+  data.tar.gz: f5a5b31881fe7485d18ffea31950d70b4a5a05a7f5defd44197ec87b78c454a8db707be376825fc28a010c9f95d0c9584beb50f14965cc508fa211602d4bd226

data/README.md

@@ -50,6 +50,60 @@ Dekiru::DataMigration::Operator.execute('Grant admin privileges to users') do
 end
 ```
 
+## Data Migration Class (Recommended)
+
+You can also define migration logic as a class, which makes testing easier:
+
+```ruby
+# scripts/20230118_demo_migration.rb
+class DemoMigration < Dekiru::DataMigration::Migration
+  def migration_targets
+    User.where("email LIKE '%sonicgarden%'").where(admin: false)
+  end
+
+  def migrate_record(user)
+    user.update!(admin: true)
+  end
+
+  def migrate
+    super
+    log "Updated user count: #{User.where(admin: true).count}"
+  end
+end
+
+DemoMigration.run
+```
+
+### Testing Migration Classes
+
+The class-based approach makes it easy to write unit tests:
+
+```ruby
+# spec/migrations/demo_migration_spec.rb
+RSpec.describe DemoMigration do
+  let(:migration) { described_class.new }
+
+  describe '#migration_targets' do
+    it 'returns correct migration targets' do
+      create_list(:user, 3, email: 'test@sonicgarden.jp', admin: false)
+      create_list(:user, 2, email: 'other@example.com', admin: false)
+
+      targets = migration.migration_targets
+      expect(targets.count).to eq(3)
+      expect(targets.all? { |u| u.email.include?('sonicgarden') }).to be true
+    end
+  end
+
+  describe '#migrate_record' do
+    it 'updates user to admin' do
+      user = create(:user, admin: false)
+      expect { migration.migrate_record(user) }
+        .to change { user.reload.admin }.from(false).to(true)
+    end
+  end
+end
+```
+
 Execution result:
 ```
 $ bin/rails r scripts/demo.rb
@@ -98,7 +152,7 @@ Are you sure to commit? (yes/no) > yes
 
 ## Generating Maintenance Scripts
 
-You can generate maintenance scripts that use `Dekiru::DataMigration::Operator` with the generator. The filename will be prefixed with the execution date.
+You can generate maintenance scripts that use `Dekiru::DataMigration::Migration` with the generator. The filename will be prefixed with the execution date.
 
 ```bash
 $ bin/rails g maintenance_script demo_migration
@@ -109,6 +163,29 @@ Generated file example:
 # scripts/20230118_demo_migration.rb
 # frozen_string_literal: true
 
+class DemoMigration < Dekiru::DataMigration::Migration
+  def migration_targets
+    # 移行対象を返すActiveRecord::Relationを定義
+    # 例: User.where(some_condition: true)
+    raise NotImplementedError, 'migration_targets method must be implemented'
+  end
+
+  def migrate_record(record)
+    # 個別レコードの更新処理を定義
+    # 例: record.update!(some_attribute: new_value)
+    raise NotImplementedError, 'migrate_record method must be implemented'
+  end
+end
+
+DemoMigration.run
+```
+
+### Legacy Block-based Approach
+
+For backward compatibility, you can still use the block-based approach:
+
+```ruby
+# scripts/legacy_demo.rb
 Dekiru::DataMigration::Operator.execute('demo_migration') do
   # write here
 end

data/lib/dekiru/data_migration/migration.rb

@@ -0,0 +1,81 @@
+# frozen_string_literal: true
+
+module Dekiru
+  module DataMigration
+    # Base class for data migration with testable method separation
+    class Migration
+      def self.run(options = {})
+        migration = new
+        title = migration.title
+
+        Operator.execute(title, options) do
+          migration.instance_variable_set(:@operator_context, self)
+          migration.migrate
+        end
+      end
+
+      def title
+        self.class.name.demodulize.underscore.humanize
+      end
+
+      def migrate
+        targets = migration_targets
+
+        log "Target count: #{targets.count}"
+        confirm?
+
+        find_each_with_progress(targets) do |record|
+          migrate_record(record)
+        end
+
+        log "Migration completed"
+      end
+
+      def migration_targets
+        raise NotImplementedError, "#{self.class}#migration_targets must be implemented"
+      end
+
+      def migrate_record(record)
+        raise NotImplementedError, "#{self.class}#migrate_record must be implemented"
+      end
+
+      private
+
+      def confirm?
+        if @operator_context
+          @operator_context.send(:confirm?)
+        else
+          # Default behavior during test (no confirmation)
+          puts "Confirmation skipped in test mode"
+        end
+      end
+
+      def log(message)
+        if @operator_context
+          @operator_context.send(:log, message)
+        else
+          # Default behavior during test
+          puts message
+        end
+      end
+
+      def find_each_with_progress(scope, options = {}, &block)
+        if @operator_context
+          @operator_context.send(:find_each_with_progress, scope, options, &block)
+        else
+          # Default behavior during test (no progress bar)
+          scope.find_each(&block)
+        end
+      end
+
+      def each_with_progress(enum, options = {}, &block)
+        if @operator_context
+          @operator_context.send(:each_with_progress, enum, options, &block)
+        else
+          # Default behavior during test (no progress bar)
+          enum.each(&block)
+        end
+      end
+    end
+  end
+end

data/lib/dekiru/data_migration/version.rb

@@ -2,6 +2,6 @@
 
 module Dekiru
   module DataMigration
-    VERSION = "0.2.0"
+    VERSION = "1.0.0"
   end
 end

data/lib/dekiru/data_migration.rb

@@ -2,6 +2,7 @@
 
 require_relative "data_migration/version"
 require_relative "data_migration/transaction_provider"
+require_relative "data_migration/migration"
 require_relative "data_migration_operator"
 
 module Dekiru

data/lib/generators/maintenance_script/maintenance_script_generator.rb

@@ -14,8 +14,11 @@ class MaintenanceScriptGenerator < Rails::Generators::NamedBase
   source_root File.expand_path("templates", __dir__)
 
   def copy_maintenance_script_file
+    @filename_date = filename_date
+    @class_name = "#{name.classify}#{@filename_date}"
+
     template "maintenance_script.rb.erb",
-             "#{Dekiru::DataMigration.configuration.maintenance_script_directory}/#{filename_date}_#{file_name}.rb"
+             "#{Dekiru::DataMigration.configuration.maintenance_script_directory}/#{@filename_date}_#{file_name}.rb"
   end
 
   private

data/lib/generators/maintenance_script/templates/maintenance_script.rb.erb

@@ -2,6 +2,20 @@
 
 using Dekiru::DataMigration::DangerousMethodGuard
 
-Dekiru::DataMigration::Operator.execute('<%= @name %>') do
-  # write here
+class <%= @class_name %> < Dekiru::DataMigration::Migration
+  def title
+    '<%= @name %>'
+  end
+
+  def migration_targets
+    # Define ActiveRecord::Relation that returns the target migration
+    # User.where(some_condition: true)
+  end
+
+  def migrate_record(record)
+    # Define the update process for individual records
+    # user.update!(some_attribute: new_value)
+  end
 end
+
+<%= @class_name %>.run if __FILE__ == $PROGRAM_NAME

data/sig/dekiru/data_migration/migration.rbs

@@ -0,0 +1,23 @@
+module Dekiru
+  module DataMigration
+    class Migration
+      def self.run: (?Hash[Symbol, untyped] options) -> bool
+
+      def title: () -> String
+
+      def migrate: () -> void
+
+      def migration_targets: () -> untyped
+
+      def migrate_record: (untyped record) -> void
+
+      private
+
+      def log: (String message) -> void
+
+      def find_each_with_progress: (untyped scope, ?Hash[Symbol, untyped] options) { (untyped) -> void } -> void
+
+      def each_with_progress: [T] (Enumerable[T] enum, ?Hash[Symbol, untyped] options) { (T) -> void } -> void
+    end
+  end
+end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: dekiru-data_migration
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 1.0.0
 platform: ruby
 authors:
 - SonicGarden
 bindir: exe
 cert_chain: []
-date: 2025-06-24 00:00:00.000000000 Z
+date: 2025-06-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -54,6 +54,7 @@ files:
 - Rakefile
 - lib/dekiru/data_migration.rb
 - lib/dekiru/data_migration/dangerous_method_guard.rb
+- lib/dekiru/data_migration/migration.rb
 - lib/dekiru/data_migration/operator.rb
 - lib/dekiru/data_migration/transaction_provider.rb
 - lib/dekiru/data_migration/version.rb
@@ -62,6 +63,7 @@ files:
 - lib/generators/maintenance_script/maintenance_script_generator.rb
 - lib/generators/maintenance_script/templates/maintenance_script.rb.erb
 - sig/dekiru/data_migration.rbs
+- sig/dekiru/data_migration/migration.rbs
 - sig/dekiru/data_migration/operator.rbs
 homepage: https://github.com/SonicGarden/dekiru-data_migration
 licenses: