riverqueue-sequel 0.6.1 → 0.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8436cdaf040ad522e33e795f753af4fdb39096899c3be0ceef7c21c7007c3001
-  data.tar.gz: 5fd60f6832b705e7a7ac762a6c987e87364a9ced39b59e086d0e082c07eeeccb
+  metadata.gz: dcba38bb9184685170e02a42f53ef69efa47ef25c364cfa6047d7ed7b457ee01
+  data.tar.gz: 95e6cc145f3faa7632a3ab8e25d0896537d54a34756ec0b246028cf736e47086
 SHA512:
-  metadata.gz: e0ccf08462f9577e3a84011d267088cdc06874713f3cdc6e47b58d1350441d76c5e61a963d1101050c72e78db32e29af17f0cda2f40012bfe5d6c2c5833a2fd2
-  data.tar.gz: e5251265a2b4587f732b7408b3ebfc7132621acbd3df7e0bb530fb4f3f6d4728a149d05dcf9c7d6aef47d2435543e88f2e3df26c5beb2a56b260b7491a14b845
+  metadata.gz: 416b64b2e91fadf29ed0b30814556dcaae74671ed74165586330518c9f4622fb40c13ed4c35cce7cb9b6d8089be9d8cb3f310b329e0682efa864e6380288d6e5
+  data.tar.gz: 45ae9a7abbf4d15e0c7f2cd5cbc9974446f5ee0da6d3718aa679e668a2f52b8164e7c1176ab1c21637896f5922afde189aed0b567fba147e2fecb8b887029e92

data/lib/driver.rb

@@ -0,0 +1,101 @@
+module River::Driver
+  # Provides a Sequel driver for River.
+  #
+  # Used in conjunction with a River client like:
+  #
+  #   DB = Sequel.connect("postgres://...")
+  #   client = River::Client.new(River::Driver::Sequel.new(DB))
+  #
+  class Sequel
+    def initialize(db)
+      @db = db
+      @db.extension(:pg_array)
+      @db.extension(:pg_json)
+    end
+
+    def job_get_by_id(id)
+      data_set = @db[:river_job].where(id: id)
+      data_set.first ? to_job_row(data_set.first) : nil
+    end
+
+    def job_insert(insert_params)
+      job_insert_many([insert_params]).first
+    end
+
+    def job_insert_many(insert_params_array)
+      @db[:river_job]
+        .insert_conflict(
+          target: [:unique_key],
+          conflict_where: ::Sequel.lit(
+            "unique_key IS NOT NULL AND unique_states IS NOT NULL AND river_job_state_in_bitmask(unique_states, state)"
+          ),
+          update: {kind: ::Sequel[:excluded][:kind]}
+        )
+        .returning(::Sequel.lit("*, (xmax != 0) AS unique_skipped_as_duplicate"))
+        .multi_insert(insert_params_array.map { |p| insert_params_to_hash(p) })
+        .map { |row| to_insert_result(row) }
+    end
+
+    def job_list
+      data_set = @db[:river_job].order_by(:id)
+      data_set.all.map { |job| to_job_row(job) }
+    end
+
+    def rollback_exception
+      ::Sequel::Rollback
+    end
+
+    def transaction(&)
+      @db.transaction(savepoint: true, &)
+    end
+
+    private def insert_params_to_hash(insert_params)
+      {
+        args: insert_params.encoded_args,
+        kind: insert_params.kind,
+        max_attempts: insert_params.max_attempts,
+        priority: insert_params.priority,
+        queue: insert_params.queue,
+        state: insert_params.state,
+        scheduled_at: insert_params.scheduled_at,
+        tags: ::Sequel.pg_array(insert_params.tags || [], :text),
+        unique_key: insert_params.unique_key ? ::Sequel.blob(insert_params.unique_key) : nil,
+        unique_states: insert_params.unique_states
+      }
+    end
+
+    private def to_insert_result(result)
+      [to_job_row(result), result[:unique_skipped_as_duplicate]]
+    end
+
+    private def to_job_row(river_job)
+      River::JobRow.new(
+        id: river_job[:id],
+        args: river_job[:args].to_h,
+        attempt: river_job[:attempt],
+        attempted_at: river_job[:attempted_at]&.getutc,
+        attempted_by: river_job[:attempted_by],
+        created_at: river_job[:created_at].getutc,
+        errors: river_job[:errors]&.map { |deserialized_error|
+          River::AttemptError.new(
+            at: Time.parse(deserialized_error["at"]),
+            attempt: deserialized_error["attempt"],
+            error: deserialized_error["error"],
+            trace: deserialized_error["trace"]
+          )
+        },
+        finalized_at: river_job[:finalized_at]&.getutc,
+        kind: river_job[:kind],
+        max_attempts: river_job[:max_attempts],
+        metadata: river_job[:metadata],
+        priority: river_job[:priority],
+        queue: river_job[:queue],
+        scheduled_at: river_job[:scheduled_at].getutc,
+        state: river_job[:state],
+        tags: river_job[:tags].to_a,
+        unique_key: river_job[:unique_key]&.to_s,
+        unique_states: ::River::UniqueBitmask.to_states(river_job[:unique_states]&.to_i(2))
+      )
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: riverqueue-sequel
 version: !ruby/object:Gem::Version
-  version: 0.6.1
+  version: 0.8.0
 platform: ruby
 authors:
 - Blake Gentry
@@ -9,7 +9,7 @@ authors:
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-08-21 00:00:00.000000000 Z
+date: 2024-12-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg
@@ -58,6 +58,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/driver.rb
 - lib/riverqueue-sequel.rb
 homepage: https://riverqueue.com
 licenses:
@@ -78,7 +79,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.20
+rubygems_version: 3.5.16
 signing_key:
 specification_version: 4
 summary: Sequel driver for the River Ruby gem.