chat_sdk-state-pg 0.2.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: b581b9863bfefec84597d4c07cea4379928f1385aec9a3ea87142dfe41df8e1e
+  data.tar.gz: 5aa8b46ba8515bc20acaba7d4985534f040e839f04a6d40b12dc6348df26927f
+SHA512:
+  metadata.gz: 524b58cf98323b21b565e33627f84ace837f93a1fc640353069e82b21110270d98f35ca1ee7c35176df910af67110282c49aad4e33801a24be1a1f5f871ad99c
+  data.tar.gz: f3db506a1c1ed0cdb1b2882ee1f80f7af18dd3001b4ec8c4e7e777cbb31d326a7f5672473bd35f8d621b41a3f557396ab123670beb7b7a60e3469df7825d454f

data/lib/chat_sdk/state/pg.rb

@@ -0,0 +1,176 @@
+# frozen_string_literal: true
+
+require "chat_sdk"
+require "pg"
+require "json"
+
+module ChatSDK
+  module State
+    class Pg < Base
+      def initialize(url: nil, connection: nil, key_prefix: "chat_sdk", auto_migrate: true)
+        @conn = connection || PG.connect(url || ENV["DATABASE_URL"] || ENV["POSTGRES_URL"] || "postgresql://localhost/chat_sdk_dev")
+        @key_prefix = key_prefix
+        ensure_tables if auto_migrate
+      end
+
+      attr_reader :conn
+
+      # Subscriptions
+
+      def subscribe(thread_id)
+        @conn.exec_params(
+          "INSERT INTO chat_sdk_subscriptions (key_prefix, thread_id) VALUES ($1, $2) ON CONFLICT DO NOTHING",
+          [@key_prefix, thread_id]
+        )
+      end
+
+      def unsubscribe(thread_id)
+        @conn.exec_params(
+          "DELETE FROM chat_sdk_subscriptions WHERE key_prefix = $1 AND thread_id = $2",
+          [@key_prefix, thread_id]
+        )
+      end
+
+      def subscribed?(thread_id)
+        result = @conn.exec_params(
+          "SELECT 1 FROM chat_sdk_subscriptions WHERE key_prefix = $1 AND thread_id = $2",
+          [@key_prefix, thread_id]
+        )
+        result.ntuples > 0
+      end
+
+      # Locks
+
+      def acquire_lock(key, owner:, ttl:)
+        ttl_seconds = ttl.to_f
+        @conn.exec_params(
+          "DELETE FROM chat_sdk_locks WHERE key_prefix = $1 AND lock_key = $2 AND expires_at < NOW()",
+          [@key_prefix, key]
+        )
+        result = @conn.exec_params(
+          "INSERT INTO chat_sdk_locks (key_prefix, lock_key, owner, expires_at) " \
+          "VALUES ($1, $2, $3, NOW() + INTERVAL '#{ttl_seconds} seconds') ON CONFLICT DO NOTHING",
+          [@key_prefix, key, owner]
+        )
+        result.cmd_tuples > 0
+      end
+
+      def release_lock(key, owner:)
+        result = @conn.exec_params(
+          "DELETE FROM chat_sdk_locks WHERE key_prefix = $1 AND lock_key = $2 AND owner = $3",
+          [@key_prefix, key, owner]
+        )
+        result.cmd_tuples > 0
+      end
+
+      def force_lock(key, owner:, ttl:)
+        ttl_seconds = ttl.to_f
+        @conn.exec_params(
+          "INSERT INTO chat_sdk_locks (key_prefix, lock_key, owner, expires_at) " \
+          "VALUES ($1, $2, $3, NOW() + INTERVAL '#{ttl_seconds} seconds') " \
+          "ON CONFLICT (key_prefix, lock_key) DO UPDATE SET owner = EXCLUDED.owner, expires_at = EXCLUDED.expires_at",
+          [@key_prefix, key, owner]
+        )
+        true
+      end
+
+      # Key-value store
+
+      def get(key)
+        result = @conn.exec_params(
+          "SELECT value FROM chat_sdk_cache WHERE key_prefix = $1 AND cache_key = $2 AND (expires_at IS NULL OR expires_at > NOW())",
+          [@key_prefix, key]
+        )
+        return JSON.parse(result[0]["value"]) if result.ntuples > 0
+
+        nil
+      end
+
+      def set(key, value, ttl: nil)
+        serialized = JSON.generate(value)
+        expires = ttl ? "NOW() + INTERVAL '#{ttl.to_f} seconds'" : "NULL"
+        @conn.exec_params(
+          "INSERT INTO chat_sdk_cache (key_prefix, cache_key, value, expires_at) " \
+          "VALUES ($1, $2, $3::jsonb, #{expires}) " \
+          "ON CONFLICT (key_prefix, cache_key) DO UPDATE SET value = EXCLUDED.value, expires_at = EXCLUDED.expires_at",
+          [@key_prefix, key, serialized]
+        )
+        value
+      end
+
+      def delete(key)
+        @conn.exec_params(
+          "DELETE FROM chat_sdk_cache WHERE key_prefix = $1 AND cache_key = $2",
+          [@key_prefix, key]
+        )
+        @conn.exec_params(
+          "DELETE FROM chat_sdk_locks WHERE key_prefix = $1 AND lock_key = $2",
+          [@key_prefix, key]
+        )
+      end
+
+      def set_if_absent(key, value, ttl: nil)
+        serialized = JSON.generate(value)
+        @conn.exec_params(
+          "DELETE FROM chat_sdk_cache WHERE key_prefix = $1 AND cache_key = $2 AND expires_at IS NOT NULL AND expires_at < NOW()",
+          [@key_prefix, key]
+        )
+        expires = ttl ? "NOW() + INTERVAL '#{ttl.to_f} seconds'" : "NULL"
+        result = @conn.exec_params(
+          "INSERT INTO chat_sdk_cache (key_prefix, cache_key, value, expires_at) " \
+          "VALUES ($1, $2, $3::jsonb, #{expires}) ON CONFLICT DO NOTHING",
+          [@key_prefix, key, serialized]
+        )
+        result.cmd_tuples > 0
+      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])
+      end
+
+      def cleanup_expired
+        @conn.exec_params(
+          "DELETE FROM chat_sdk_cache WHERE key_prefix = $1 AND expires_at IS NOT NULL AND expires_at < NOW()",
+          [@key_prefix]
+        )
+        @conn.exec_params(
+          "DELETE FROM chat_sdk_locks WHERE key_prefix = $1 AND expires_at < NOW()",
+          [@key_prefix]
+        )
+      end
+
+      private
+
+      def ensure_tables
+        @conn.exec(<<~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 NOW(),
+            PRIMARY KEY (key_prefix, thread_id)
+          );
+
+          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)
+          );
+
+          CREATE TABLE IF NOT EXISTS chat_sdk_cache (
+            key_prefix VARCHAR(255) NOT NULL,
+            cache_key VARCHAR(512) NOT NULL,
+            value JSONB NOT NULL,
+            expires_at TIMESTAMP,
+            PRIMARY KEY (key_prefix, cache_key)
+          );
+        SQL
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,71 @@
+--- !ruby/object:Gem::Specification
+name: chat_sdk-state-pg
+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: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+description: PostgreSQL-backed conversation state storage for the ChatSDK framework
+  with auto-migration and JSONB storage
+email:
+- quentin@rootly.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/chat_sdk/state/pg.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: PostgreSQL state backend for ChatSDK
+test_files: []