wal 0.0.23 → 0.0.25

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3a7144e35e8bfe1df54e18fa1e4b9c94e4bb6926ed0de011e0e4006ce08c44cd
-  data.tar.gz: 21196f54b92390273288aac5fed7b5fdcd79842dda0d63cbb0cada583bca8cd3
+  metadata.gz: 7ff2f0a5d9392268db27728b053f29ac897649314c72b857b0d43303bb8face1
+  data.tar.gz: a378a4d4c4e5f3f1e6d9a988fe61947683c52614223e27a67371b4042434a2cb
 SHA512:
-  metadata.gz: 88be295b55c90871dc4aa8731e21c03fff1b890e59a3b9ec000276c5d0b2dad96660b1686328af6e208084fc5389f90c8925dbc27fb29de56e535063eaf20899
-  data.tar.gz: a13bd85ed047f5b7bfcfe839bb0ff8643587ae6c6ca81ef4f56809a96e4daff6a6f7423c9eff3a715cfa72f27e84b99765f003b4d40b0eacabf0d73cc7311bbc
+  metadata.gz: 5514c83c92062744f25cdc2f7f08aaa79cdbd4bc7cd49c6028de964f3d5c02a0a0b8d2d3b8f3a6cdf69954adb45f5a89866f5fdd193cebfd1cb1fc24049d3ba9
+  data.tar.gz: a29bb6e5c5cc9795e82d88d15f4e820cc88b651f58950c67120dc988454aa6230d0e85af25e016289402c5bb5a6b1db9341e039fdc350155a86710190be6bad2

data/exe/wal

@@ -69,7 +69,7 @@ class Wal::LoggingReplicator
   end
 end
 
-if cli["watch"]
+def watch(db_config, cli)
   watcher = cli["--watcher"].constantize.new
   use_temporary_slot = cli["--tmp-slot"] || false
   replication_slot = cli["--slot"]
@@ -82,46 +82,50 @@ if cli["watch"]
   replicator = Wal::LoggingReplicator.new(replication_slot, replicator)
   replicator.replicate_forever(watcher, publications:)
   puts "Watcher finished for #{replication_slot}"
+end
 
-elsif cli["start"]
-  slots = YAML.load_file(cli["<config-file>"])["slots"]
-
-  workers = slots.map do |slot, config|
-    watcher = config["watcher"].constantize.new
-    temporary = config["temporary"] || false
-    publications = config["publications"] || []
-    replicator_class = config["replicator"].presence&.constantize || Wal::Replicator
-    max_retries = config["retries"]&.to_i || (2**32 - 1)
-    retries = 0
-    backoff = config["retry_backoff"]&.to_f || 1
-    backoff_expoent = config["retry_backoff_expoent"]&.to_f
-
-    Thread.new(slot, watcher, temporary, publications) do |replication_slot, watcher, use_temporary_slot, publications|
-      replication_slot = "#{replication_slot}_#{SecureRandom.alphanumeric(4)}" if use_temporary_slot
-      puts "Watcher started for #{replication_slot} slot (#{publications.join(", ")})"
-
+def start_worker(db_config, slot, config)
+  watcher = config["watcher"].constantize.new
+  temporary = config["temporary"] || false
+  publications = config["publications"] || []
+  replicator_class = config["replicator"].presence&.constantize || Wal::Replicator
+  max_retries = config["retries"]&.to_i || (2**32 - 1)
+  retries = 0
+  backoff = config["retry_backoff"]&.to_f || 1
+  backoff_expoent = config["retry_backoff_expoent"]&.to_f
+
+  Thread.new(slot, watcher, temporary, publications) do |replication_slot, watcher, use_temporary_slot, publications|
+    replication_slot = "#{replication_slot}_#{SecureRandom.alphanumeric(4)}" if use_temporary_slot
+    puts "Watcher started for #{replication_slot} slot (#{publications.join(", ")})"
+
+    begin
       replicator = replicator_class.new(replication_slot:, use_temporary_slot:, db_config:)
       replicator = Wal::LoggingReplicator.new(replication_slot, replicator)
-
-      begin
-        replicator.replicate_forever(watcher, publications:)
-      rescue StandardError => err
-        if retries < max_retries
-          Wal.logger&.error("[#{replication_slot}] Error #{err}")
-          retries += 1
-          backoff_time = backoff_expoent ? (backoff * retries) ** backoff_expoent : backoff
-          puts "Restarting #{replication_slot} in #{backoff_time.floor(2)}s..."
-          sleep backoff_time
-          puts "Restarting #{replication_slot}"
-          retry
-        end
-        raise
+      replicator.replicate_forever(watcher, publications:)
+    rescue StandardError => err
+      if retries < max_retries
+        Wal.logger&.error("[#{replication_slot}] Error #{err}")
+        retries += 1
+        backoff_time = backoff_expoent ? (backoff * retries) ** backoff_expoent : backoff
+        puts "Restarting #{replication_slot} in #{backoff_time.floor(2)}s..."
+        sleep backoff_time
+        puts "Restarting #{replication_slot}"
+        retry
       end
+      raise
+    end
 
-      puts "Watcher finished for #{replication_slot}"
+    puts "Watcher finished for #{replication_slot}"
 
-      Process.kill("TERM", Process.pid)
-    end
+    Process.kill("TERM", Process.pid)
+  end
+end
+
+def start(db_config, cli)
+  slots = YAML.load_file(cli["<config-file>"])["slots"]
+
+  workers = slots.map do |slot, config|
+    start_worker(db_config, slot, config)
   end
 
   ping = Thread.new do
@@ -147,3 +151,9 @@ elsif cli["start"]
 
   workers.each(&:join)
 end
+
+if cli["watch"]
+  watch(db_config, cli)
+elsif cli["start"]
+  start(db_config, cli)
+end

data/lib/wal/railtie.rb

@@ -2,7 +2,7 @@
 
 require "rails/railtie"
 
-module Waldit
+module Wal
   class Railtie < Rails::Railtie
     generators do
       require_relative "generators/migration"
@@ -13,5 +13,19 @@ module Waldit
       ActiveRecord::Migration.include Wal::Migration
       ActiveRecord::Schema.include Wal::Migration
     end
+
+    config.after_initialize do
+      # "Borrowed" from Sidekiq
+      # https://github.com/sidekiq/sidekiq/blob/61e27d20b1ed62f203eee6ae2b549f2e53db14c9/lib/sidekiq/rails.rb#L38-L58
+      unless Rails.logger == Wal.logger || ActiveSupport::Logger.logger_outputs_to?(Rails.logger, $stdout)
+        if Rails.logger.respond_to?(:broadcast_to)
+          Rails.logger.broadcast_to(Wal.logger)
+        elsif ActiveSupport::Logger.respond_to?(:broadcast)
+          Rails.logger.extend(ActiveSupport::Logger.broadcast(Wal.logger))
+        else
+          Rails.logger = ActiveSupport::BroadcastLogger.new(Rails.logger, Wal.logger)
+        end
+      end
+    end
   end
 end

data/lib/wal/version.rb

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

data/lib/wal.rb

@@ -14,6 +14,10 @@ require_relative "wal/version"
 module Wal
   class << self
     attr_accessor :logger
+
+    def logger
+      @logger ||= Logger.new($stdout, level: :info)
+    end
   end
 
   def self.configure(&block)

data/rbi/wal.rbi

@@ -7,7 +7,7 @@ module Wal
       UpdateEvent,
       DeleteEvent,
     ) }
-  VERSION = "0.0.21"
+  VERSION = "0.0.25"
 
   class << self
     sig { returns(T.class_of(Logger)) }

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: wal
 version: !ruby/object:Gem::Version
-  version: 0.0.23
+  version: 0.0.25
 platform: ruby
 authors:
 - Rodrigo Navarro
 bindir: exe
 cert_chain: []
-date: 2025-10-29 00:00:00.000000000 Z
+date: 2026-01-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg