waldit 0.0.17 → 0.0.20

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6c66dca1c2f51bdaa3f85e6f3f13930536e9352e6d5958f8b8eab140a108cab4
-  data.tar.gz: 43d8b34c506b98f5213372e624ab35d19f586570c47cf3abc5e692ec85ead638
+  metadata.gz: e89f236a107159502d3d667c1073d39152cee8ba5923f3ef2a7f008e6ae78814
+  data.tar.gz: 1bc36fdd5dd1580f5ebbf3bc8cbe301cc6a9be74d467c561998289fe6d38e84e
 SHA512:
-  metadata.gz: 261e9f0b9bf6948c6fa6ac2f2bfd3dc2baf15541aa7ef18818b9aef76af08925427cf90a195bc32008c17bd674991687ac24cb1fe95b65d7cef08574c57aee08
-  data.tar.gz: 3a3dbf1982be4498d393f6ff10ce5be88835c34cdc0272be8d692db1ed5886ff66af1ea41f3e2f1dd79eab24a1ddf2d695c1a93ae74b38117d54de38e85695a0
+  metadata.gz: ebf0fe008ddcc05e90ec6d6931a22153a45c58d7f73f94b01af3d7a33ba23406dd99a38b265d932b764ce0db38ca36f7c2df37a74c194d97b3b1a79df78a8e21
+  data.tar.gz: 9b4abee181b3c3a04399c7e2614018c9bc9bc65026cd6c28e89e3b74a7c47a4e3262cfbea882d59cf725a26f98c6f13d6dc2419c61ab6b5c7d871e2c6c62b361

data/lib/waldit/version.rb

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

data/lib/waldit/watcher.rb

@@ -52,16 +52,22 @@ module Waldit
               .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(",")
+            log_new = (changes[:new] || []).map { |table| "#{table}" }
+            log_old = (changes[:old] || []).map { |table| "#{table}" }
+            log_diff = (changes[:diff] || []).map { |table| "#{table}" }
 
             @connection.exec_prepared("waldit_finish", [
               event.transaction_id,
               event.timestamp,
-              "{#{log_new}}",
-              "{#{log_old}}",
-              "{#{log_diff}}",
+              "{#{log_new.join(",")}}",
+              "{#{log_old.join(",")}}",
+              "{#{log_diff.join(",")}}",
+            ])
+
+            @connection.exec_prepared("waldit_cleanup", [
+              event.transaction_id,
+              "{#{(log_new + log_old).join(",")}}",
+              "{#{log_diff.join(",")}}",
             ])
 
             # We sucessful retried a connection, let's reset our retry state
@@ -112,12 +118,13 @@ module Waldit
     private
 
     def initialize_connection
-      @connection = record.connection.raw_connection
+      @connection = record.connection_pool.checkout.raw_connection
       prepare_insert
       prepare_update
       prepare_delete
       prepare_delete_cleanup
       prepare_finish
+      prepare_cleanup
     end
 
     def prepare_insert
@@ -168,10 +175,10 @@ module Waldit
         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,
+          new = CASE WHEN action = 'insert' OR table_name = ANY ($3::varchar[]) THEN new ELSE null END,
+          old = CASE WHEN action = 'delete' OR table_name = ANY ($4::varchar[]) THEN old ELSE null END,
           diff =
-            CASE WHEN table_name = ANY ($5::varchar[]) THEN (
+            CASE WHEN action = 'update' AND table_name = ANY ($5::varchar[]) THEN (
               SELECT
                 jsonb_object_agg(
                   coalesce(old_kv.key, new_kv.key),
@@ -187,5 +194,20 @@ module Waldit
       SQL
     rescue PG::DuplicatePstatement
     end
+
+    def prepare_cleanup
+      @connection.prepare("waldit_cleanup", <<~SQL)
+        DELETE FROM #{record.table_name}
+        WHERE
+          transaction_id = $1
+          AND action = 'update'::waldit_action
+          AND (
+            (diff IS NULL AND table_name = ANY ($3::varchar[]))
+            OR
+            (new = old AND table_name = ANY ($2::varchar[]))
+          )
+      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.17"
+  VERSION = "0.0.19"
 
   class << self
     sig { returns(String) }

metadata

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