active_postgres 0.9.0 → 0.9.2

data/lib/active_postgres/ssh_executor.rb

@@ -1,6 +1,7 @@
 require 'sshkit'
 require 'sshkit/dsl'
 require 'securerandom'
+require 'stringio'
 
 module ActivePostgres
   class SSHExecutor
@@ -294,18 +295,21 @@ module ActivePostgres
         SSHKit.config.format = :pretty
       end
 
-      return unless File.exist?(config.ssh_key)
-
       SSHKit::Backend::Netssh.configure do |ssh|
-        ssh.ssh_options = {
-          keys: [config.ssh_key],
-          keys_only: true,
+        options = {
           forward_agent: false,
           auth_methods: ['publickey'],
           verify_host_key: config.ssh_host_key_verification || :always,
           timeout: 10,
           number_of_password_prompts: 0
         }
+
+        if config.ssh_key && File.exist?(config.ssh_key)
+          options[:keys] = [config.ssh_key]
+          options[:keys_only] = true
+        end
+
+        ssh.ssh_options = options
       end
     end
   end

data/lib/active_postgres/version.rb

@@ -1,3 +1,3 @@
 module ActivePostgres
-  VERSION = '0.9.0'.freeze
+  VERSION = '0.9.2'.freeze
 end

data/lib/active_postgres.rb

@@ -23,6 +23,7 @@ require_relative 'active_postgres/installer'
 require_relative 'active_postgres/ssh_executor'
 require_relative 'active_postgres/direct_executor'
 require_relative 'active_postgres/health_checker'
+require_relative 'active_postgres/overview'
 require_relative 'active_postgres/failover'
 require_relative 'active_postgres/performance_tuner'
 require_relative 'active_postgres/connection_pooler'

data/lib/tasks/postgres.rake

@@ -150,6 +150,15 @@ namespace :postgres do
     health_checker.show_status
   end
 
+  desc 'Show control tower overview'
+  task overview: :environment do
+    require 'active_postgres'
+
+    config = ActivePostgres::Configuration.load
+    overview = ActivePostgres::Overview.new(config)
+    overview.show
+  end
+
   desc 'Visualize cluster nodes and topology'
   task nodes: :environment do
     require 'active_postgres'
@@ -270,6 +279,50 @@ namespace :postgres do
     puts "\n#{'=' * 80}\n"
   end
 
+  desc 'Show help for PostgreSQL rake tasks'
+  task help: :environment do
+    puts "\nPostgreSQL Rake Tasks"
+    puts '=' * 70
+    puts "\nTip: set RAILS_ENV=production for production targets"
+
+    puts "\nSetup & Maintenance"
+    puts "  rake postgres:setup                  # Deploy HA cluster (CLEAN=true for fresh install)"
+    puts "  rake postgres:purge                  # Destroy cluster (DESTRUCTIVE)"
+    puts "  rake postgres:setup:core             # PostgreSQL config (postgresql.conf, pg_hba.conf)"
+    puts "  rake postgres:setup:repmgr           # HA + failover (repmgr)"
+    puts "  rake postgres:setup:pgbouncer        # PgBouncer pooling"
+    puts "  rake postgres:setup:pgbackrest       # Backups (pgBackRest)"
+    puts "  rake postgres:setup:monitoring       # postgres_exporter"
+    puts "  rake postgres:setup:ssl              # SSL certs"
+
+    puts "\nStatus & Health"
+    puts "  rake postgres:status                 # Cluster status (SSH by default)"
+    puts "  rake postgres:overview               # Control tower overview"
+    puts "  rake postgres:nodes                  # Topology view"
+    puts "  rake postgres:verify                 # Comprehensive checklist"
+    puts "  ACTIVE_POSTGRES_STATUS_MODE=ssh|direct rake postgres:status"
+
+    puts "\nBackups"
+    puts "  rake postgres:backup:full            # Full backup"
+    puts "  rake postgres:backup:incremental     # Incremental backup"
+    puts "  rake postgres:backup:list            # List backups"
+    puts "  rake postgres:backup:restore[ID]     # Restore backup set"
+    puts "  rake postgres:backup:restore_at[\"YYYY-MM-DD HH:MM:SS\",promote]  # PITR"
+
+    puts "\nMigrations"
+    puts "  rake postgres:migrate                # Run migrations on primary only"
+
+    puts "\nPgBouncer"
+    puts "  rake postgres:pgbouncer:update_userlist[users]  # Update userlist"
+    puts "  rake postgres:pgbouncer:stats        # Service status + stats"
+
+    puts "\nTests"
+    puts "  rake postgres:test:replication[rows] # Replication stress test"
+    puts "  rake postgres:test:pgbouncer[connections]  # PgBouncer load test"
+
+    puts
+  end
+
   desc 'Promote standby to primary'
   task :promote, [:host] => :environment do |_t, args|
     require 'active_postgres'
@@ -325,6 +378,20 @@ namespace :postgres do
       installer.run_restore(args[:backup_id])
     end
 
+    desc 'Restore to a point in time (PITR)'
+    task :restore_at, [:target_time, :target_action] => :environment do |_t, args|
+      require 'active_postgres'
+
+      unless args[:target_time]
+        puts 'Usage: rake postgres:backup:restore_at["2026-01-29 01:15:00",promote]'
+        exit 1
+      end
+
+      config = ActivePostgres::Configuration.load
+      installer = ActivePostgres::Installer.new(config)
+      installer.run_restore_at(args[:target_time], target_action: args[:target_action] || 'promote')
+    end
+
     desc 'List available backups'
     task list: :environment do
       require 'active_postgres'

data/templates/repmgr.conf.erb

@@ -31,6 +31,16 @@ follow_command='repmgr standby follow -f /etc/repmgr.conf --upstream-node-id=%n'
 reconnect_attempts=<%= repmgr_config[:reconnect_attempts] || 6 %>
 reconnect_interval=<%= repmgr_config[:reconnect_interval] || 10 %>
 
+service_start_command='sudo systemctl start postgresql@<%= config.version %>-main'
+service_stop_command='sudo systemctl stop postgresql@<%= config.version %>-main'
+service_restart_command='sudo systemctl restart postgresql@<%= config.version %>-main'
+service_reload_command='sudo systemctl reload postgresql@<%= config.version %>-main'
+
+ssh_options='-i /var/lib/postgresql/.ssh/active_postgres_dns -o StrictHostKeyChecking=no'
+<% if repmgr_config.key?(:use_rewind) %>
+use_rewind=<%= repmgr_config[:use_rewind] ? 'yes' : 'no' %>
+<% end %>
+
 log_level=INFO
 log_facility=STDERR
 log_file='/var/log/postgresql/repmgr.log'

data/templates/repmgr_dns_failover.sh.erb

@@ -6,30 +6,44 @@ DNS_USER="<%= dns_user %>"
 DNS_SERVERS=(<%= dns_servers.map(&:to_s).join(' ') %>)
 DNS_SSH_KEY="<%= dns_ssh_key_path %>"
 DNSMASQ_FILE="/etc/dnsmasq.d/active_postgres.conf"
-PRIMARY_RECORD="<%= primary_record %>"
-REPLICA_RECORD="<%= replica_record %>"
+PRIMARY_RECORDS=(<%= primary_records.map(&:to_s).join(' ') %>)
+REPLICA_RECORDS=(<%= replica_records.map(&:to_s).join(' ') %>)
 SSH_STRICT_HOST_KEY="<%= ssh_strict_host_key %>"
 SSH_KNOWN_HOSTS="/var/lib/postgresql/.ssh/known_hosts"
 LOG_TAG="active_postgres_dns"
 
-cluster_csv=$(repmgr -f "$REPMGR_CONF" cluster show --csv 2>/dev/null || true)
-if [[ -z "$cluster_csv" ]]; then
+REPMGR_DB=$(grep -oP "dbname=\K[^ ']+" "$REPMGR_CONF" | head -1)
+REPMGR_USER=$(grep -oP "user=\K[^ ']+" "$REPMGR_CONF" 2>/dev/null | head -1)
+REPMGR_USER="${REPMGR_USER:-repmgr}"
+
+cluster_data=$(sudo -u postgres psql -h /var/run/postgresql -d "$REPMGR_DB" -tAF',' -c "SELECT type, conninfo FROM repmgr.nodes WHERE active = true" 2>/dev/null || true)
+if [[ -z "$cluster_data" ]]; then
   exit 0
 fi
 
-primary_host=$(printf "%s\n" "$cluster_csv" | awk -F',' 'NR>1 && tolower($3) ~ /primary/ {print $NF; exit}' | sed -n 's/.*host=\\([^ ]*\\).*/\\1/p')
-standby_hosts=$(printf "%s\n" "$cluster_csv" | awk -F',' 'NR>1 && tolower($3) ~ /standby/ {print $NF}' | sed -n 's/.*host=\\([^ ]*\\).*/\\1/p' | sort -u)
+primary_host=$(printf "%s\n" "$cluster_data" | awk -F',' '$1 == "primary" {print $2; exit}' | sed -n 's/.*host=\([^ ]*\).*/\1/p')
+standby_hosts=$(printf "%s\n" "$cluster_data" | awk -F',' '$1 == "standby" {print $2}' | sed -n 's/.*host=\([^ ]*\).*/\1/p' | sort -u)
 
 if [[ -z "$primary_host" ]]; then
   exit 0
 fi
 
 content=$'# Managed by active_postgres\n'
-printf -v content '%saddress=/%s/%s\n' "$content" "$PRIMARY_RECORD" "$primary_host"
+for record in "${PRIMARY_RECORDS[@]}"; do
+  if [[ -n "$record" ]]; then
+    printf -v content '%saddress=/%s/%s\n' "$content" "$record" "$primary_host"
+  fi
+done
 
 if [[ -n "$standby_hosts" ]]; then
-  for host in $standby_hosts; do
-    printf -v content '%saddress=/%s/%s\n' "$content" "$REPLICA_RECORD" "$host"
+  for record in "${REPLICA_RECORDS[@]}"; do
+    if [[ -z "$record" ]]; then
+      continue
+    fi
+
+    for host in $standby_hosts; do
+      printf -v content '%saddress=/%s/%s\n' "$content" "$record" "$host"
+    done
   done
 fi
 
@@ -41,9 +55,9 @@ for server in "${DNS_SERVERS[@]}"; do
   fi
 
   if ! /usr/bin/ssh "${ssh_opts[@]}" "${DNS_USER}@${server}" \
-    "sudo bash -c 'cat > ${DNSMASQ_FILE} << \"EOF\"\\n${content}EOF\\n' && (sudo systemctl reload dnsmasq || sudo systemctl restart dnsmasq)"; then
+    "sudo bash -c 'cat > ${DNSMASQ_FILE} << \"EOF\"\\n${content}EOF\\n' && sudo systemctl restart dnsmasq"; then
     /usr/bin/logger -t "$LOG_TAG" "Failed updating dnsmasq on ${server}"
   fi
 done
 
-/usr/bin/logger -t "$LOG_TAG" "Updated DNS records for ${PRIMARY_RECORD} and ${REPLICA_RECORD} (primary=${primary_host})"
+/usr/bin/logger -t "$LOG_TAG" "Updated DNS records for ${PRIMARY_RECORDS[*]} and ${REPLICA_RECORDS[*]} (primary=${primary_host})"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: active_postgres
 version: !ruby/object:Gem::Version
-  version: 0.9.0
+  version: 0.9.2
 platform: ruby
 authors:
 - BoringCache
@@ -179,6 +179,7 @@ files:
 - lib/active_postgres/installer.rb
 - lib/active_postgres/log_sanitizer.rb
 - lib/active_postgres/logger.rb
+- lib/active_postgres/overview.rb
 - lib/active_postgres/performance_tuner.rb
 - lib/active_postgres/rails/database_config.rb
 - lib/active_postgres/rails/migration_guard.rb