pgbus 0.0.1 → 0.1.1

data/lib/pgbus/configuration.rb

@@ -31,9 +31,21 @@ module Pgbus
     # LISTEN/NOTIFY
     attr_accessor :listen_notify, :notify_throttle_ms
 
+    # PGMQ schema installation mode (:auto, :extension, :embedded)
+    attr_reader :pgmq_schema_mode
+
     # Event consumers
     attr_accessor :event_consumers
 
+    # Recurring jobs
+    attr_accessor :recurring_tasks, :recurring_schedule_interval, :recurring_tasks_file,
+                  :skip_recurring, :recurring_execution_retention
+
+    # Multi-database support (optional separate database for pgbus tables)
+    # Set to { database: { writing: :pgbus, reading: :pgbus } } to use a separate database.
+    # Requires a matching entry in config/database.yml under the "pgbus" key.
+    attr_accessor :connects_to
+
     # Web dashboard
     attr_accessor :web_auth, :web_refresh_interval, :web_per_page, :web_live_updates, :web_data_source
 
@@ -66,8 +78,18 @@ module Pgbus
       @listen_notify = true
       @notify_throttle_ms = 250
 
+      @pgmq_schema_mode = :auto
+
       @event_consumers = nil
 
+      @recurring_tasks = nil
+      @recurring_schedule_interval = 1.0
+      @recurring_tasks_file = nil
+      @skip_recurring = false
+      @recurring_execution_retention = 7 * 24 * 3600 # 7 days
+
+      @connects_to = nil
+
       @web_auth = nil
       @web_refresh_interval = 5000
       @web_per_page = 25
@@ -83,6 +105,17 @@ module Pgbus
       "#{queue_name(name)}#{dead_letter_queue_suffix}"
     end
 
+    VALID_PGMQ_SCHEMA_MODES = %i[auto extension embedded].freeze
+
+    def pgmq_schema_mode=(mode)
+      mode = mode.to_sym
+      unless VALID_PGMQ_SCHEMA_MODES.include?(mode)
+        raise ArgumentError, "Invalid pgmq_schema_mode: #{mode}. Must be one of: #{VALID_PGMQ_SCHEMA_MODES.join(", ")}"
+      end
+
+      @pgmq_schema_mode = mode
+    end
+
     def connection_options
       if database_url
         database_url

data/lib/pgbus/engine.rb

@@ -11,9 +11,23 @@ module Pgbus
       Pgbus::ConfigLoader.load(config_path) if config_path.exist?
     end
 
+    initializer "pgbus.recurring" do |app|
+      recurring_path = app.root.join("config", "recurring.yml")
+      if recurring_path.exist? && !Pgbus.configuration.recurring_tasks
+        Pgbus.configuration.recurring_tasks = Pgbus::Recurring::ConfigLoader.load(recurring_path)
+        Pgbus.configuration.recurring_tasks_file ||= recurring_path.to_s
+      end
+    end
+
+    initializer "pgbus.db" do
+      ActiveSupport.on_load(:active_record) do
+        Pgbus::ApplicationRecord.connects_to(**Pgbus.configuration.connects_to) if Pgbus.configuration.connects_to
+      end
+    end
+
     initializer "pgbus.active_job" do
       ActiveSupport.on_load(:active_job) do
-        require "pgbus/active_job/adapter"
+        include Pgbus::Concurrency
       end
     end
 
@@ -23,6 +37,10 @@ module Pgbus
       end
     end
 
+    rake_tasks do
+      load File.expand_path("../tasks/pgbus_pgmq.rake", __dir__)
+    end
+
     initializer "pgbus.web" do
       require "pgbus/web/authentication"
       require "pgbus/web/data_source"

data/lib/pgbus/event_bus/handler.rb

@@ -52,23 +52,13 @@ module Pgbus
       end
 
       def already_processed?(event_id)
-        return false unless defined?(ActiveRecord::Base)
-
-        ActiveRecord::Base.connection.select_value(
-          "SELECT 1 FROM pgbus_processed_events WHERE event_id = $1 AND handler_class = $2",
-          "Pgbus Idempotency Check",
-          [event_id, self.class.name]
-        )
+        ProcessedEvent.exists?(event_id: event_id, handler_class: self.class.name)
       end
 
       def mark_processed!(event_id)
-        return unless defined?(ActiveRecord::Base)
-
-        ActiveRecord::Base.connection.exec_insert(
-          "INSERT INTO pgbus_processed_events (event_id, handler_class, processed_at) " \
-          "VALUES ($1, $2, $3) ON CONFLICT (event_id, handler_class) DO NOTHING",
-          "Pgbus Idempotency Mark",
-          [event_id, self.class.name, Time.now.utc]
+        ProcessedEvent.upsert(
+          { event_id: event_id, handler_class: self.class.name, processed_at: Time.now.utc },
+          unique_by: %i[event_id handler_class]
         )
       end
     end

data/lib/pgbus/instrumentation.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module Pgbus
+  # Lightweight instrumentation via ActiveSupport::Notifications.
+  #
+  # All events are prefixed with "pgbus." and carry timing information
+  # automatically when used with the block form of AS::Notifications.instrument.
+  #
+  # Events emitted:
+  #   pgbus.client.send_message   — single message enqueue
+  #   pgbus.client.send_batch     — batch enqueue
+  #   pgbus.client.read_batch     — batch dequeue
+  #   pgbus.client.read_message   — single message dequeue
+  #   pgbus.executor.execute      — full job execution (deserialize + perform + archive)
+  #   pgbus.serializer.serialize  — job/event serialization
+  #   pgbus.serializer.deserialize — job/event deserialization
+  #
+  module Instrumentation
+    module_function
+
+    def instrument(event, payload = {}, &block)
+      if defined?(ActiveSupport::Notifications)
+        ActiveSupport::Notifications.instrument(event, payload, &block)
+      elsif block
+        yield payload
+      end
+    end
+  end
+end