riverqueue-sequel 0.13.0 → 0.15.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bb0d975eb6921905735667d53aeba24da81b090f92af870935bd112cbc9a7b5e
-  data.tar.gz: 9f4b514ebc0ac579caeb7eacec5227a9e9e2b497cbfbd9d15366a3795e271ee9
+  metadata.gz: 7a3869013c8703e42675430a9ac21c77baf72994de9a8398c524ff080b09b374
+  data.tar.gz: 4282e30b7d4fe282f9600cf10b4ed86cb415930d861070d0bf4c80086e99827e
 SHA512:
-  metadata.gz: 9dc23c4d8a76b28a7d39e3422b011db46d648c3e5c7adfd652e6fa391643a003617362f5edc1e17e71722c1aa5f52d0db35fc95315242c28ab9d4f534c557620
-  data.tar.gz: 8fdce85797d7deefb7b82223ac81531bc61962b32a43b96bd29452a8f88e8352dedd2fac1da555e5e194991f281f9c3d5d4468bed82a2c6410404218b63adaf1
+  metadata.gz: '081f77c7894c5ca3418c0ebb47d8c233b6a58554b2bfcec04aa7c89e901be347c3e69accb212749fa0f46854051544a68104c6816af397c45cd3efe232d4d4c9'
+  data.tar.gz: 21a5b271f12cbdedc4bf93803db4bab136ea35777cd101eb4723fd39b333a2dd54b81389c891017404d2f7c5b0c6bee929bff011b3f75cbd53eae35c996ad979

data/lib/driver.rb

@@ -3,7 +3,7 @@
 require "securerandom"
 
 module River::Driver
-  # Provides a Sequel driver for River supporting PostgreSQL, YugabyteDB, and SQLite.
+  # Provides a Sequel driver for River supporting Postgres, YugabyteDB, and SQLite.
   #
   # Used in conjunction with a River client like:
   #
@@ -18,7 +18,7 @@ module River::Driver
   class Sequel
     include River::Driver::Runtime
 
-    # Creates a driver backed by a connected Sequel::Database for PostgreSQL or
+    # Creates a driver backed by a connected Sequel::Database for Postgres or
     # SQLite.
     def initialize(db)
       @db = db
@@ -31,6 +31,16 @@ module River::Driver
       end
     end
 
+    # Runs the callback only if this transaction and its savepoints commit.
+    def after_commit(&block)
+      @db.after_commit(savepoint: true, &block)
+    end
+
+    # Reports whether this thread has a transaction on the configured database.
+    def in_transaction?
+      @db.in_transaction?
+    end
+
     def job_get_by_id(id)
       if @is_sqlite
         row = sqlite_job_rows("WHERE id = ? LIMIT 1", id).first
@@ -79,11 +89,13 @@ module River::Driver
       end
     end
 
-    # Runs the block in a Sequel transaction or savepoint and returns the
-    # block's result.
+    # Runs the block in a Sequel transaction and returns the block's result.
+    # Joins an open transaction instead of opening a savepoint, unless that
+    # transaction uses Sequel's `auto_savepoint`.
     def transaction
       block_error = nil
-      @db.transaction(savepoint: true) do
+      options = runtime_legacy_subtransactions? ? {savepoint: true} : {}
+      @db.transaction(**options) do
         yield
       rescue ArgumentError => error
         block_error = error
@@ -145,7 +157,7 @@ module River::Driver
     private_constant :SQLITE_CONFLICT_WHERE, :SQLITE_JOB_COLUMNS, :UNIQUE_INSERT_METADATA_KEY
 
     private def format_time(time)
-      time.getutc.round(3).strftime("%Y-%m-%d %H:%M:%S.%3N")
+      normalize_time(time).strftime("%Y-%m-%d %H:%M:%S.%3N")
     end
 
     private def postgres_insert_params_to_hash(insert_params, nonce)
@@ -209,6 +221,15 @@ module River::Driver
       end
     end
 
+    private def runtime_notification_connection
+      params, schema = @db.synchronize do |connection|
+        [connection.conninfo_hash, connection.exec("SELECT current_schema()").getvalue(0, 0)]
+      end
+      params.compact!
+      params[:connect_timeout] = "5" unless params[:connect_timeout].to_i.positive?
+      [::PG.connect(params), schema]
+    end
+
     private def runtime_postgres?
       !@is_sqlite
     end
@@ -221,17 +242,16 @@ module River::Driver
       @db.literal(value)
     end
 
-    private def runtime_unique_violation_class
-      ::Sequel::UniqueConstraintViolation
-    end
-
     private def runtime_value(row, key)
       row[key] || row[key.to_s]
     end
 
     private def sqlite_insert_params_to_hash(insert_params, nonce)
       {
-        args: JSON.parse(insert_params.encoded_args),
+        # Embed the encoded args as a JSON string so they reach SQLite
+        # byte-for-byte. `json_extract` unwraps the string back to the
+        # encoded JSON text before `jsonb` parses it.
+        args: insert_params.encoded_args,
         kind: insert_params.kind,
         max_attempts: insert_params.max_attempts,
         metadata: {UNIQUE_INSERT_METADATA_KEY => nonce},
@@ -250,7 +270,7 @@ module River::Driver
     # matching migration 007 and newer River databases.
     private def sqlite_job_insert_many(insert_params_array)
       nonces = sqlite_insert_nonces(insert_params_array)
-      @db.transaction(savepoint: true) do
+      transaction do
         jobs = insert_params_array.zip(nonces).map { |param, nonce| sqlite_insert_params_to_hash(param, nonce) }
         inserted_nonces = nonces.to_h { |nonce| [nonce, true] }
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: riverqueue-sequel
 version: !ruby/object:Gem::Version
-  version: 0.13.0
+  version: 0.15.0
 platform: ruby
 authors:
 - Blake Gentry
@@ -14,9 +14,9 @@ dependencies:
   name: sequel
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '5.20'
     - - "<"
       - !ruby/object:Gem::Version
         version: '1000'
@@ -24,9 +24,9 @@ dependencies:
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '5.20'
     - - "<"
       - !ruby/object:Gem::Version
         version: '1000'
@@ -36,15 +36,15 @@ dependencies:
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 0.13.0
+        version: 0.15.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 0.13.0
-description: Sequel PostgreSQL and SQLite driver for inserting and working River jobs
+        version: 0.15.0
+description: Sequel Postgres and SQLite driver for inserting and working River jobs
   in Ruby.
 email: brandur@brandur.org
 executables: []
@@ -73,5 +73,5 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 rubygems_version: 4.0.9
 specification_version: 4
-summary: Sequel PostgreSQL and SQLite driver for the River Ruby gem.
+summary: Sequel Postgres and SQLite driver for the River Ruby gem.
 test_files: []