after_migrate 0.2.4 → 0.2.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 52bea11a45a6765c9bff5e82d42d527595e7ef6c91e84028ca2dde978d8d527e
-  data.tar.gz: 3dcd46239d300faa855998e427585870899a0ea0d013e21fb6eb770ecd5f5be4
+  metadata.gz: 23144c2fac25e5be5c8965155dc54b03b8859063fe00a43d033e9c3bed415f8a
+  data.tar.gz: e40e59a694376551d73261e59d214e8ebfe460ee0ab5961116873a22bc9b019f
 SHA512:
-  metadata.gz: e21f8c6f9db4d2db100b5ef4c93f87ce0a7b69af5a7da29b79ce91a4a47e585ffdce9f33726929aed3962693ad9afee40418a40dc76a813dc2ee717b5333a59d
-  data.tar.gz: d6da80e272cc1895bb753b6fa01b423fc8eb9c3ea4f93d34d691f64743d1269689c3147764d953114957907d36fea5cac8910b03baac009fea08984924c4154b
+  metadata.gz: a1ff650538cacd552d26399df13507a0e4f5a71581e32bff0d1e2686ea3d8b8a86e44ef96a7143892ef7cef79a720135489ac174b09ade876b722465b916fc18
+  data.tar.gz: 977f7a020023cad34941b24046982d0563f30340cfefe6a196feb83915c7ec5d6a148707cc596749ad0d2ced9a5e64ec55c926d6869d3bd0c853edbd4617dbd8

data/.gitlocalexclude

@@ -0,0 +1,2 @@
+.ai/
+.claude/

data/CLAUDE.md

@@ -22,7 +22,7 @@ This is a Rails gem that automatically runs database maintenance (`ANALYZE`, `VA
 
 2. **`Collector`** (`lib/after_migrate/collector.rb`) — receives every SQL notification, filters to DDL/DML statements (CREATE/ALTER/DROP/INSERT/UPDATE/DELETE), calls the adapter-specific parser, and accumulates table names into `AfterMigrate.affected_tables` (a `Concurrent::Map<schema, Concurrent::Set<table_name>>`).
 
-3. **`AfterMigrate.store`** (`lib/after_migrate/store.rb`) — defaults to an in-memory `Concurrent::Map` wrapped by `AfterMigrate::Stores::Memory`. `AfterMigrate.affected_tables`, `merge_tables`, and `reset!` delegate to the store. The memory store persists across multiple rake task invocations inside one Ruby process and is cleared by `AfterMigrate.run!` (or `AfterMigrate.reset!`). This replaces the old `Current` (`ActiveSupport::CurrentAttributes`) which was reset after every migration.
+3. **`AfterMigrate.store`** (`lib/after_migrate/store.rb`) — defaults to an in-memory `Concurrent::Map` wrapped by `AfterMigrate::Stores::Memory`. `AfterMigrate.affected_tables`, `merge_tables`, and `reset!` delegate to the store. The memory store persists across multiple rake task invocations inside one Ruby process and is cleared by `AfterMigrate.run!` (or `AfterMigrate.reset!`). This replaces the old `Current` (`ActiveSupport::CurrentAttributes`) which was reset after every migration. Alternate backends: `FileStore` (JSON, cross-invocation) and `RedisStore` (cross-process, for parallel/multi-tenant runs). `RedisStore#resolved_redis` memoizes the resolved client per store instance, so `config.redis` is only materialized once — pass a `ConnectionPool` (the store calls `pool.with`) or a memoized client, never a `-> { Redis.new }` proc that builds a connection per call (that opens one connection per SQL statement and exhausts the Redis connection limit under concurrency).
 
 4. **`Executor`** (`lib/after_migrate/executor.rb`) — iterates `AfterMigrate.affected_tables` and calls the correct adapter's `optimize_tables`. Respects the `analyze` config option (`only_affected_tables` / `all_tables` / `none`). Always calls `AfterMigrate.reset!` in its `ensure` block.
 
@@ -45,13 +45,13 @@ This is a Rails gem that automatically runs database maintenance (`ANALYZE`, `VA
 
 Config values are in `AfterMigrate::Configuration` (initialized in `lib/after_migrate.rb`):
 
-| Option                | Default                  | Values                                             |
-|-----------------------|--------------------------|----------------------------------------------------|
-| `enabled`             | `true`                   | bool                                               |
-| `verbose`             | `true`                   | bool                                               |
-| `vacuum`              | `true`                   | bool (PostgreSQL only)                             |
-| `analyze`             | `"only_affected_tables"` | `"only_affected_tables"`, `"all_tables"`, `"none"` |
-| `rake_tasks_enhanced` | `true`                   | bool                                               |
+| Option                | Default                  | Values                                                  |
+|-----------------------|--------------------------|---------------------------------------------------------|
+| `enabled`             | `true`                   | bool                                                    |
+| `verbose`             | `true`                   | bool                                                    |
+| `vacuum`              | `true`                   | bool (PostgreSQL only)                                  |
+| `analyze`             | `"only_affected_tables"` | `"only_affected_tables"`, `"all_tables"`, `"none"`      |
+| `rake_tasks_enhanced` | `true`                   | bool                                                    |
 | `defer`               | `true`                   | bool — skip auto-run; call `AfterMigrate.run!` manually |
 
 ## Multi-tenant usage

data/README.md

@@ -81,6 +81,41 @@ AfterMigrate.configure do |config|
   config.rake_tasks_enhanced = true
 end
 ```
+
+### Store backends
+
+The collected table names are kept in a store. Choose one with `config.store`:
+
+- `:memory` (default) — a process-local `Concurrent::Map`. Fine for a single-process migration.
+- `:file` — JSON file, shared across rake invocations in the same run.
+- `:redis` — shared across processes; use this for parallel / multi-tenant migrations.
+
+```ruby
+config.store = :redis
+```
+
+> [!IMPORTANT]
+> **Redis: always pass a pooled or memoized client — never a proc that builds a new connection per call.**
+>
+> `config.redis` is read on *every* collected SQL statement. If it returns a
+> fresh `Redis.new` each time (e.g. `config.redis = -> { Redis.new(...) }`),
+> every statement opens a new, unpooled connection. Under parallel or
+> multi-tenant migrations this exhausts the Redis server connection limit
+> (`Connection refused` / `max number of clients reached`).
+>
+> Recommended — a thread-safe `ConnectionPool` (the store uses `pool.with`):
+>
+> ```ruby
+> config.store = :redis
+> config.redis = ConnectionPool.new(size: 2, timeout: 5) { Redis.new(url: ENV["REDIS_URL"]) }
+> ```
+>
+> A memoized single client also works for sequential migrations:
+>
+> ```ruby
+> config.redis = -> { @after_migrate_redis ||= Redis.new(url: ENV["REDIS_URL"]) }
+> ```
+
 ---
 
 ## 🚢 Releasing

data/lib/after_migrate/store.rb

@@ -181,12 +181,18 @@ module AfterMigrate
         block.call(redis)
       end
 
+      # Memoized so a plain client or a `-> { Redis.new }` proc only ever
+      # produces one connection per store instance. Without this, `with_redis`
+      # re-invokes the proc on every merge_tables/affected_tables call, opening
+      # a fresh unpooled connection per SQL statement — which exhausts the
+      # Redis server's connection limit under parallel/multi-tenant migrations.
+      # Pass a ConnectionPool as config.redis for thread-safe concurrency.
       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'
+        @resolved_redis ||= begin
+          client = @redis.respond_to?(:call) ? @redis.call : @redis
+          client ||= Redis.new if defined?(Redis)
+          client || 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.4'
+  VERSION = '0.2.5'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: after_migrate
 version: !ruby/object:Gem::Version
-  version: 0.2.4
+  version: 0.2.5
 platform: ruby
 authors:
 - Nikolay Moskvin
@@ -48,6 +48,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".gitignore"
+- ".gitlocalexclude"
 - ".rspec"
 - ".rubocop.yml"
 - ".ruby-version"