chat_sdk-state-pg 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5663959fe7eb030f9e94cd6ef2bbfe92d6762e14facbe66fa128f832e372b715
-  data.tar.gz: 5aa8b46ba8515bc20acaba7d4985534f040e839f04a6d40b12dc6348df26927f
+  metadata.gz: dcbe0e018a443c2717ee1e8a5f02941854d14a67cfc5079e4547f50c85392dd7
+  data.tar.gz: 1a8a8d102f3a188b28dfeeb4e9474bbdf3e34367585a5c2ba6f817d67c50279a
 SHA512:
-  metadata.gz: 11027cec1805e2d7e4f07532842ab2bd234d612e5caf47b1fe7d096f6d8fb28b1ff525a2baca1fd7f1286cd3563fd3b8060e2c6714238a72879231625050f837
-  data.tar.gz: f3db506a1c1ed0cdb1b2882ee1f80f7af18dd3001b4ec8c4e7e777cbb31d326a7f5672473bd35f8d621b41a3f557396ab123670beb7b7a60e3469df7825d454f
+  metadata.gz: ec9ab286bd28012c6a1572c8df4915c160a44b4f08c4cbaa557e59fe4d08b3883f67a4895ec100ad4f58d77ffc40ace6a1e694172ac9d8e8448ea9beb3de635c
+  data.tar.gz: 202eb940a1d559384448f75e560c373f37908522c06c32c9e4e18e8b4abf8aec1ad4498c9ef01ef30484348ed960470a53eaace1bb9a0d041c08e00b51a99232

data/lib/chat_sdk/state/pg.rb

@@ -74,6 +74,15 @@ module ChatSDK
         true
       end
 
+      def extend_lock(key, owner:, ttl:)
+        result = @conn.exec_params(
+          "UPDATE chat_sdk_locks SET expires_at = NOW() + ($4 * INTERVAL '1 second') " \
+          "WHERE key_prefix = $1 AND lock_key = $2 AND owner = $3 AND expires_at > NOW()",
+          [@key_prefix, key, owner, ttl.to_f]
+        )
+        result.cmd_tuples > 0
+      end
+
       # Key-value store
 
       def get(key)
@@ -124,12 +133,56 @@ module ChatSDK
         result.cmd_tuples > 0
       end
 
+      def enqueue(key, value, max_size:, drop: :drop_oldest)
+        @conn.transaction do |conn|
+          conn.exec_params("SELECT pg_advisory_xact_lock(hashtext($1))", ["#{@key_prefix}:#{key}"])
+          depth = conn.exec_params(
+            "SELECT COUNT(*) AS count FROM chat_sdk_queues WHERE key_prefix = $1 AND queue_key = $2",
+            [@key_prefix, key]
+          )[0]["count"].to_i
+          return depth if depth >= max_size && drop.to_sym == :drop_newest
+
+          conn.exec_params(
+            "INSERT INTO chat_sdk_queues (key_prefix, queue_key, value) VALUES ($1, $2, $3::jsonb)",
+            [@key_prefix, key, JSON.generate(value)]
+          )
+          conn.exec_params(
+            "DELETE FROM chat_sdk_queues WHERE id IN (" \
+            "SELECT id FROM chat_sdk_queues WHERE key_prefix = $1 AND queue_key = $2 " \
+            "ORDER BY id ASC LIMIT (SELECT GREATEST(COUNT(*) - $3, 0) FROM chat_sdk_queues " \
+            "WHERE key_prefix = $1 AND queue_key = $2))",
+            [@key_prefix, key, max_size]
+          )
+          [depth + 1, max_size].min
+        end
+      end
+
+      def drain_queue(key)
+        @conn.transaction do |conn|
+          rows = conn.exec_params(
+            "DELETE FROM chat_sdk_queues WHERE id IN (" \
+            "SELECT id FROM chat_sdk_queues WHERE key_prefix = $1 AND queue_key = $2 ORDER BY id ASC FOR UPDATE) " \
+            "RETURNING value, id",
+            [@key_prefix, key]
+          ).to_a.sort_by { |row| row["id"].to_i }
+          rows.map { |row| JSON.parse(row["value"].is_a?(String) ? row["value"] : JSON.generate(row["value"])) }
+        end
+      end
+
+      def queue_depth(key)
+        @conn.exec_params(
+          "SELECT COUNT(*) AS count FROM chat_sdk_queues WHERE key_prefix = $1 AND queue_key = $2",
+          [@key_prefix, key]
+        )[0]["count"].to_i
+      end
+
       # Cleanup
 
       def clear
         @conn.exec_params("DELETE FROM chat_sdk_subscriptions WHERE key_prefix = $1", [@key_prefix])
         @conn.exec_params("DELETE FROM chat_sdk_locks WHERE key_prefix = $1", [@key_prefix])
         @conn.exec_params("DELETE FROM chat_sdk_cache WHERE key_prefix = $1", [@key_prefix])
+        @conn.exec_params("DELETE FROM chat_sdk_queues WHERE key_prefix = $1", [@key_prefix])
       end
 
       def cleanup_expired
@@ -169,6 +222,17 @@ module ChatSDK
             expires_at TIMESTAMP,
             PRIMARY KEY (key_prefix, cache_key)
           );
+
+          CREATE TABLE IF NOT EXISTS chat_sdk_queues (
+            id BIGSERIAL PRIMARY KEY,
+            key_prefix VARCHAR(255) NOT NULL,
+            queue_key VARCHAR(512) NOT NULL,
+            value JSONB NOT NULL,
+            created_at TIMESTAMP DEFAULT NOW()
+          );
+
+          CREATE INDEX IF NOT EXISTS chat_sdk_queues_lookup
+            ON chat_sdk_queues (key_prefix, queue_key, id);
         SQL
       end
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: chat_sdk-state-pg
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Quentin Rousseau