after_migrate 0.2.2 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 62f5a309584cee52788e0ea45c222648cc58c4db46bc08ce6188c6325f6c62dc
-  data.tar.gz: e36efca18e936086cdace33a4fe422c2d34f63502ddf01c53090cf8048ed92c6
+  metadata.gz: b9d807edc34ee3c61bcda0ef4fd2319f80bac36c3bcd1cd1e2a46eb9f5e44f28
+  data.tar.gz: 16547159bf93c5a192a38a1e92d441bcd9690412b8d58ae163f59f8bc2da71bb
 SHA512:
-  metadata.gz: 2590c965f71fc07b5171cea88869115516687331138b0a6a20b5c4bdd3008a13be1c7a7382dd8886112a02e076904d57f59e2f24db44bfa23d1c36dd30700b40
-  data.tar.gz: 4a2f1dd2df2bb18e8f663b1ca150db22a6d4d5474b651ea0ee693129640926266ce77168e370921cab4ac214620a601fe02f6204a97829575ec7e417c944a7d4
+  metadata.gz: a7e76304080055fe4b9669d9895b9fb3709266c89e29deb2e39b3d408a29ba88ef1b99ebe951e693da446d2c97fda40ba416b71fae517234b65975e250b6635d
+  data.tar.gz: 14823ec925953300f85d28586a232111f945c2b9b8fb20bdaa4bf9d3057f8467dec2eb76defda50317a5811bbd6f7071bf71bb195f188950ab2a340030e0b2a5

data/CHANGELOG.md

@@ -2,6 +2,18 @@
 
 All notable changes to this project will be documented in this file.
 
+## [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.3'
 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, :store_path, :run_id, :redis, :redis_key_prefix, :redis_ttl
 
     def initialize
       @enabled = true
@@ -20,6 +21,9 @@ module AfterMigrate
       @store = :memory
       @store_path = 'tmp/after_migrate/affected_tables.json'
       @run_id = nil
+      @redis = nil
+      @redis_key_prefix = 'after_migrate'
+      @redis_ttl = 24 * 60 * 60
     end
   end
 
@@ -58,21 +62,41 @@ 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.store_path.to_s,
+        configuration.run_id.to_s,
+        configuration.redis_key_prefix.to_s,
+        configuration.redis_ttl.to_s
+      ]
+    end
+
     def build_store
       case configuration.store.to_s
       when 'file'
         Stores::FileStore.new(path: configuration.store_path, run_id: configuration.run_id)
+      when 'redis'
+        redis_store
       else
         Stores::Memory.new
       end
     end
+
+    def redis_store
+      Stores::RedisStore.new(
+        redis: configuration.redis,
+        key_prefix: configuration.redis_key_prefix,
+        run_id: configuration.run_id,
+        ttl: configuration.redis_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.3
 platform: ruby
 authors:
 - Nikolay Moskvin