wal 0.0.7 → 0.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f69fdfa02c0ec9015408b179140ab2086041ffc4f895e2d0cfd8f1702e32ad5b
-  data.tar.gz: a6bbc4b59d098e39f159875e782e174f4d2700b5bad3281979e1f25deb40ad12
+  metadata.gz: '08cf62e21e9cd21bca29345d2e666d2d390e886d4ea8c43b9926b2807d7732ac'
+  data.tar.gz: 6396204641b2b13f6dc64291f61072f2283a808a271fbed4f1a6b84dfe59245f
 SHA512:
-  metadata.gz: fd490b22e2db6e1a0a6e5f2b667234c3a2f8c6fc488f54bedfcb21660f2259ffbf47498f0227da72ce94ac59ee0177b2363c3fc244a8b300fde48777ee3ad79b
-  data.tar.gz: f7d8e93fd8a91e0e1d4d37d7fc25d5d698eadf9e145db45d4078a137e9ffc5e5ca8a9c1f3f26ee19c9e7920a863dd6482cc2322d6a81b296635ab003e8f0bb92
+  metadata.gz: 4aa20fe9dd40ba5c66eb8d9499be96590bd2ab51e835c49d09e91307c1b978809e266e781312e502264bd446294e273a5796ad98dd9eece2ea7cd9a98f86a6e3
+  data.tar.gz: ed72b21ca9e0e6d770d0cf2121f8ec9fb22aacf47b6cfd46d70946ec5200bc2d57feeb211d7edc65875171540f48f94dc376ccb0d3e4c38ffc855812eb6fb089

data/lib/wal/generators/migration.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+require "rails/generators"
+require "rails/generators/active_record"
+
+module Wal
+  module Generators
+    class Migration < Rails::Generators::Base
+      include Rails::Generators::Migration
+
+      source_root File.expand_path("templates", __dir__)
+
+      argument :watcher, type: :string
+
+      def self.next_migration_number(dir)
+        ::ActiveRecord::Generators::Base.next_migration_number(dir)
+      end
+
+      def create_migration_file
+        migration_template("migration.rb.erb", "db/migrate/#{migration_class_name.underscore}.rb")
+      end
+
+      private
+
+      def activerecord_migration_class
+        if ActiveRecord::Migration.respond_to? :current_version
+          "ActiveRecord::Migration[#{ActiveRecord::Migration.current_version}]"
+        else
+          "ActiveRecord::Migration"
+        end
+      end
+
+      def publication_name
+        "#{watcher.underscore}_publication"
+      end
+
+      def class_name
+        watcher.camelize
+      end
+
+      def migration_class_name
+        "Set#{class_name}Publication"
+      end
+
+      def publication_tables
+        if (watcher = class_name.constantize) && watcher < Wal::RecordWatcher
+          tables = watcher
+            .change_callbacks
+            .map { |table, configs| [table.to_s, configs.flat_map { |config| config[:changed] || [] }.uniq] }
+            .to_h
+
+          watcher
+            .delete_callbacks
+            .keys
+            .reduce(tables) { |tables, table| tables.reverse_merge(table => []) }
+        else
+          []
+        end
+      end
+    end
+  end
+end

data/lib/wal/generators/templates/migration.rb.erb

@@ -0,0 +1,9 @@
+class <%= migration_class_name %> < <%= activerecord_migration_class %>
+  def change
+    define_publication :<%= publication_name %> do |p|
+<% publication_tables.each do |table, columns| -%>
+      p.table :<%= table %><%= columns.present? ? ", columns: #{columns}" : "" %>
+<% end -%>
+    end
+  end
+end

data/lib/wal/migration.rb

@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+
+module Wal
+  module Migration
+    class PublicationDefinition
+      def initialize(name)
+        @name = name
+        @tables = {}
+      end
+
+      def table(name, columns: [])
+        @tables[name] ||= [].to_set
+        @tables[name].merge(columns.map(&:to_s))
+      end
+
+      def tables
+        @tables.keys
+      end
+
+      def to_sql
+        tables_sql = @tables.reduce([]) do |sql, (table, columns)|
+          table_sql = "TABLE #{table}"
+          table_sql << " (#{columns.join(", ")})" unless columns.empty?
+          sql << table_sql
+        end
+        "ALTER PUBLICATION #{@name} SET #{tables_sql.join(", ")}"
+      end
+    end
+
+    def define_publication(name)
+      execute <<~SQL
+        DO $$
+        BEGIN
+        CASE WHEN NOT EXISTS (SELECT 1 FROM pg_publication WHERE pubname = '#{name}') THEN
+          CREATE PUBLICATION #{name};
+        END CASE;
+        END;
+        $$;
+      SQL
+
+      definition = PublicationDefinition.new(name)
+      yield definition
+
+      execute definition.to_sql
+    end
+  end
+end

data/lib/wal/railtie.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "rails/railtie"
+
+module Waldit
+  class Railtie < Rails::Railtie
+    generators do
+      require_relative "generators/migration"
+    end
+
+    config.before_configuration do
+      require_relative "migration"
+      ActiveRecord::Migration.include Wal::Migration
+      ActiveRecord::Schema.include Wal::Migration
+    end
+  end
+end

data/lib/wal/record_watcher.rb

@@ -65,6 +65,14 @@ module Wal
       @@delete_callbacks[table].push(block: block)
     end
 
+    def self.change_callbacks
+      @@change_callbacks
+    end
+
+    def self.delete_callbacks
+      @@delete_callbacks
+    end
+
     def on_record_changed(event)
       case event
       when InsertEvent
@@ -104,8 +112,8 @@ module Wal
     #
     # These strategies can be defined per transaction, and by default it will uses the memory one, and only fallback
     # to the temporary table if the transaction size is roughly 2 gigabytes or more.
-    def aggregation_strategy(event)
-      if event.estimated_size > 1024.pow(3) * 2
+    def aggregation_strategy(begin_transaction_event)
+      if begin_transaction_event.estimated_size > 1024.pow(3) * 2
         :temporary_table
       else
         :memory

data/lib/wal/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Wal
-  VERSION = "0.0.7"
+  VERSION = "0.0.9"
 end

data/lib/wal.rb

@@ -8,6 +8,7 @@ require_relative "wal/record_watcher"
 require_relative "wal/streaming_watcher"
 require_relative "wal/replicator"
 require_relative "wal/active_record_context_extension"
+require_relative "wal/railtie"
 require_relative "wal/version"
 
 module Wal

data/rbi/wal.rbi

@@ -7,7 +7,7 @@ module Wal
       UpdateEvent,
       DeleteEvent,
     ) }
-  VERSION = "0.0.3"
+  VERSION = "0.0.9"
 
   class BeginTransactionEvent < T::Struct
     prop :transaction_id, Integer, immutable: true

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: wal
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.0.9
 platform: ruby
 authors:
 - Rodrigo Navarro
 bindir: exe
 cert_chain: []
-date: 2025-08-10 00:00:00.000000000 Z
+date: 2025-08-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg
@@ -93,7 +93,11 @@ files:
 - exe/wal
 - lib/wal.rb
 - lib/wal/active_record_context_extension.rb
+- lib/wal/generators/migration.rb
+- lib/wal/generators/templates/migration.rb.erb
+- lib/wal/migration.rb
 - lib/wal/noop_watcher.rb
+- lib/wal/railtie.rb
 - lib/wal/record_watcher.rb
 - lib/wal/replicator.rb
 - lib/wal/streaming_watcher.rb