chat_sdk-state-mysql 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c625807962a1731ba27361a51f2e78b47fcb9b1299769dc16644862d2133afde
-  data.tar.gz: ee821cebf2ec30e940c8d9e64495fe71f28323628eb58a562c25e17eb3a7cc92
+  metadata.gz: 4113d277f2fa7cb1f2c0ae346240c6c589bfaf4b03449be422200645f1db310d
+  data.tar.gz: 8c093057fa1677ef220cbacbbe96bdec2158447a886250fd273c0f64c86902da
 SHA512:
-  metadata.gz: 76500e001f1879bc588cf2a15dacac85e42e4dd94418202b7ececf1b457717afba1c0ad1c3fa92a57d180252ecbb6b763ed06fb1b69528fbcf191f8488bb6b05
-  data.tar.gz: 582293dd5c2675748260e3d2242fdfb88b620f7d51fcccc999d14fa30a0b20b8713a98c12ebc2e75524964cd4e4ecbfc85605cc5df7ca623d616ed0d50bad206
+  metadata.gz: 77310f954281b2193b02141847c2edc0d458825e7ef6f1863c16b63c1c225c34d88a2c7ec9e174ee5b7823159b8b6323bd504417bd1ff2ed9cfa781c7fdf7aac
+  data.tar.gz: a0bb0eacef3a7f175880fd1b108c0bcb84e3e5a4fac09e47e2dd2212301f0cc3a58e83c0d152fab24eaa4ebdac8cf207c7b126ad4660f07905a143804ad714c1

data/lib/chat_sdk/state/mysql.rb

@@ -74,6 +74,15 @@ module ChatSDK
         true
       end
 
+      def extend_lock(key, owner:, ttl:)
+        stmt = @conn.prepare(
+          "UPDATE chat_sdk_locks SET expires_at = DATE_ADD(NOW(), INTERVAL #{ttl.to_f} SECOND) " \
+          "WHERE key_prefix = ? AND lock_key = ? AND owner = ? AND expires_at > NOW()"
+        )
+        stmt.execute(@key_prefix, key, owner)
+        @conn.affected_rows > 0
+      end
+
       # Key-value store
 
       def get(key)
@@ -138,6 +147,52 @@ module ChatSDK
         @conn.affected_rows > 0
       end
 
+      def enqueue(key, value, max_size:, drop: :drop_oldest)
+        @conn.query("START TRANSACTION")
+        count_stmt = @conn.prepare(
+          "SELECT id FROM chat_sdk_queues WHERE key_prefix = ? AND queue_key = ? ORDER BY id FOR UPDATE"
+        )
+        ids = count_stmt.execute(@key_prefix, key).map { |row| row["id"] }
+        if ids.length >= max_size && drop.to_sym == :drop_newest
+          @conn.query("COMMIT")
+          return ids.length
+        end
+
+        insert = @conn.prepare("INSERT INTO chat_sdk_queues (key_prefix, queue_key, value) VALUES (?, ?, ?)")
+        insert.execute(@key_prefix, key, JSON.generate(value))
+        overflow = ids.length + 1 - max_size
+        if overflow.positive?
+          placeholders = Array.new(overflow, "?").join(", ")
+          remove = @conn.prepare("DELETE FROM chat_sdk_queues WHERE id IN (#{placeholders})")
+          remove.execute(*ids.first(overflow))
+        end
+        @conn.query("COMMIT")
+        [ids.length + 1, max_size].min
+      rescue
+        @conn.query("ROLLBACK")
+        raise
+      end
+
+      def drain_queue(key)
+        @conn.query("START TRANSACTION")
+        select = @conn.prepare(
+          "SELECT id, value FROM chat_sdk_queues WHERE key_prefix = ? AND queue_key = ? ORDER BY id FOR UPDATE"
+        )
+        rows = select.execute(@key_prefix, key).to_a
+        delete = @conn.prepare("DELETE FROM chat_sdk_queues WHERE key_prefix = ? AND queue_key = ?")
+        delete.execute(@key_prefix, key)
+        @conn.query("COMMIT")
+        rows.map { |row| JSON.parse(row["value"]) }
+      rescue
+        @conn.query("ROLLBACK")
+        raise
+      end
+
+      def queue_depth(key)
+        stmt = @conn.prepare("SELECT COUNT(*) AS count FROM chat_sdk_queues WHERE key_prefix = ? AND queue_key = ?")
+        stmt.execute(@key_prefix, key).first["count"].to_i
+      end
+
       # Cleanup
 
       def clear
@@ -147,6 +202,8 @@ module ChatSDK
         s2.execute(@key_prefix)
         s3 = @conn.prepare("DELETE FROM chat_sdk_cache WHERE key_prefix = ?")
         s3.execute(@key_prefix)
+        s4 = @conn.prepare("DELETE FROM chat_sdk_queues WHERE key_prefix = ?")
+        s4.execute(@key_prefix)
       end
 
       def cleanup_expired
@@ -193,6 +250,17 @@ module ChatSDK
             PRIMARY KEY (key_prefix, cache_key)
           ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4
         SQL
+        @conn.query(<<~SQL)
+          CREATE TABLE IF NOT EXISTS chat_sdk_queues (
+            id BIGINT NOT NULL AUTO_INCREMENT,
+            key_prefix VARCHAR(255) NOT NULL,
+            queue_key VARCHAR(512) NOT NULL,
+            value JSON NOT NULL,
+            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+            PRIMARY KEY (id),
+            INDEX chat_sdk_queues_lookup (key_prefix, queue_key, id)
+          ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4
+        SQL
       end
     end
   end

metadata

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