pgbus 0.3.1 → 0.3.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6daa2060697774db5018b30b2d7b08ef4f3f962332e627f8fa90ce49cfe136da
-  data.tar.gz: 9fe928dba25e277212093a6e1a0443c6451336b2bc996e06375d88eaaf2dd148
+  metadata.gz: d895049be3463d74d039e29485fc7d26c4c06ca96419e8a46327a4be6538db1f
+  data.tar.gz: f685673eaf24780e820486b9eb1f752493caf1765241b0063293d1d8d4b98e25
 SHA512:
-  metadata.gz: 1977d39318814b8878eef04065e4abf2a73588005542e5b15b679248848fd470c11b685a3a7c2d89d26ecbee0b897fa2968d498b757b0435bf213eba61e48c3d
-  data.tar.gz: 0d999aa34400b0a9f85e030d9cd5ac9f5bb0b57e467eee50615a3cf6f19ee70f8c33da688b3d78abf575c4c053e0dab7ecc3fbc2c841390c97d9089b2f238135
+  metadata.gz: 33cb5c4fbf7ae36a8bb82a780d39bd0a32fd2c3167d0d7aaaed74a60608778a90e32eac53c1db2808ffa5956184298a83478488a8d11edcc89b052ec5fe8b0a5
+  data.tar.gz: 4a9cc394b02ee99e1681ab26a86db25c57f38f92e901dab0c2c48c1025224bb4089009abcd865a2432c3b37a988b51511faa64be976709069a4dbef70e3ec5ed

data/Rakefile

@@ -31,7 +31,12 @@ namespace :bench do
     ruby "benchmarks/memory_profile.rb"
   end
 
-  desc "Run all benchmarks"
+  desc "Run integration benchmarks (requires PGBUS_DATABASE_URL)"
+  task :integration do
+    ruby "benchmarks/integration_bench.rb"
+  end
+
+  desc "Run all benchmarks (unit-level, no DB required)"
   task all: %i[serialization client executor]
 end
 

data/app/models/pgbus/job_lock.rb

@@ -16,42 +16,50 @@ module Pgbus
     # Atomically try to acquire a lock.
     # Cleans up expired locks for this key first (crash recovery at acquire time).
     # Returns true if acquired, false if already locked.
-    def self.acquire!(lock_key, job_class:, ttl:, job_id: nil, state: "queued", owner_pid: nil, owner_hostname: nil)
+    #
+    # Uses raw SQL on the hot path to minimize ActiveRecord allocations
+    # (~29 objects vs ~304 per acquire+release cycle with AR query builder).
+    def self.acquire!(lock_key, job_class:, ttl:, job_id: nil, state: "queued", owner_pid: nil, owner_hostname: nil) # rubocop:disable Naming/PredicateMethod
+      expires_at = Time.current + ttl
+
       # Remove any expired lock for this key inline (last-resort TTL recovery)
-      where(lock_key: lock_key).where("expires_at < ?", Time.current).delete_all
+      connection.exec_delete(
+        "DELETE FROM #{table_name} WHERE lock_key = $1 AND expires_at < $2",
+        "JobLock Expire", [lock_key, Time.current]
+      )
 
-      result = insert(
-        {
-          lock_key: lock_key, job_class: job_class, job_id: job_id,
-          state: state, owner_pid: owner_pid, owner_hostname: owner_hostname,
-          expires_at: Time.current + ttl
-        },
-        unique_by: :lock_key
+      result = connection.exec_query(
+        "INSERT INTO #{table_name} (lock_key, job_class, job_id, state, owner_pid, owner_hostname, expires_at) " \
+        "VALUES ($1, $2, $3, $4, $5, $6, $7) ON CONFLICT (lock_key) DO NOTHING RETURNING id",
+        "JobLock Acquire", [lock_key, job_class, job_id, state, owner_pid, owner_hostname, expires_at]
       )
       result.rows.any?
-    rescue ActiveRecord::RecordNotUnique
-      false
     end
 
     # Transition a queued lock to executing state and claim ownership.
     # Called when a worker starts executing a job that was locked at enqueue time.
     def self.claim_for_execution!(lock_key, owner_pid:, owner_hostname:, ttl:)
-      where(lock_key: lock_key).update_all(
-        state: "executing",
-        owner_pid: owner_pid,
-        owner_hostname: owner_hostname,
-        expires_at: Time.current + ttl
+      connection.exec_update(
+        "UPDATE #{table_name} SET state = $1, owner_pid = $2, owner_hostname = $3, expires_at = $4 " \
+        "WHERE lock_key = $5",
+        "JobLock Claim", ["executing", owner_pid, owner_hostname, Time.current + ttl, lock_key]
       )
     end
 
     # Release a lock by key.
     def self.release!(lock_key)
-      where(lock_key: lock_key).delete_all
+      connection.exec_delete(
+        "DELETE FROM #{table_name} WHERE lock_key = $1",
+        "JobLock Release", [lock_key]
+      )
     end
 
     # Check if a lock is currently held (regardless of expiry — reaper handles orphans).
     def self.locked?(lock_key)
-      where(lock_key: lock_key).exists?
+      result = connection.select_value(
+        "SELECT 1 FROM #{table_name} WHERE lock_key = $1 LIMIT 1", "JobLock Check", [lock_key]
+      )
+      !result.nil?
     end
 
     # Reap orphaned locks: locks in 'executing' state whose owner_pid

data/lib/pgbus/version.rb

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

metadata

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