polyrun 2.2.0 → 2.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8b7bedc5bda9c9ee341927dab2ba6aedafaebe4c26ad73750d8ed7ad4845f4e3
-  data.tar.gz: bac9cd71c5ab3372c0ff5dda891dd354a0ceaac3555f5949870ce8611646debc
+  metadata.gz: cf97ce76499f3784cf3a11d8e1ebf640631dc8247785ff74a518b70fb3218459
+  data.tar.gz: c4cc16110c1eb5696b1fab2fa771780a4102856909b8ecb620f2f2da3540b758
 SHA512:
-  metadata.gz: fe09eaccf1eaf3fba5a998d49f2f2df9cfd6a1da6b1eef960e6cbe894f2269b05777be459fe394a67c3038d72750a8401ae8e224726901ac4bd437e3d01a93ed
-  data.tar.gz: f8090b77fba080b7ca1996afeec0c2680ca26e0d0a4571453210f421f4921685bd8c5bcfeb79724556e377ff26968a914b52bf24b4a1a433a9500d656aa43f41
+  metadata.gz: 4d24dd153dcb88eb2e1bd7092fd47d72268ca6848061af4b64616fc6ea1307594086cfaa84e6ea06908a82b037b6404684315e66ed3cd7e923fbbaa50409e2f0
+  data.tar.gz: 9569227c3d67cc36eb4e79df95b36c9357aed6c79e853a428546d2c270ccd6accd900a6a8869b250cab3c2bbcf91021d29a4d9a9d5b53292355a6981229e4b64

data/CHANGELOG.md

@@ -1,5 +1,10 @@
 # CHANGELOG
 
+## 2.2.1 (2026-07-13)
+
+- Fix per-example timeout leaving ActiveRecord connection pools in a bad state after `Timeout` interrupts database I/O; disconnect all pools on timeout (adapter-agnostic).
+- Fix `DEBUG_PROSOPITE=1`, `DEBUG_SQL` / `POLYRUN_DEBUG_SQL`, and `DEBUG_TRACE` / `POLYRUN_DEBUG_TRACE` requiring `POLYRUN_EXAMPLE_DEBUG=1`; each flag is standalone again.
+
 ## 2.2.0 (2026-07-13)
 
 - Fix `install_example_rails_logging!` raising `NameError` when example-debug Rails logging runs under `POLYRUN_EXAMPLE_DEBUG=1`.

data/docs/SETUP_PROFILE.md

@@ -69,14 +69,14 @@ Orchestration tracing stays on `DEBUG=1` / `POLYRUN_DEBUG=1`. Per-example toolin
 
 | Env | Effect |
 |--------|--------|
-| `POLYRUN_EXAMPLE_DEBUG=1` | Enable example debug installers |
+| `POLYRUN_EXAMPLE_DEBUG=1` | Enable print helpers, Rails logging; disable per-example timeouts |
 | `POLYRUN_DEBUG_SQL=1` or `DEBUG_SQL=1` | Log mutating SQL |
 | `POLYRUN_DEBUG_TRACE=1` or `DEBUG_TRACE=1` | TracePoint under app root |
 | `DEBUG_PROSOPITE=1` | Prosopite N+1 scan when gem is loaded |
 | `DEBUG_LOG_LEVEL` | Rails / ActiveRecord logger severity: `debug`, `info`, `warn`, `error`, `fatal`, or Ruby Logger integers (`0` debug … `4` fatal); default `debug` |
-| `RSPEC_EXAMPLE_TIMEOUT_SEC` | Per-example timeout (disabled while example debug is on) |
+| `RSPEC_EXAMPLE_TIMEOUT_SEC` | Per-example timeout (disabled while example debug is on); on timeout, ActiveRecord connection pools are disconnected so the next example gets fresh connections (any adapter) |
 
-In `spec_helper` / `spec/support`: `require "polyrun/rspec"` then `Polyrun::RSpec.install_sharded_formatter_compat!` (silences per-worker seed/summary/pending under `POLYRUN_SHARD_*`), `install_example_debug!`, `install_example_rails_logging!`, `install_example_timeout!`. Pair with `install_worker_ping!` for idle timeouts.
+In `spec_helper` / `spec/support`: `require "polyrun/rspec"` then `Polyrun::RSpec.install_sharded_formatter_compat!` (silences per-worker seed/summary/pending under `POLYRUN_SHARD_*`), `install_example_debug!`, `install_example_rails_logging!`, `install_example_prosopite!`, `install_example_timeout!`. Pair with `install_worker_ping!` for idle timeouts.
 
 ## 6. Coverage and CI reports
 

data/lib/polyrun/cli/help.rb

@@ -26,7 +26,7 @@ module Polyrun
           Per-worker wall timeout: --worker-timeout SEC or POLYRUN_WORKER_TIMEOUT_SEC. Exit 124; parent stops remaining workers.
           Per-worker idle timeout: --worker-idle-timeout SEC or POLYRUN_WORKER_IDLE_TIMEOUT_SEC after a progress ping (POLYRUN_WORKER_PING_FILE). Enable pings in test setup. Exit 125. Optional periodic pings: POLYRUN_WORKER_PING_THREAD=1 (POLYRUN_WORKER_PING_INTERVAL_SEC).
           Worker output routing (opt-in): POLYRUN_WORKER_OUTPUT_ROUTING=1 or POLYRUN_WORKER_LOG_DIR; per-shard logs under tmp/polyrun/workers (POLYRUN_WORKER_OUTPUT_PREFIX=0 for log-only)
-          Example debug (RSpec, opt-in): POLYRUN_EXAMPLE_DEBUG=1; POLYRUN_DEBUG_SQL / POLYRUN_DEBUG_TRACE
+          Example debug (RSpec, opt-in): POLYRUN_EXAMPLE_DEBUG=1 (print/Rails logging/timeouts); POLYRUN_DEBUG_SQL / DEBUG_SQL; POLYRUN_DEBUG_TRACE / DEBUG_TRACE; DEBUG_PROSOPITE=1
           Sharded formatter compat: silences per-worker seed, summary, and pending lines under POLYRUN_SHARD_* (see docs/SETUP_PROFILE.md)
           Orchestration warnings on process stderr: POLYRUN_ORCHESTRATION_STDERR=1
           Spec quality (opt-in): POLYRUN_SPEC_QUALITY=1; run-shards --merge-spec-quality; merge-spec-quality / report-spec-quality

data/lib/polyrun/rspec/active_record_timeout_recovery.rb

@@ -0,0 +1,51 @@
+module Polyrun
+  module RSpec
+    # Drops pooled connections after per-example Timeout interrupts database I/O.
+    # Uses only ActiveRecord connection-pool APIs (PostgreSQL, MySQL, SQLite, SQL Server, etc.).
+    module ActiveRecordTimeoutRecovery
+      module_function
+
+      def disconnect_all_connection_pools!
+        return unless active_record_loaded?
+
+        if defined?(ActiveRecord) && ActiveRecord.respond_to?(:disconnect_all!)
+          ActiveRecord.disconnect_all!
+          return
+        end
+
+        each_connection_pool(&:disconnect!)
+      end
+
+      def active_record_loaded?
+        defined?(ActiveRecord::Base) &&
+          ActiveRecord::Base.respond_to?(:connection_handler)
+      end
+
+      def each_connection_pool
+        connection_handlers.each do |handler|
+          pools_for(handler).each { |pool| yield pool }
+        end
+      end
+
+      def connection_handlers
+        base = ActiveRecord::Base
+        if base.respond_to?(:connection_handlers)
+          base.connection_handlers.values
+        else
+          [base.connection_handler]
+        end
+      end
+
+      def pools_for(handler)
+        if handler.respond_to?(:all_connection_pools)
+          handler.all_connection_pools
+        elsif handler.respond_to?(:connection_pool_list)
+          handler.connection_pool_list
+        else
+          []
+        end
+      end
+      private_class_method :active_record_loaded?, :each_connection_pool, :connection_handlers, :pools_for
+    end
+  end
+end

data/lib/polyrun/rspec/example_debug.rb

@@ -6,9 +6,10 @@ module Polyrun
   module RSpec
     # Per-example RSpec debugging for local investigation (SQL, TracePoint, timeouts).
     #
-    # +POLYRUN_EXAMPLE_DEBUG=1+ enables installers; +DEBUG=1+ / +POLYRUN_DEBUG=1+ trace orchestration only.
+    # +POLYRUN_EXAMPLE_DEBUG=1+ enables print helpers, Rails logging, and disables per-example timeouts.
     # +POLYRUN_DEBUG_SQL=1+ or legacy +DEBUG_SQL=1+ logs mutating SQL.
     # +POLYRUN_DEBUG_TRACE=1+ or legacy +DEBUG_TRACE=1+ traces :call/:raise under the app root.
+    # +DEBUG_PROSOPITE=1+ runs Prosopite N+1 scans when the gem is loaded.
     # +DEBUG_LOG_LEVEL+ accepts Ruby Logger severities as integers (0 debug … 4 fatal) or names.
     module ExampleDebug
       LOG_LEVEL_BY_NAME = {
@@ -26,15 +27,15 @@ module Polyrun
       end
 
       def sql_enabled?
-        enabled? && (truthy?(ENV["POLYRUN_DEBUG_SQL"]) || truthy?(ENV["DEBUG_SQL"]))
+        truthy?(ENV["POLYRUN_DEBUG_SQL"]) || truthy?(ENV["DEBUG_SQL"])
       end
 
       def trace_enabled?
-        enabled? && (truthy?(ENV["POLYRUN_DEBUG_TRACE"]) || truthy?(ENV["DEBUG_TRACE"]))
+        truthy?(ENV["POLYRUN_DEBUG_TRACE"]) || truthy?(ENV["DEBUG_TRACE"])
       end
 
       def prosopite_enabled?
-        enabled? && truthy?(ENV["DEBUG_PROSOPITE"])
+        truthy?(ENV["DEBUG_PROSOPITE"])
       end
 
       def print_spec_enabled?
@@ -99,14 +100,22 @@ module Polyrun
         return if seconds <= 0
         return if example_timeout_disabled?
 
+        require_relative "active_record_timeout_recovery"
+
         rspec_config.around(:each) do |example|
-          if example.metadata[:benchmark] || example.metadata[:slow]
-            example.run
-          else
-            Timeout.timeout(seconds) { example.run }
+          timed_out = false
+          begin
+            if example.metadata[:benchmark] || example.metadata[:slow]
+              example.run
+            else
+              Timeout.timeout(seconds) { example.run }
+            end
+          rescue Timeout::Error
+            timed_out = true
+            raise "Example timed out after #{seconds}s (#{example.location})"
+          ensure
+            ActiveRecordTimeoutRecovery.disconnect_all_connection_pools! if timed_out
           end
-        rescue Timeout::Error
-          raise "Example timed out after #{seconds}s (#{example.location})"
         end
       end
 

data/lib/polyrun/version.rb

@@ -1,3 +1,3 @@
 module Polyrun
-  VERSION = "2.2.0"
+  VERSION = "2.2.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: polyrun
 version: !ruby/object:Gem::Version
-  version: 2.2.0
+  version: 2.2.1
 platform: ruby
 authors:
 - Andrei Makarov
@@ -338,6 +338,7 @@ files:
 - lib/polyrun/reporting/rspec_failure_fragment_formatter.rb
 - lib/polyrun/reporting/rspec_junit.rb
 - lib/polyrun/rspec.rb
+- lib/polyrun/rspec/active_record_timeout_recovery.rb
 - lib/polyrun/rspec/example_debug.rb
 - lib/polyrun/rspec/example_debug_instrumentation.rb
 - lib/polyrun/rspec/sharded_formatter_compat.rb