process_bot 0.1.15 → 0.1.17

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d3abcda452c729ce0582b53ff117aa7ec8658d16ad52f05f7ff1be5e383e2422
-  data.tar.gz: 7499b97fa9e2ed082b739b241549baa4e511a5d0db0f024e921b4d174d842c20
+  metadata.gz: e0a67de01a927a07fc17e06814e4752b2da954eca493143f7850b18e5a5b1aa9
+  data.tar.gz: f0fc56095299f93a43a45549c9399118d4ddf7e5e112ce701baac8b86e1b3756
 SHA512:
-  metadata.gz: 3680d58298af7cb511b3e5007abfafadfd3457b22230936e36f0925d6f8245b07de3116356f28b6beb18bf5cb36b86604b5bfeb323aa357adb2313d9ac20d733
-  data.tar.gz: 4bbbafc629c0b3af2f47f1c82d3aab2bceeb003dd7820df85c5764427e654e96f08e232ee4e237f1df12e0c9a28fd807b0e9902ea1520c673902c79a95b0b3c8
+  metadata.gz: 5cad374bb6dacaa0c2a87e7b1f0b1a7add878c8466a6d54de5836ed285f0ec7a4fde8f2a9dcac94ebc8432992dc92632953f587ab992883c2b3b5aadcb1497ee
+  data.tar.gz: d597eaac6ecbaa04c6926018f3bf3872438c7138f2a64bdb52691bf2b1183c0854f34a05f5681c576a648fd69c21c90ea6395bf30c7700868dd9fecfcb332158

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    process_bot (0.1.15)
+    process_bot (0.1.17)
       knjrbfw (>= 0.0.116)
       pry
       rake

data/README.md

@@ -44,6 +44,7 @@ ProcessBot provides these Sidekiq tasks:
 - `process_bot:sidekiq:stop`
 - `process_bot:sidekiq:graceful` (stops fetching new jobs and waits for running jobs by default)
 - `process_bot:sidekiq:graceful_no_wait` (stops fetching new jobs and returns immediately)
+- `process_bot:sidekiq:ensure_running` (starts missing processes, including replacements for graceful shutdowns)
 - `process_bot:sidekiq:restart`
 
 You can also skip waiting for graceful completion:

data/lib/process_bot/capistrano/sidekiq.rake

@@ -70,6 +70,52 @@ namespace :process_bot do
       end
     end
 
+    desc "Ensure the configured number of Sidekiq ProcessBots are running (starts replacements for graceful shutdowns)"
+    task :ensure_running do
+      on roles fetch(:sidekiq_roles) do |role|
+        git_plugin.switch_user(role) do
+          desired_processes = fetch(:sidekiq_processes).to_i
+          running_processes = git_plugin.running_process_bot_processes
+
+          graceful_processes = running_processes.select do |process_bot_data|
+            git_plugin.sidekiq_process_graceful?(process_bot_data)
+          end
+
+          active_processes = running_processes - graceful_processes
+
+          active_indexes = active_processes.filter_map do |process_bot_data|
+            git_plugin.process_bot_sidekiq_index(process_bot_data)
+          end
+
+          graceful_indexes = graceful_processes.filter_map do |process_bot_data|
+            git_plugin.process_bot_sidekiq_index(process_bot_data)
+          end
+
+          if graceful_indexes.any?
+            puts "ProcessBot Sidekiq in graceful shutdown: #{graceful_indexes.join(", ")}"
+            puts "Starting replacements for graceful shutdowns to reach #{desired_processes} running processes"
+          end
+
+          missing_indexes = git_plugin.missing_sidekiq_indexes(desired_processes, active_indexes)
+          missing_count = desired_processes - active_indexes.count
+
+          if missing_count.negative?
+            puts "Found #{active_indexes.count} running ProcessBot Sidekiq processes; desired is #{desired_processes}"
+            missing_count = 0
+          end
+
+          if missing_indexes.any?
+            missing_indexes.each do |idx|
+              puts "Starting Sidekiq with ProcessBot #{idx} (missing)"
+              git_plugin.start_sidekiq(idx)
+            end
+          elsif missing_count.zero?
+            puts "All ProcessBot Sidekiq processes are running (#{active_indexes.count}/#{desired_processes})"
+          end
+        end
+      end
+    end
+
     desc "Restart Sidekiq and ProcessBot"
     task :restart do
       invoke! "process_bot:sidekiq:stop"

data/lib/process_bot/capistrano/sidekiq_helpers.rb

@@ -93,6 +93,37 @@ module ProcessBot::Capistrano::SidekiqHelpers # rubocop:disable Metrics/ModuleLe
     processes
   end
 
+  def process_bot_sidekiq_index(process_bot_data)
+    process_bot_id = process_bot_data["id"].to_s
+    match = process_bot_id.match(/-(\d+)\z/)
+    return nil unless match
+
+    match[1].to_i
+  end
+
+  def sidekiq_command_graceful?(command)
+    return false unless command
+
+    normalized_command = command.to_s.downcase
+    normalized_command.include?("stopping") || normalized_command.include?("quiet")
+  end
+
+  def sidekiq_process_graceful?(process_bot_data)
+    sidekiq_pid = process_bot_data["pid"]
+    return false unless sidekiq_pid
+
+    command = backend.capture(:ps, "-o", "command=", "-p", sidekiq_pid.to_s).strip
+    sidekiq_command_graceful?(command)
+  rescue SSHKit::Command::Failed
+    false
+  end
+
+  def missing_sidekiq_indexes(desired_processes, active_indexes)
+    desired = desired_processes.to_i
+    desired_indexes = (0...desired).to_a
+    desired_indexes - active_indexes
+  end
+
   def sidekiq_user(role = nil)
     if role.nil?
       fetch(:sidekiq_user)

data/lib/process_bot/process/handlers/sidekiq.rb

@@ -215,5 +215,15 @@ class ProcessBot::Process::Handlers::Sidekiq
     logger.logs "Wait for no jobs and Stop sidekiq"
     wait_for_no_jobs
     stop
+    wait_for_sidekiq_exit
+  end
+
+  def wait_for_sidekiq_exit
+    return unless current_pid
+
+    while process_running?(current_pid)
+      logger.logs "Waiting for Sidekiq PID #{current_pid} to stop"
+      sleep 1
+    end
   end
 end

data/lib/process_bot/process.rb

@@ -106,7 +106,8 @@ class ProcessBot::Process
 
   def send_control_command(command, **command_options)
     logger.logs "Sending #{command} command"
-    client.send_command(command: command, options: options.options.merge(command_options))
+    response = client.send_command(command: command, options: options.options.merge(command_options))
+    raise "No response from ProcessBot while sending #{command}" if response == :nil
   rescue Errno::ECONNREFUSED => e
     raise e unless options[:ignore_no_process_bot]
   end

data/lib/process_bot/version.rb

@@ -1,3 +1,3 @@
 module ProcessBot
-  VERSION = "0.1.15".freeze
+  VERSION = "0.1.17".freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: process_bot
 version: !ruby/object:Gem::Version
-  version: 0.1.15
+  version: 0.1.17
 platform: ruby
 authors:
 - kaspernj
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2026-01-05 00:00:00.000000000 Z
+date: 2026-01-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: knjrbfw