waldit 0.0.16 → 0.0.17

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a323599011304e72762f4fc8662221c438056c64d4bc0e1918f88749c4e4079b
-  data.tar.gz: d2495606705b6d2f85eb003a5a56c2be98895f168ec6c4fb5d656f6b4c8f2134
+  metadata.gz: 6c66dca1c2f51bdaa3f85e6f3f13930536e9352e6d5958f8b8eab140a108cab4
+  data.tar.gz: 43d8b34c506b98f5213372e624ab35d19f586570c47cf3abc5e692ec85ead638
 SHA512:
-  metadata.gz: '08cc2d9380a1a4f085657bcc49784b123f73dae3861483c7e19114d67790e7085ba60b7e267df20d438c611b8c2b4b311347d824091f7a74da76ba23f3e3b9de'
-  data.tar.gz: 32991fc4275d025c28fea6852f08c052d14f9c1295002252a2f007fdd48dc7341d8a0b2c72ea7af73aacf32b6af680dd110741c37cadc296528f3f9444f6a652
+  metadata.gz: 261e9f0b9bf6948c6fa6ac2f2bfd3dc2baf15541aa7ef18818b9aef76af08925427cf90a195bc32008c17bd674991687ac24cb1fe95b65d7cef08574c57aee08
+  data.tar.gz: 3a3dbf1982be4498d393f6ff10ce5be88835c34cdc0272be8d692db1ed5886ff66af1ea41f3e2f1dd79eab24a1ddf2d695c1a93ae74b38117d54de38e85695a0

data/lib/waldit/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Waldit
-  VERSION = "0.0.16"
+  VERSION = "0.0.17"
 end

data/lib/waldit/watcher.rb

@@ -35,72 +35,58 @@ module Waldit
       end
     end
 
+    def initialize(*)
+      super
+      initialize_connection
+      @retry = false
+    end
+
     def on_transaction_events(events)
-      record.transaction do
-        @connection = record.connection.raw_connection
+      @connection.transaction do
         tables = Set.new
-        insert_prepared = false
-        update_prepared = false
-        delete_prepared = false
 
         events.each do |event|
           case event
           when CommitTransactionEvent
-            record.where(transaction_id: event.transaction_id).update_all(committed_at: event.timestamp)
-
             changes = [:old, :new, :diff]
               .map { |diff| [diff, tables.filter { |table| Waldit.store_changes.call(table).include? diff }] }
               .to_h
 
-            log_new = (changes[:new] || []).map { |table| "'#{table}'" }.join(",")
-            log_old = (changes[:old] || []).map { |table| "'#{table}'" }.join(",")
-            log_diff = (changes[:diff] || []).map { |table| "'#{table}'" }.join(",")
-
-            record.where(transaction_id: event.transaction_id, action: "update").update_all(<<~SQL)
-              new = CASE WHEN table_name = ANY (ARRAY[#{log_new}]::varchar[]) THEN new ELSE null END,
-              old = CASE WHEN table_name = ANY (ARRAY[#{log_old}]::varchar[]) THEN old ELSE null END,
-              diff =
-                CASE WHEN table_name = ANY (ARRAY[#{log_diff}]::varchar[]) THEN (
-                  SELECT
-                    jsonb_object_agg(
-                      coalesce(old_kv.key, new_kv.key),
-                      jsonb_build_array(old_kv.value, new_kv.value)
-                    )
-                  FROM jsonb_each(old) AS old_kv
-                  FULL OUTER JOIN jsonb_each(new) AS new_kv ON old_kv.key = new_kv.key
-                  WHERE old_kv.value IS DISTINCT FROM new_kv.value
-                )
-                ELSE null
-                END
-            SQL
+            log_new = (changes[:new] || []).map { |table| "#{table}" }.join(",")
+            log_old = (changes[:old] || []).map { |table| "#{table}" }.join(",")
+            log_diff = (changes[:diff] || []).map { |table| "#{table}" }.join(",")
+
+            @connection.exec_prepared("waldit_finish", [
+              event.transaction_id,
+              event.timestamp,
+              "{#{log_new}}",
+              "{#{log_old}}",
+              "{#{log_diff}}",
+            ])
+
+            # We sucessful retried a connection, let's reset our retry state
+            @retry = false
 
           when InsertEvent
             tables << event.table
-            unless insert_prepared
-              prepare_insert
-              insert_prepared = true
-            end
             audit_event(event)
 
           when UpdateEvent
             tables << event.table
-            unless update_prepared
-              prepare_update
-              update_prepared = true
-            end
             audit_event(event)
 
           when DeleteEvent
             tables << event.table
-            unless delete_prepared
-              prepare_delete
-              prepare_delete_cleanup
-              delete_prepared = true
-            end
             audit_event(event)
           end
         end
       end
+    rescue PG::ConnectionBad
+      raise if @retry
+      # Let's try to fetch a new connection and reprocess the transaction
+      initialize_connection
+      @retry = true
+      retry
     end
 
     def should_watch_table?(table)
@@ -125,6 +111,15 @@ module Waldit
 
     private
 
+    def initialize_connection
+      @connection = record.connection.raw_connection
+      prepare_insert
+      prepare_update
+      prepare_delete
+      prepare_delete_cleanup
+      prepare_finish
+    end
+
     def prepare_insert
       @connection.prepare("waldit_insert", <<~SQL)
         INSERT INTO #{record.table_name} (transaction_id, lsn, table_name, primary_key, action, context, new)
@@ -132,6 +127,7 @@ module Waldit
         ON CONFLICT (table_name, primary_key, transaction_id)
         DO UPDATE SET new = #{record.table_name}.new
       SQL
+    rescue PG::DuplicatePstatement
     end
 
     def prepare_update
@@ -141,6 +137,7 @@ module Waldit
         ON CONFLICT (table_name, primary_key, transaction_id)
         DO UPDATE SET new = excluded.new
       SQL
+    rescue PG::DuplicatePstatement
     end
 
     def prepare_delete
@@ -150,6 +147,7 @@ module Waldit
         ON CONFLICT (table_name, primary_key, transaction_id)
         DO UPDATE SET old = #{record.table_name}.old
       SQL
+    rescue PG::DuplicatePstatement
     end
 
     def prepare_delete_cleanup
@@ -162,6 +160,32 @@ module Waldit
           AND action IN ('insert'::waldit_action, 'update'::waldit_action)
         RETURNING action, old
       SQL
+    rescue PG::DuplicatePstatement
+    end
+
+    def prepare_finish
+      @connection.prepare("waldit_finish", <<~SQL)
+        UPDATE #{record.table_name}
+        SET
+          committed_at = $2,
+          new = CASE WHEN table_name = ANY ($3::varchar[]) THEN new ELSE null END,
+          old = CASE WHEN table_name = ANY ($4::varchar[]) THEN old ELSE null END,
+          diff =
+            CASE WHEN table_name = ANY ($5::varchar[]) THEN (
+              SELECT
+                jsonb_object_agg(
+                  coalesce(old_kv.key, new_kv.key),
+                  jsonb_build_array(old_kv.value, new_kv.value)
+                )
+              FROM jsonb_each(old) AS old_kv
+              FULL OUTER JOIN jsonb_each(new) AS new_kv ON old_kv.key = new_kv.key
+              WHERE old_kv.value IS DISTINCT FROM new_kv.value
+            )
+            ELSE null
+            END
+        WHERE transaction_id = $1
+      SQL
+    rescue PG::DuplicatePstatement
     end
   end
 end

data/rbi/waldit.rbi

@@ -2,7 +2,7 @@
 module Waldit
   extend T::Sig
   extend Waldit::Context
-  VERSION = "0.0.16"
+  VERSION = "0.0.17"
 
   class << self
     sig { returns(String) }

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: waldit
 version: !ruby/object:Gem::Version
-  version: 0.0.16
+  version: 0.0.17
 platform: ruby
 authors:
 - Rodrigo Navarro
 bindir: exe
 cert_chain: []
-date: 2025-09-05 00:00:00.000000000 Z
+date: 2025-09-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: wal