pgbus 0.1.7 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 38d33f96cb4a6aa5255b33951818a1a975168d5d40b2b9bc391d8967e00e9854
-  data.tar.gz: 3c5dae7b2e4f31c63b81b1f11a4d6b96239f8bb5b7589a4465ae0a0834f88f8c
+  metadata.gz: b1e18f777c6b6f2dbc0f927994e19587a489f0551be9f5e228c4d47b3f346fa0
+  data.tar.gz: 433faca464a6f881a21cbcf950c3238cdd34e1e11fac0108a320dd98e23dfae1
 SHA512:
-  metadata.gz: 97330cb59c83f37706c2a1ae5b6ce3c309ab4113287b92e077b34de05242d1fa65b4b0eda488fdc050e5bf0fb896eca0837986f85f3781dbaf0c6ff9ea3248b3
-  data.tar.gz: ea013b795cf6b677373a1ecf161c7b85633d18f2abf95e0155d2bb870fa0f6f222c499511392d21c9ea9b1ee9ade9862a561d12e38c975f58873ed1abe0f0d13
+  metadata.gz: 89418720dfb7433c1716cff8e238e74fee2be4c04a8023a4c2dc3b67fdf393b9ddb9f7c4df70b435a6cdcc9a3668c1c24e78bedce83c0c22d93d196ab1ee1574
+  data.tar.gz: b10d5f71b80c97e42573b2b034c0c1b4444342d6e729fef0fd0e8a97dba269eb35cdca8fc4bfcb4446dd17e630683859c1b5d0043467218baf15b72b05fb3e38

data/lib/pgbus/client.rb

@@ -42,6 +42,12 @@ module Pgbus
             "PGMQ schema is not available (#{e.message}). Run `rails db:migrate` for the pgbus database."
     end
 
+    def ensure_all_queues
+      queue_names = collect_configured_queues
+      Pgbus.logger.info { "[Pgbus] Bootstrapping #{queue_names.size} queue(s): #{queue_names.join(", ")}" }
+      queue_names.each { |name| ensure_queue(name) }
+    end
+
     def ensure_dead_letter_queue(name)
       dlq_name = config.dead_letter_queue_name(name)
       return if @queues_created[dlq_name]
@@ -228,6 +234,26 @@ module Pgbus
 
     private
 
+    def collect_configured_queues
+      queues = Set.new
+      queues << config.default_queue
+
+      # Queues from worker configs
+      (config.workers || []).each do |w|
+        worker_queues = w[:queues] || w["queues"] || [config.default_queue]
+        worker_queues.each { |q| queues << q unless q == "*" }
+      end
+
+      # Queues from recurring tasks
+      (config.recurring_tasks || {}).each_value do |opts|
+        opts = opts.transform_keys(&:to_s) if opts.is_a?(Hash)
+        queue = opts["queue"] || opts[:queue]
+        queues << queue if queue
+      end
+
+      queues.to_a
+    end
+
     def ensure_single_queue(full_name)
       return if @queues_created[full_name]
 

data/lib/pgbus/process/supervisor.rb

@@ -70,6 +70,7 @@ module Pgbus
           restore_signals
           setup_child_process
           load_rails_app
+          bootstrap_queues
           worker = Worker.new(
             queues: queues, threads: threads, config: config,
             single_active_consumer: single_active, consumer_priority: priority
@@ -121,6 +122,7 @@ module Pgbus
           setup_child_process
           load_rails_app
           load_recurring_config
+          bootstrap_queues
           scheduler = Recurring::Scheduler.new(config: config)
           scheduler.run
         end
@@ -280,6 +282,12 @@ module Pgbus
         end
       end
 
+      def bootstrap_queues
+        Pgbus.client.ensure_all_queues
+      rescue StandardError => e
+        Pgbus.logger.error { "[Pgbus] Failed to bootstrap queues: #{e.message}" }
+      end
+
       def load_rails_app
         return unless defined?(Rails)
 

data/lib/pgbus/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Pgbus
-  VERSION = "0.1.7"
+  VERSION = "0.1.9"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pgbus
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.9
 platform: ruby
 authors:
 - Mikael Henriksson