after_migrate 0.2.2 → 0.2.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 62f5a309584cee52788e0ea45c222648cc58c4db46bc08ce6188c6325f6c62dc
-  data.tar.gz: e36efca18e936086cdace33a4fe422c2d34f63502ddf01c53090cf8048ed92c6
+  metadata.gz: 52bea11a45a6765c9bff5e82d42d527595e7ef6c91e84028ca2dde978d8d527e
+  data.tar.gz: 3dcd46239d300faa855998e427585870899a0ea0d013e21fb6eb770ecd5f5be4
 SHA512:
-  metadata.gz: 2590c965f71fc07b5171cea88869115516687331138b0a6a20b5c4bdd3008a13be1c7a7382dd8886112a02e076904d57f59e2f24db44bfa23d1c36dd30700b40
-  data.tar.gz: 4a2f1dd2df2bb18e8f663b1ca150db22a6d4d5474b651ea0ee693129640926266ce77168e370921cab4ac214620a601fe02f6204a97829575ec7e417c944a7d4
+  metadata.gz: e21f8c6f9db4d2db100b5ef4c93f87ce0a7b69af5a7da29b79ce91a4a47e585ffdce9f33726929aed3962693ad9afee40418a40dc76a813dc2ee717b5333a59d
+  data.tar.gz: d6da80e272cc1895bb753b6fa01b423fc8eb9c3ea4f93d34d691f64743d1269689c3147764d953114957907d36fea5cac8910b03baac009fea08984924c4154b

data/CHANGELOG.md

@@ -2,6 +2,27 @@
 
 All notable changes to this project will be documented in this file.
 
+## [0.2.4] - 2026-06-01
+
+### Added
+- `config.store_options` for store-specific settings, e.g. `config.store_options_for(:redis)[:client]`
+
+### Changed
+- Store-specific configuration now flows through `config.store_options` while preserving compatibility accessors such as `config.store_path`, `config.redis`, `config.redis_key_prefix`, and `config.redis_ttl`
+- `config.run_id` now defaults from `AFTER_MIGRATE_RUN_ID`, falling back to `default`
+
+## [0.2.3] - 2026-06-01
+
+### Added
+- Redis-backed store via `config.store = :redis` for sharing collected migration tables across processes
+- `config.redis` for passing a Redis client, connection pool, or callable client provider
+- `config.redis_key_prefix` and `config.redis_ttl` for namespacing and expiring Redis store keys
+- Redis store support for run isolation via `config.run_id`
+- Redis store specs for persistence, merging, reset behavior, connection pools, `Redis.new` fallback, and missing-client errors
+
+### Changed
+- Store cache keys now include Redis-specific configuration, so changing Redis store settings rebuilds the active store instance
+
 ## [0.2.2] - 2026-05-31
 
 ### Added

data/lib/after_migrate/store.rb

@@ -100,5 +100,94 @@ module AfterMigrate
         end
       end
     end
+
+    class RedisStore < Memory
+      attr_reader :key_prefix, :run_id, :ttl
+
+      def initialize(redis:, key_prefix: 'after_migrate', run_id: nil, ttl: 24 * 60 * 60)
+        super()
+        @redis = redis
+        @key_prefix = key_prefix.to_s
+        @run_id = run_id.to_s.presence || 'default'
+        @ttl = ttl.to_i
+      end
+
+      def affected_tables
+        load_into_memory
+        super
+      end
+
+      def merge_tables(schema, table_names)
+        return if table_names.blank?
+
+        with_redis do |redis|
+          redis.sadd(index_key, schema)
+          redis.sadd(schema_key(schema), table_names.to_a)
+          expire_keys(redis, schema)
+        end
+        merge_into_memory(schema => table_names)
+      end
+
+      def reset!
+        super
+        with_redis do |redis|
+          schemas = redis.smembers(index_key)
+          keys = schemas.map { |schema| schema_key(schema) }
+          redis.del(*(keys + [index_key])) if keys.any?
+          redis.del(index_key) if keys.empty?
+        end
+      end
+
+      private
+
+      def load_into_memory
+        with_redis do |redis|
+          redis.smembers(index_key).each do |schema|
+            merge_into_memory(schema => redis.smembers(schema_key(schema)))
+          end
+        end
+      end
+
+      def merge_into_memory(schemas)
+        schemas.each do |schema, table_names|
+          set = @affected_tables.compute_if_absent(schema) { Concurrent::Set.new }
+          set.merge(table_names)
+        end
+      end
+
+      def index_key
+        "#{base_key}:schemas"
+      end
+
+      def schema_key(schema)
+        "#{base_key}:schema:#{schema}"
+      end
+
+      def base_key
+        "#{key_prefix}:#{run_id}"
+      end
+
+      def expire_keys(redis, schema)
+        return unless ttl.positive?
+
+        redis.expire(index_key, ttl)
+        redis.expire(schema_key(schema), ttl)
+      end
+
+      def with_redis(&block)
+        redis = resolved_redis
+        return redis.with { |connection| block.call(connection) } if redis.respond_to?(:with)
+
+        block.call(redis)
+      end
+
+      def resolved_redis
+        client = @redis.respond_to?(:call) ? @redis.call : @redis
+        return client if client
+        return Redis.new if defined?(Redis)
+
+        raise 'AfterMigrate Redis store requires config.redis or the redis gem'
+      end
+    end
   end
 end

data/lib/after_migrate/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AfterMigrate
-  VERSION = '0.2.2'
+  VERSION = '0.2.4'
 end

data/lib/after_migrate.rb

@@ -8,7 +8,8 @@ require 'after_migrate/railtie'
 
 module AfterMigrate
   class Configuration
-    attr_accessor :enabled, :verbose, :vacuum, :analyze, :rake_tasks_enhanced, :defer, :store, :store_path, :run_id
+    attr_accessor :enabled, :verbose, :vacuum, :analyze, :rake_tasks_enhanced, :defer,
+                  :store, :run_id, :store_options
 
     def initialize
       @enabled = true
@@ -18,8 +19,53 @@ module AfterMigrate
       @rake_tasks_enhanced = true
       @defer = true
       @store = :memory
-      @store_path = 'tmp/after_migrate/affected_tables.json'
-      @run_id = nil
+      @run_id = ENV.fetch('AFTER_MIGRATE_RUN_ID', 'default')
+      @store_options = {
+        file: {
+          path: 'tmp/after_migrate/affected_tables.json'
+        },
+        redis: {
+          client: nil,
+          key_prefix: 'after_migrate',
+          ttl: 24 * 60 * 60
+        }
+      }
+    end
+
+    def store_path
+      store_options_for(:file)[:path]
+    end
+
+    def store_path=(value)
+      store_options_for(:file)[:path] = value
+    end
+
+    def redis
+      store_options_for(:redis)[:client]
+    end
+
+    def redis=(value)
+      store_options_for(:redis)[:client] = value
+    end
+
+    def redis_key_prefix
+      store_options_for(:redis)[:key_prefix]
+    end
+
+    def redis_key_prefix=(value)
+      store_options_for(:redis)[:key_prefix] = value
+    end
+
+    def redis_ttl
+      store_options_for(:redis)[:ttl]
+    end
+
+    def redis_ttl=(value)
+      store_options_for(:redis)[:ttl] = value
+    end
+
+    def store_options_for(store_name)
+      store_options[store_name.to_sym] ||= {}
     end
   end
 
@@ -58,21 +104,48 @@ module AfterMigrate
     end
 
     def store
-      key = [configuration.store.to_s, configuration.store_path.to_s, configuration.run_id.to_s]
-      @store = nil if @store_key != key
-      @store_key = key
+      @store = nil if @store_key != store_key
+      @store_key = store_key
       @store ||= build_store
     end
 
     private
 
+    def store_key
+      [
+        configuration.store.to_s,
+        configuration.run_id.to_s,
+        configuration.store_options_for(configuration.store).hash
+      ]
+    end
+
     def build_store
       case configuration.store.to_s
       when 'file'
-        Stores::FileStore.new(path: configuration.store_path, run_id: configuration.run_id)
+        file_store
+      when 'redis'
+        redis_store
       else
         Stores::Memory.new
       end
     end
+
+    def file_store
+      options = configuration.store_options_for(:file)
+      Stores::FileStore.new(
+        path: options.fetch(:path),
+        run_id: options.fetch(:run_id, configuration.run_id)
+      )
+    end
+
+    def redis_store
+      options = configuration.store_options_for(:redis)
+      Stores::RedisStore.new(
+        redis: options[:client],
+        key_prefix: options.fetch(:key_prefix),
+        run_id: options.fetch(:run_id, configuration.run_id),
+        ttl: options.fetch(:ttl)
+      )
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: after_migrate
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.2.4
 platform: ruby
 authors:
 - Nikolay Moskvin