pgbus 0.3.8 → 0.3.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 76f2d5b122b3a379c36fcef393e8b7eabcfdbc7c8534d76948b1349dd97cb58b
-  data.tar.gz: 4059948c68407aebeebd6f45d3b550158c6d38455c95cf255ccf1faf2d10889a
+  metadata.gz: 0dd5ee830e98f086b75601600307b11a2729d20076495563151ba6a0745b43bd
+  data.tar.gz: 2bb7b5f38f4a2c3186aae1406f7aa843a7d18601aab1065dbbcc3303784682e8
 SHA512:
-  metadata.gz: b21741c2de21716ce47c0246c8de08bb440e9a4dfc750bc60e95484b97fe9c7626b357c47c4b2b7583c6d8273fef08b363aa84ed5424bf21a64bb64519ce64b7
-  data.tar.gz: 48983a1abcbfc000936caf9ff2cac0d7451633aee2ee05c68c4168b84c0aed4ee38e4108501214b91c5045e9cc97f269f55c3e9e6050d778ac06b07a11c6c45d
+  metadata.gz: e33636a93205ce7bb09bef53b626bb275905c980cf6792ec9716baed0bc3265d150816dcaffd5f77f6cd08541ab03fceb0cf76315e9bce9ad4bfe76035b4dad8
+  data.tar.gz: 1aff3f96586bc6a8593cac6cb7ea7f7206899ee1d43907b3717f9cd1433cc9c818537f6ad1d8ecad2dad7f492dbb216d17c0e4b80dd0276f6a45b5e4c619dc6d

data/lib/pgbus/configuration.rb

@@ -202,15 +202,49 @@ module Pgbus
       elsif connection_params
         connection_params
       elsif defined?(ActiveRecord::Base)
-        if connects_to
-          -> { Pgbus::BusRecord.connection.raw_connection }
-        else
-          -> { ActiveRecord::Base.connection.raw_connection }
-        end
+        # Extract connection config from ActiveRecord so pgmq-ruby creates its
+        # own dedicated PG connections. Sharing AR's raw_connection via a Proc
+        # is NOT thread-safe: the ConnectionPool caches the PG::Connection from
+        # whichever thread first called the Proc, then hands that same object to
+        # other threads — causing nil results, segfaults, and data corruption
+        # when AR and PGMQ issue concurrent queries on the same connection.
+        extract_ar_connection_hash
       else
         raise ConfigurationError, "No database connection configured. " \
                                   "Set Pgbus.configuration.database_url, connection_params, or use with Rails."
       end
     end
+
+    private
+
+    def extract_ar_connection_hash
+      base = connects_to ? Pgbus::BusRecord : ActiveRecord::Base
+      db_config = base.connection_db_config
+
+      # Rails 7.1+ db_config.configuration_hash returns the full config
+      config_hash = db_config.configuration_hash
+
+      {
+        host: config_hash[:host] || "localhost",
+        port: (config_hash[:port] || 5432).to_i,
+        dbname: config_hash[:database],
+        user: config_hash[:username],
+        password: config_hash[:password]
+      }.compact
+    rescue StandardError => e
+      # Fallback to Proc path if AR config extraction fails (e.g., adapter
+      # doesn't expose standard config keys). Log a warning since this path
+      # is not thread-safe.
+      Pgbus.logger.warn do
+        "[Pgbus] Could not extract AR connection config (#{e.class}: #{e.message}). " \
+          "Falling back to shared raw_connection — this is NOT thread-safe with multiple workers. " \
+          "Set Pgbus.configuration.database_url explicitly for thread-safe operation."
+      end
+      if connects_to
+        -> { Pgbus::BusRecord.connection.raw_connection }
+      else
+        -> { ActiveRecord::Base.connection.raw_connection }
+      end
+    end
   end
 end

data/lib/pgbus/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Pgbus
-  VERSION = "0.3.8"
+  VERSION = "0.3.9"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pgbus
 version: !ruby/object:Gem::Version
-  version: 0.3.8
+  version: 0.3.9
 platform: ruby
 authors:
 - Mikael Henriksson