waldit 0.0.6 → 0.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e8e8ecb8b8fc2c3c4b855fb06b204d858b7cfbf24f12dc016b4c3f5ca10849a2
-  data.tar.gz: b72218c4df4b396e497f1aa9d96fd159305fabb7faf0d481392044536da5141a
+  metadata.gz: 8dd9fb06d63894b50126cf5b577ef2c0ab62f58ffc095632ad26ee18f8f9d86e
+  data.tar.gz: 4d99542390d1a30ba64055515bcf9e30d5c83351fd402fe0e933f96a9a37425e
 SHA512:
-  metadata.gz: 5471e11fa41cb7c3576dba2e0c1095ee3b2a7b7fd4eda88cd46e10523cf54ecdcda5a2d3470fdb9e97f3aebc066ce84bba7aa63be8507a0f0e37927b6604520f
-  data.tar.gz: 11dd782e95115cb77e8df0bed87f4dd8422454f5ef6ce84c35b9fe39af931c9855b37b07685cb3833ace964757c7d81b3d64a55a23330b52177ada7b60dd7ea8
+  metadata.gz: 28290886fa42d5ef259186281f73dd3f6539f561adef547c458215127fd40253d374ffc67a3e48db9ec32e47317676cc173cea9a52bd1d569e715c7831d3b84d
+  data.tar.gz: ce9ef4fc3d0df78721537c55226f57049f28dedaa7b7d1f072b74c139a0fdb7c2ab2918c500a6666645517d2be0f9a27c9f4a86331b4900ad0a3509fd0050344

data/lib/waldit/migration.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+module Waldit
+  module Migration
+    def create_waldit_table(name = "waldit", &block)
+      reversible do |dir|
+        dir.up { execute "CREATE TYPE waldit_action AS ENUM ('insert', 'update', 'delete')" }
+        dir.down { execute "DROP TYPE waldit_action" }
+      end
+
+      create_table :waldit do |t|
+        t.column :action, :waldit_action, null: false
+        t.string :table_name, null: false
+        t.string :primary_key
+        t.bigint :transaction_id, null: false
+        t.decimal :lsn, null: false, precision: 20, scale: 0
+        t.timestamptz :commited_at
+        t.jsonb :context, null: false, default: {}
+        t.jsonb :old, null: true
+        t.jsonb :new, null: true
+        t.jsonb :diff, null: true
+        block.call(t)
+      end
+
+      add_index :waldit, [:table_name, :primary_key, :transaction_id], unique: true
+      add_index :waldit, [:transaction_id, :lsn]
+      add_index :waldit, :commited_at
+      add_index :waldit, :context, using: :gin, opclass: :jsonb_path_ops
+    end
+
+    def create_waldit_publication
+      reversible do |dir|
+        dir.up { execute "CREATE PUBLICATION waldit" }
+        dir.down { execute "DROP PUBLICATION waldit" }
+      end
+    end
+
+    def add_table_to_waldit(table)
+      reversible do |dir|
+        dir.up do
+          execute "ALTER TABLE #{table} REPLICA IDENTITY FULL"
+          execute "ALTER PUBLICATION waldit ADD TABLE #{table}"
+        end
+        dir.down do
+          execute "ALTER PUBLICATION waldit DROP TABLE #{table}"
+          execute "ALTER TABLE #{table} REPLICA IDENTITY DEFAULT"
+        end
+      end
+    end
+
+    def remove_table_from_waldit(table)
+      reversible do |dir|
+        dir.up do
+          execute "ALTER PUBLICATION waldit DROP TABLE #{table}"
+          execute "ALTER TABLE #{table} REPLICA IDENTITY DEFAULT"
+        end
+        dir.down do
+          execute "ALTER TABLE #{table} REPLICA IDENTITY FULL"
+          execute "ALTER PUBLICATION waldit ADD TABLE #{table}"
+        end
+      end
+    end
+  end
+end

data/lib/waldit/railtie.rb

@@ -5,11 +5,16 @@ require "rails/railtie"
 module Waldit
   class Railtie < Rails::Railtie
     config.before_configuration do
-      ActiveRecord::ConnectionAdapters.register(
-        "postgresqlwaldit",
-        "Waldit::PostgreSQLAdapter",
-        "waldit/postgresql_adapter",
-      )
+      ["waldit", "postgresqlwaldit"].each do |adapter_name|
+        ActiveRecord::ConnectionAdapters.register(
+          adapter_name,
+          "Waldit::PostgreSQLAdapter",
+          "waldit/postgresql_adapter",
+        )
+      end
+
+      require_relative "migration"
+      ActiveRecord::Migration.include Waldit::Migration
     end
   end
 end

data/lib/waldit/sidekiq.rb

@@ -17,8 +17,13 @@ module Waldit
       include ::Sidekiq::ServerMiddleware
 
       def call(job_instance, job, queue, &block)
-        context = deserialize_context(job) || {}
-        Waldit.with_context(context.merge(background_job: job_instance.class.to_s), &block)
+        background_job = case job["class"]
+        in "ActiveJob::QueueAdapters::SidekiqAdapter::JobWrapper"
+          job["args"][0]["job_class"]
+        in klass
+          klass
+        end
+        Waldit.with_context((deserialize_context(job) || {}).merge(background_job:), &block)
       end
 
       private

data/lib/waldit/version.rb

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

data/lib/waldit/watcher.rb

@@ -44,14 +44,19 @@ module Waldit
           when CommitTransactionEvent
             record.where(transaction_id: event.transaction_id).update_all(commited_at: event.timestamp)
 
-            log_new = tables.filter { |table| Waldit.store_changes.call(table).include? :new }
-            log_old = tables.filter { |table| Waldit.store_changes.call(table).include? :old }
-            log_diff = tables.filter { |table| Waldit.store_changes.call(table).include? :diff }
-            record.where(transaction_id: event.transaction_id).update_all(<<~SQL)
-              new = CASE WHEN table_name = ANY (ARRAY[#{log_new.map { |table| "'#{table}'" }.join(",")}]::varchar[]) THEN new ELSE null END,
-              old = CASE WHEN table_name = ANY (ARRAY[#{log_old.map { |table| "'#{table}'" }.join(",")}]::varchar[]) THEN old ELSE null END,
+            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.map { |table| "'#{table}'" }.join(",")}]::varchar[]) THEN (
+                CASE WHEN table_name = ANY (ARRAY[#{log_diff}]::varchar[]) THEN (
                   SELECT
                     jsonb_object_agg(
                       coalesce(old_kv.key, new_kv.key),

data/lib/waldit.rb

@@ -27,10 +27,10 @@ module Waldit
       case changes
       when Symbol
         changes = [changes].to_set
-        @store_changes = -> { changes }
+        @store_changes = -> table { changes }
       when Array
         changes = changes.map(&:to_sym).to_set
-        @store_changes = -> { changes }
+        @store_changes = -> table { changes }
       else
         @store_changes = changes
       end
@@ -51,7 +51,7 @@ module Waldit
 
     config.watched_tables = -> table { table != "waldit" }
 
-    config.store_changes = -> table { %i[old new diff] }
+    config.store_changes = -> table { %i[old new] }
 
     config.ignored_columns = -> table { %w[created_at updated_at] }
 

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: waldit
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.8
 platform: ruby
 authors:
 - Rodrigo Navarro
 bindir: exe
 cert_chain: []
-date: 2025-08-12 00:00:00.000000000 Z
+date: 2025-08-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: wal
@@ -64,6 +64,7 @@ files:
 - Rakefile
 - lib/waldit.rb
 - lib/waldit/context.rb
+- lib/waldit/migration.rb
 - lib/waldit/postgresql_adapter.rb
 - lib/waldit/railtie.rb
 - lib/waldit/record.rb