pgbus 0.6.8 → 0.6.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b4d9e527a08c8660868590840ae8f3f311e9f4d7fe20f1f2b83018c9cc30d904
-  data.tar.gz: 9113b68be4f3772f74488e4229685220a1daa649d3d8726d2fd044afbe1b070a
+  metadata.gz: 84c8f674c6aa6d83d0120f586b93f0fc4126ccb5d72378ec04a9b8cb3d464677
+  data.tar.gz: 2544b1be7c34ccf50493ce0af717c3069e979d95ded26197340d8133d637378f
 SHA512:
-  metadata.gz: 745010fa781503049ab585e6cc08b21420b8d118017be91eaab35d3c94c681836ff5f98c99d5f099fce5249cb45fc7d8b725e06dc569d1c040084a6b2d086911
-  data.tar.gz: 50df9a8d99532bf4efa3c03dd0be48e24f123f8ea937f0f7475ff8dfcf1d0242267568f426bdcc76089bacdd1538926b748f032ff977841b08f3665723e4c792
+  metadata.gz: bb329bb54c6eb7da4341eb8050657007d4fe057e862db7b8ef3c6cb0612abff33eaa383d7ea0ce00844dab1a532bd4b45ad3c52618c9ef45b7f9cf9d693ba199
+  data.tar.gz: 8ed028896561f51cb8a904f8e692125e6e61335501fceae366f5cda3f22daf7379b1b83d472df940c4965e2f0b2e00aed30ab90e3227d185e797ee17fb9654b6

data/lib/generators/pgbus/add_failed_events_index_generator.rb

@@ -2,11 +2,13 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require_relative "migration_path"
 
 module Pgbus
   module Generators
     class AddFailedEventsIndexGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
+      include MigrationPath
 
       source_root File.expand_path("templates", __dir__)
 
@@ -18,13 +20,8 @@ module Pgbus
                    desc: "Use a separate database for pgbus tables (e.g. --database=pgbus)"
 
       def create_migration_file
-        if separate_database?
-          migration_template "add_failed_events_unique_index.rb.erb",
-                             "db/pgbus_migrate/add_pgbus_failed_events_unique_index.rb"
-        else
-          migration_template "add_failed_events_unique_index.rb.erb",
-                             "db/migrate/add_pgbus_failed_events_unique_index.rb"
-        end
+        migration_template "add_failed_events_unique_index.rb.erb",
+                           File.join(pgbus_migrate_path, "add_pgbus_failed_events_unique_index.rb")
       end
 
       def display_post_install
@@ -42,10 +39,6 @@ module Pgbus
       def migration_version
         "[#{ActiveRecord::Migration.current_version}]"
       end
-
-      def separate_database?
-        options[:database].present?
-      end
     end
   end
 end

data/lib/generators/pgbus/add_job_locks_generator.rb

@@ -2,11 +2,13 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require_relative "migration_path"
 
 module Pgbus
   module Generators
     class AddJobLocksGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
+      include MigrationPath
 
       source_root File.expand_path("templates", __dir__)
 
@@ -18,13 +20,8 @@ module Pgbus
                    desc: "Use a separate database for pgbus tables (e.g. --database=pgbus)"
 
       def create_migration_file
-        if separate_database?
-          migration_template "add_job_locks.rb.erb",
-                             "db/pgbus_migrate/add_pgbus_job_locks.rb"
-        else
-          migration_template "add_job_locks.rb.erb",
-                             "db/migrate/add_pgbus_job_locks.rb"
-        end
+        migration_template "add_job_locks.rb.erb",
+                           File.join(pgbus_migrate_path, "add_pgbus_job_locks.rb")
       end
 
       def display_post_install
@@ -43,10 +40,6 @@ module Pgbus
       def migration_version
         "[#{ActiveRecord::Migration.current_version}]"
       end
-
-      def separate_database?
-        options[:database].present?
-      end
     end
   end
 end

data/lib/generators/pgbus/add_job_stats_generator.rb

@@ -2,11 +2,13 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require_relative "migration_path"
 
 module Pgbus
   module Generators
     class AddJobStatsGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
+      include MigrationPath
 
       source_root File.expand_path("templates", __dir__)
 
@@ -18,13 +20,8 @@ module Pgbus
                    desc: "Use a separate database for pgbus tables (e.g. --database=pgbus)"
 
       def create_migration_file
-        if separate_database?
-          migration_template "add_job_stats.rb.erb",
-                             "db/pgbus_migrate/add_pgbus_job_stats.rb"
-        else
-          migration_template "add_job_stats.rb.erb",
-                             "db/migrate/add_pgbus_job_stats.rb"
-        end
+        migration_template "add_job_stats.rb.erb",
+                           File.join(pgbus_migrate_path, "add_pgbus_job_stats.rb")
       end
 
       def display_post_install
@@ -43,10 +40,6 @@ module Pgbus
       def migration_version
         "[#{ActiveRecord::Migration.current_version}]"
       end
-
-      def separate_database?
-        options[:database].present?
-      end
     end
   end
 end

data/lib/generators/pgbus/add_job_stats_latency_generator.rb

@@ -2,11 +2,13 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require_relative "migration_path"
 
 module Pgbus
   module Generators
     class AddJobStatsLatencyGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
+      include MigrationPath
 
       source_root File.expand_path("templates", __dir__)
 
@@ -18,13 +20,8 @@ module Pgbus
                    desc: "Use a separate database for pgbus tables (e.g. --database=pgbus)"
 
       def create_migration_file
-        if separate_database?
-          migration_template "add_job_stats_latency.rb.erb",
-                             "db/pgbus_migrate/add_pgbus_job_stats_latency.rb"
-        else
-          migration_template "add_job_stats_latency.rb.erb",
-                             "db/migrate/add_pgbus_job_stats_latency.rb"
-        end
+        migration_template "add_job_stats_latency.rb.erb",
+                           File.join(pgbus_migrate_path, "add_pgbus_job_stats_latency.rb")
       end
 
       def display_post_install
@@ -43,10 +40,6 @@ module Pgbus
       def migration_version
         "[#{ActiveRecord::Migration.current_version}]"
       end
-
-      def separate_database?
-        options[:database].present?
-      end
     end
   end
 end

data/lib/generators/pgbus/add_job_stats_queue_index_generator.rb

@@ -2,11 +2,13 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require_relative "migration_path"
 
 module Pgbus
   module Generators
     class AddJobStatsQueueIndexGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
+      include MigrationPath
 
       source_root File.expand_path("templates", __dir__)
 
@@ -18,13 +20,8 @@ module Pgbus
                    desc: "Use a separate database for pgbus tables (e.g. --database=pgbus)"
 
       def create_migration_file
-        if separate_database?
-          migration_template "add_job_stats_queue_index.rb.erb",
-                             "db/pgbus_migrate/add_pgbus_job_stats_queue_index.rb"
-        else
-          migration_template "add_job_stats_queue_index.rb.erb",
-                             "db/migrate/add_pgbus_job_stats_queue_index.rb"
-        end
+        migration_template "add_job_stats_queue_index.rb.erb",
+                           File.join(pgbus_migrate_path, "add_pgbus_job_stats_queue_index.rb")
       end
 
       def display_post_install
@@ -44,10 +41,6 @@ module Pgbus
       def migration_version
         "[#{ActiveRecord::Migration.current_version}]"
       end
-
-      def separate_database?
-        options[:database].present?
-      end
     end
   end
 end

data/lib/generators/pgbus/add_outbox_generator.rb

@@ -2,11 +2,13 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require_relative "migration_path"
 
 module Pgbus
   module Generators
     class AddOutboxGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
+      include MigrationPath
 
       source_root File.expand_path("templates", __dir__)
 
@@ -18,13 +20,8 @@ module Pgbus
                    desc: "Use a separate database for pgbus tables (e.g. --database=pgbus)"
 
       def create_migration_file
-        if separate_database?
-          migration_template "add_outbox.rb.erb",
-                             "db/pgbus_migrate/add_pgbus_outbox.rb"
-        else
-          migration_template "add_outbox.rb.erb",
-                             "db/migrate/add_pgbus_outbox.rb"
-        end
+        migration_template "add_outbox.rb.erb",
+                           File.join(pgbus_migrate_path, "add_pgbus_outbox.rb")
       end
 
       def display_post_install
@@ -43,10 +40,6 @@ module Pgbus
       def migration_version
         "[#{ActiveRecord::Migration.current_version}]"
       end
-
-      def separate_database?
-        options[:database].present?
-      end
     end
   end
 end

data/lib/generators/pgbus/add_presence_generator.rb

@@ -2,11 +2,13 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require_relative "migration_path"
 
 module Pgbus
   module Generators
     class AddPresenceGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
+      include MigrationPath
 
       source_root File.expand_path("templates", __dir__)
 
@@ -18,13 +20,8 @@ module Pgbus
                    desc: "Use a separate database for pgbus tables (e.g. --database=pgbus)"
 
       def create_migration_file
-        if separate_database?
-          migration_template "add_presence.rb.erb",
-                             "db/pgbus_migrate/add_pgbus_presence.rb"
-        else
-          migration_template "add_presence.rb.erb",
-                             "db/migrate/add_pgbus_presence.rb"
-        end
+        migration_template "add_presence.rb.erb",
+                           File.join(pgbus_migrate_path, "add_pgbus_presence.rb")
       end
 
       def display_post_install
@@ -46,10 +43,6 @@ module Pgbus
       def migration_version
         "[#{ActiveRecord::Migration.current_version}]"
       end
-
-      def separate_database?
-        options[:database].present?
-      end
     end
   end
 end

data/lib/generators/pgbus/add_queue_states_generator.rb

@@ -2,11 +2,13 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require_relative "migration_path"
 
 module Pgbus
   module Generators
     class AddQueueStatesGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
+      include MigrationPath
 
       source_root File.expand_path("templates", __dir__)
 
@@ -18,13 +20,8 @@ module Pgbus
                    desc: "Use a separate database for pgbus tables (e.g. --database=pgbus)"
 
       def create_migration_file
-        if separate_database?
-          migration_template "add_queue_states.rb.erb",
-                             "db/pgbus_migrate/add_pgbus_queue_states.rb"
-        else
-          migration_template "add_queue_states.rb.erb",
-                             "db/migrate/add_pgbus_queue_states.rb"
-        end
+        migration_template "add_queue_states.rb.erb",
+                           File.join(pgbus_migrate_path, "add_pgbus_queue_states.rb")
       end
 
       def display_post_install
@@ -42,10 +39,6 @@ module Pgbus
       def migration_version
         "[#{ActiveRecord::Migration.current_version}]"
       end
-
-      def separate_database?
-        options[:database].present?
-      end
     end
   end
 end

data/lib/generators/pgbus/add_recurring_generator.rb

@@ -2,11 +2,13 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require_relative "migration_path"
 
 module Pgbus
   module Generators
     class AddRecurringGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
+      include MigrationPath
 
       source_root File.expand_path("templates", __dir__)
 
@@ -18,13 +20,8 @@ module Pgbus
                    desc: "Use a separate database for pgbus tables (e.g. --database=pgbus)"
 
       def create_migration_file
-        if separate_database?
-          migration_template "add_recurring_tables.rb.erb",
-                             "db/pgbus_migrate/add_pgbus_recurring_tables.rb"
-        else
-          migration_template "add_recurring_tables.rb.erb",
-                             "db/migrate/add_pgbus_recurring_tables.rb"
-        end
+        migration_template "add_recurring_tables.rb.erb",
+                           File.join(pgbus_migrate_path, "add_pgbus_recurring_tables.rb")
       end
 
       def create_recurring_config
@@ -47,10 +44,6 @@ module Pgbus
       def migration_version
         "[#{ActiveRecord::Migration.current_version}]"
       end
-
-      def separate_database?
-        options[:database].present?
-      end
     end
   end
 end

data/lib/generators/pgbus/add_stream_stats_generator.rb

@@ -2,11 +2,13 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require_relative "migration_path"
 
 module Pgbus
   module Generators
     class AddStreamStatsGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
+      include MigrationPath
 
       source_root File.expand_path("templates", __dir__)
 
@@ -18,13 +20,8 @@ module Pgbus
                    desc: "Use a separate database for pgbus tables (e.g. --database=pgbus)"
 
       def create_migration_file
-        if separate_database?
-          migration_template "add_stream_stats.rb.erb",
-                             "db/pgbus_migrate/add_pgbus_stream_stats.rb"
-        else
-          migration_template "add_stream_stats.rb.erb",
-                             "db/migrate/add_pgbus_stream_stats.rb"
-        end
+        migration_template "add_stream_stats.rb.erb",
+                           File.join(pgbus_migrate_path, "add_pgbus_stream_stats.rb")
       end
 
       def display_post_install
@@ -45,10 +42,6 @@ module Pgbus
       def migration_version
         "[#{ActiveRecord::Migration.current_version}]"
       end
-
-      def separate_database?
-        options[:database].present?
-      end
     end
   end
 end

data/lib/generators/pgbus/install_generator.rb

@@ -2,11 +2,13 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require_relative "migration_path"
 
 module Pgbus
   module Generators
     class InstallGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
+      include MigrationPath
 
       source_root File.expand_path("templates", __dir__)
 
@@ -25,13 +27,8 @@ module Pgbus
                          "Migrations go to db/pgbus_migrate/ and schema to db/pgbus_schema.rb"
 
       def create_migration_file
-        if separate_database?
-          migration_template "migration.rb.erb",
-                             "db/pgbus_migrate/create_pgbus_tables.rb"
-        else
-          migration_template "migration.rb.erb",
-                             "db/migrate/create_pgbus_tables.rb"
-        end
+        migration_template "migration.rb.erb",
+                           File.join(pgbus_migrate_path, "create_pgbus_tables.rb")
       end
 
       def create_config_file
@@ -121,10 +118,6 @@ module Pgbus
       def database_name
         options[:database]
       end
-
-      def separate_database?
-        database_name.present?
-      end
     end
   end
 end

data/lib/generators/pgbus/migrate_job_locks_generator.rb

@@ -2,11 +2,13 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require_relative "migration_path"
 
 module Pgbus
   module Generators
     class MigrateJobLocksGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
+      include MigrationPath
 
       source_root File.expand_path("templates", __dir__)
 
@@ -18,13 +20,8 @@ module Pgbus
                    desc: "Use a separate database for pgbus tables (e.g. --database=pgbus)"
 
       def create_migration_file
-        if separate_database?
-          migration_template "migrate_job_locks_to_uniqueness_keys.rb.erb",
-                             "db/pgbus_migrate/migrate_pgbus_job_locks_to_uniqueness_keys.rb"
-        else
-          migration_template "migrate_job_locks_to_uniqueness_keys.rb.erb",
-                             "db/migrate/migrate_pgbus_job_locks_to_uniqueness_keys.rb"
-        end
+        migration_template "migrate_job_locks_to_uniqueness_keys.rb.erb",
+                           File.join(pgbus_migrate_path, "migrate_pgbus_job_locks_to_uniqueness_keys.rb")
       end
 
       def display_post_install
@@ -47,10 +44,6 @@ module Pgbus
       def migration_version
         "[#{ActiveRecord::Migration.current_version}]"
       end
-
-      def separate_database?
-        options[:database].present?
-      end
     end
   end
 end

data/lib/generators/pgbus/migration_path.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+module Pgbus
+  module Generators
+    # Shared migration path logic for all pgbus generators.
+    #
+    # When --database is passed, uses Rails' built-in db_migrate_path which
+    # reads migrations_paths from database.yml. This is the standard Rails
+    # multi-database mechanism and respects custom paths like db/pgbus_migrate/.
+    #
+    # Falls back to db/migrate/ when no --database is set (single-database mode).
+    module MigrationPath
+      private
+
+      def pgbus_migrate_path
+        if separate_database?
+          db_migrate_path
+        else
+          "db/migrate"
+        end
+      end
+
+      def separate_database?
+        options[:database].present?
+      end
+    end
+  end
+end

data/lib/generators/pgbus/tune_autovacuum_generator.rb

@@ -2,11 +2,13 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require_relative "migration_path"
 
 module Pgbus
   module Generators
     class TuneAutovacuumGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
+      include MigrationPath
 
       source_root File.expand_path("templates", __dir__)
 
@@ -18,13 +20,8 @@ module Pgbus
                    desc: "Use a separate database for pgbus tables (e.g. --database=pgbus)"
 
       def create_migration_file
-        if separate_database?
-          migration_template "tune_autovacuum.rb.erb",
-                             "db/pgbus_migrate/tune_pgbus_autovacuum.rb"
-        else
-          migration_template "tune_autovacuum.rb.erb",
-                             "db/migrate/tune_pgbus_autovacuum.rb"
-        end
+        migration_template "tune_autovacuum.rb.erb",
+                           File.join(pgbus_migrate_path, "tune_pgbus_autovacuum.rb")
       end
 
       def display_post_install
@@ -46,10 +43,6 @@ module Pgbus
       def migration_version
         "[#{ActiveRecord::Migration.current_version}]"
       end
-
-      def separate_database?
-        options[:database].present?
-      end
     end
   end
 end

data/lib/generators/pgbus/upgrade_pgmq_generator.rb

@@ -2,11 +2,13 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require_relative "migration_path"
 
 module Pgbus
   module Generators
     class UpgradePgmqGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
+      include MigrationPath
 
       source_root File.expand_path("templates", __dir__)
 
@@ -18,13 +20,8 @@ module Pgbus
                    desc: "Use a separate database for pgbus tables (e.g. --database=pgbus)"
 
       def create_migration_file
-        if separate_database?
-          migration_template "upgrade_pgmq.rb.erb",
-                             "db/pgbus_migrate/upgrade_pgmq_to_v#{target_version_slug}.rb"
-        else
-          migration_template "upgrade_pgmq.rb.erb",
-                             "db/migrate/upgrade_pgmq_to_v#{target_version_slug}.rb"
-        end
+        migration_template "upgrade_pgmq.rb.erb",
+                           File.join(pgbus_migrate_path, "upgrade_pgmq_to_v#{target_version_slug}.rb")
       end
 
       def display_post_upgrade
@@ -51,10 +48,6 @@ module Pgbus
       def target_version_slug
         target_version.tr(".", "_")
       end
-
-      def separate_database?
-        options[:database].present?
-      end
     end
   end
 end

data/lib/pgbus/engine.rb

@@ -47,6 +47,7 @@ module Pgbus
     rake_tasks do
       load File.expand_path("../tasks/pgbus_pgmq.rake", __dir__)
       load File.expand_path("../tasks/pgbus_streams.rake", __dir__)
+      load File.expand_path("../tasks/pgbus_autovacuum.rake", __dir__)
     end
 
     initializer "pgbus.i18n" do

data/lib/pgbus/version.rb

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

data/lib/tasks/pgbus_autovacuum.rake

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+namespace :pgbus do
+  desc "Apply autovacuum tuning to PGMQ queue/archive tables and high-churn pgbus tables"
+  task tune_autovacuum: :environment do
+    require "pgbus/autovacuum_tuning"
+
+    conn = Pgbus.configuration.connects_to ? Pgbus::BusRecord.connection : ActiveRecord::Base.connection
+
+    # Only run if pgmq schema exists (tables may not be created yet during
+    # initial setup — the install migration handles tuning itself).
+    pgmq_exists = conn.select_value(
+      "SELECT 1 FROM information_schema.schemata WHERE schema_name = 'pgmq'"
+    )
+
+    unless pgmq_exists
+      puts "[pgbus] PGMQ schema not found — skipping autovacuum tuning."
+      next
+    end
+
+    puts "[pgbus] Applying autovacuum tuning to PGMQ queue/archive tables..."
+    conn.execute(Pgbus::AutovacuumTuning.sql_for_all_queues)
+
+    puts "[pgbus] Applying autovacuum tuning to high-churn pgbus tables..."
+    conn.execute(Pgbus::AutovacuumTuning.sql_for_high_churn_tables)
+
+    puts "[pgbus] Autovacuum tuning complete."
+  end
+end
+
+# Reapply autovacuum settings after schema:load since Ruby-format schema.rb
+# does not preserve ALTER TABLE ... SET (reloptions). This is a no-op if the
+# tables don't exist yet (IF EXISTS guards in the SQL).
+%w[db:schema:load db:schema:load:pgbus].each do |task_name|
+  next unless Rake::Task.task_defined?(task_name)
+
+  Rake::Task[task_name].enhance do
+    Rake::Task["pgbus:tune_autovacuum"].invoke if Rake::Task.task_defined?("pgbus:tune_autovacuum")
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pgbus
 version: !ruby/object:Gem::Version
-  version: 0.6.8
+  version: 0.6.9
 platform: ruby
 authors:
 - Mikael Henriksson
@@ -215,6 +215,7 @@ files:
 - lib/generators/pgbus/add_stream_stats_generator.rb
 - lib/generators/pgbus/install_generator.rb
 - lib/generators/pgbus/migrate_job_locks_generator.rb
+- lib/generators/pgbus/migration_path.rb
 - lib/generators/pgbus/templates/add_failed_events_unique_index.rb.erb
 - lib/generators/pgbus/templates/add_job_locks.rb.erb
 - lib/generators/pgbus/templates/add_job_stats.rb.erb
@@ -319,6 +320,7 @@ files:
 - lib/pgbus/web/streamer/registry.rb
 - lib/pgbus/web/streamer/stream_event_dispatcher.rb
 - lib/puma/plugin/pgbus_streams.rb
+- lib/tasks/pgbus_autovacuum.rake
 - lib/tasks/pgbus_pgmq.rake
 - lib/tasks/pgbus_streams.rake
 homepage: https://github.com/mhenrixon/pgbus