chat_sdk-state-mysql 0.2.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 6400f4beca077fbd4362accc33b1d2207721972cbc14aaf4dae2eb9f2e22883e
+  data.tar.gz: ee821cebf2ec30e940c8d9e64495fe71f28323628eb58a562c25e17eb3a7cc92
+SHA512:
+  metadata.gz: 586a519179e2bb027ef69fc1fb19284f8c5001d9676e6c48dafb9482fc0a98e79d1a45e8deae449a0cdefeba38b0ff671a6c293e0f3da8942f77c525be04c342
+  data.tar.gz: 582293dd5c2675748260e3d2242fdfb88b620f7d51fcccc999d14fa30a0b20b8713a98c12ebc2e75524964cd4e4ecbfc85605cc5df7ca623d616ed0d50bad206

data/lib/chat_sdk/state/mysql.rb

@@ -0,0 +1,199 @@
+# frozen_string_literal: true
+
+require "chat_sdk"
+require "mysql2"
+require "json"
+
+module ChatSDK
+  module State
+    class Mysql < Base
+      def initialize(url: nil, connection: nil, key_prefix: "chat_sdk", auto_migrate: true)
+        @conn = connection || connect(url || ENV["MYSQL_URL"] || ENV["DATABASE_URL"] || "mysql2://root@localhost/chat_sdk_dev")
+        @key_prefix = key_prefix
+        ensure_tables if auto_migrate
+      end
+
+      attr_reader :conn
+
+      # Subscriptions
+
+      def subscribe(thread_id)
+        stmt = @conn.prepare(
+          "INSERT IGNORE INTO chat_sdk_subscriptions (key_prefix, thread_id) VALUES (?, ?)"
+        )
+        stmt.execute(@key_prefix, thread_id)
+      end
+
+      def unsubscribe(thread_id)
+        stmt = @conn.prepare(
+          "DELETE FROM chat_sdk_subscriptions WHERE key_prefix = ? AND thread_id = ?"
+        )
+        stmt.execute(@key_prefix, thread_id)
+      end
+
+      def subscribed?(thread_id)
+        stmt = @conn.prepare(
+          "SELECT 1 FROM chat_sdk_subscriptions WHERE key_prefix = ? AND thread_id = ?"
+        )
+        result = stmt.execute(@key_prefix, thread_id)
+        result.count > 0
+      end
+
+      # Locks
+
+      def acquire_lock(key, owner:, ttl:)
+        ttl_seconds = ttl.to_f
+        del_stmt = @conn.prepare(
+          "DELETE FROM chat_sdk_locks WHERE key_prefix = ? AND lock_key = ? AND expires_at < NOW()"
+        )
+        del_stmt.execute(@key_prefix, key)
+        ins_stmt = @conn.prepare(
+          "INSERT IGNORE INTO chat_sdk_locks (key_prefix, lock_key, owner, expires_at) " \
+          "VALUES (?, ?, ?, DATE_ADD(NOW(), INTERVAL #{ttl_seconds} SECOND))"
+        )
+        ins_stmt.execute(@key_prefix, key, owner)
+        @conn.affected_rows > 0
+      end
+
+      def release_lock(key, owner:)
+        stmt = @conn.prepare(
+          "DELETE FROM chat_sdk_locks WHERE key_prefix = ? AND lock_key = ? AND owner = ?"
+        )
+        stmt.execute(@key_prefix, key, owner)
+        @conn.affected_rows > 0
+      end
+
+      def force_lock(key, owner:, ttl:)
+        ttl_seconds = ttl.to_f
+        stmt = @conn.prepare(
+          "INSERT INTO chat_sdk_locks (key_prefix, lock_key, owner, expires_at) " \
+          "VALUES (?, ?, ?, DATE_ADD(NOW(), INTERVAL #{ttl_seconds} SECOND)) " \
+          "ON DUPLICATE KEY UPDATE owner = VALUES(owner), expires_at = VALUES(expires_at)"
+        )
+        stmt.execute(@key_prefix, key, owner)
+        true
+      end
+
+      # Key-value store
+
+      def get(key)
+        stmt = @conn.prepare(
+          "SELECT value FROM chat_sdk_cache WHERE key_prefix = ? AND cache_key = ? AND (expires_at IS NULL OR expires_at > NOW())"
+        )
+        result = stmt.execute(@key_prefix, key)
+        row = result.first
+        return JSON.parse(row["value"]) if row
+
+        nil
+      end
+
+      def set(key, value, ttl: nil)
+        serialized = JSON.generate(value)
+        stmt = if ttl
+          @conn.prepare(
+            "INSERT INTO chat_sdk_cache (key_prefix, cache_key, value, expires_at) " \
+            "VALUES (?, ?, ?, DATE_ADD(NOW(), INTERVAL #{ttl.to_f} SECOND)) " \
+            "ON DUPLICATE KEY UPDATE value = VALUES(value), expires_at = VALUES(expires_at)"
+          )
+        else
+          @conn.prepare(
+            "INSERT INTO chat_sdk_cache (key_prefix, cache_key, value, expires_at) " \
+            "VALUES (?, ?, ?, NULL) " \
+            "ON DUPLICATE KEY UPDATE value = VALUES(value), expires_at = VALUES(expires_at)"
+          )
+        end
+        stmt.execute(@key_prefix, key, serialized)
+        value
+      end
+
+      def delete(key)
+        stmt1 = @conn.prepare(
+          "DELETE FROM chat_sdk_cache WHERE key_prefix = ? AND cache_key = ?"
+        )
+        stmt1.execute(@key_prefix, key)
+        stmt2 = @conn.prepare(
+          "DELETE FROM chat_sdk_locks WHERE key_prefix = ? AND lock_key = ?"
+        )
+        stmt2.execute(@key_prefix, key)
+      end
+
+      def set_if_absent(key, value, ttl: nil)
+        serialized = JSON.generate(value)
+        del_stmt = @conn.prepare(
+          "DELETE FROM chat_sdk_cache WHERE key_prefix = ? AND cache_key = ? AND expires_at IS NOT NULL AND expires_at < NOW()"
+        )
+        del_stmt.execute(@key_prefix, key)
+        stmt = if ttl
+          @conn.prepare(
+            "INSERT IGNORE INTO chat_sdk_cache (key_prefix, cache_key, value, expires_at) " \
+            "VALUES (?, ?, ?, DATE_ADD(NOW(), INTERVAL #{ttl.to_f} SECOND))"
+          )
+        else
+          @conn.prepare(
+            "INSERT IGNORE INTO chat_sdk_cache (key_prefix, cache_key, value, expires_at) " \
+            "VALUES (?, ?, ?, NULL)"
+          )
+        end
+        stmt.execute(@key_prefix, key, serialized)
+        @conn.affected_rows > 0
+      end
+
+      # Cleanup
+
+      def clear
+        s1 = @conn.prepare("DELETE FROM chat_sdk_subscriptions WHERE key_prefix = ?")
+        s1.execute(@key_prefix)
+        s2 = @conn.prepare("DELETE FROM chat_sdk_locks WHERE key_prefix = ?")
+        s2.execute(@key_prefix)
+        s3 = @conn.prepare("DELETE FROM chat_sdk_cache WHERE key_prefix = ?")
+        s3.execute(@key_prefix)
+      end
+
+      def cleanup_expired
+        s1 = @conn.prepare(
+          "DELETE FROM chat_sdk_cache WHERE key_prefix = ? AND expires_at IS NOT NULL AND expires_at < NOW()"
+        )
+        s1.execute(@key_prefix)
+        s2 = @conn.prepare(
+          "DELETE FROM chat_sdk_locks WHERE key_prefix = ? AND expires_at < NOW()"
+        )
+        s2.execute(@key_prefix)
+      end
+
+      private
+
+      def connect(url)
+        Mysql2::Client.new(url)
+      end
+
+      def ensure_tables
+        @conn.query(<<~SQL)
+          CREATE TABLE IF NOT EXISTS chat_sdk_subscriptions (
+            key_prefix VARCHAR(255) NOT NULL,
+            thread_id VARCHAR(512) NOT NULL,
+            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+            PRIMARY KEY (key_prefix, thread_id)
+          ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4
+        SQL
+        @conn.query(<<~SQL)
+          CREATE TABLE IF NOT EXISTS chat_sdk_locks (
+            key_prefix VARCHAR(255) NOT NULL,
+            lock_key VARCHAR(512) NOT NULL,
+            owner VARCHAR(255) NOT NULL,
+            expires_at TIMESTAMP NOT NULL,
+            PRIMARY KEY (key_prefix, lock_key)
+          ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4
+        SQL
+        @conn.query(<<~SQL)
+          CREATE TABLE IF NOT EXISTS chat_sdk_cache (
+            key_prefix VARCHAR(255) NOT NULL,
+            cache_key VARCHAR(512) NOT NULL,
+            value JSON NOT NULL,
+            expires_at TIMESTAMP NULL DEFAULT NULL,
+            PRIMARY KEY (key_prefix, cache_key)
+          ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4
+        SQL
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,71 @@
+--- !ruby/object:Gem::Specification
+name: chat_sdk-state-mysql
+version: !ruby/object:Gem::Version
+  version: 0.2.1
+platform: ruby
+authors:
+- Quentin Rousseau
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: chat_sdk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+- !ruby/object:Gem::Dependency
+  name: mysql2
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.5'
+description: MySQL-backed conversation state storage for the ChatSDK framework with
+  auto-migration and JSON storage
+email:
+- quentin@rootly.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/chat_sdk/state/mysql.rb
+homepage: https://github.com/rootlyhq/chat-sdk
+licenses:
+- MIT
+metadata:
+  rubygems_mfa_required: 'true'
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.2.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 4.0.16
+specification_version: 4
+summary: MySQL state backend for ChatSDK
+test_files: []