process_bot 0.1.17 → 0.1.18

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e0a67de01a927a07fc17e06814e4752b2da954eca493143f7850b18e5a5b1aa9
-  data.tar.gz: f0fc56095299f93a43a45549c9399118d4ddf7e5e112ce701baac8b86e1b3756
+  metadata.gz: 00371cf3c9ed3fa904d5d3a8655be99178ce77a729ab2aad459572fc63bb4779
+  data.tar.gz: db9145067629c7872c810c03e1e97caa6c626837ec26d07fa6ceeebf1a623bdd
 SHA512:
-  metadata.gz: 5cad374bb6dacaa0c2a87e7b1f0b1a7add878c8466a6d54de5836ed285f0ec7a4fde8f2a9dcac94ebc8432992dc92632953f587ab992883c2b3b5aadcb1497ee
-  data.tar.gz: d597eaac6ecbaa04c6926018f3bf3872438c7138f2a64bdb52691bf2b1183c0854f34a05f5681c576a648fd69c21c90ea6395bf30c7700868dd9fecfcb332158
+  metadata.gz: 89a0dde6cbadf510efb51b07f690ec90ca03af46ba06c9a1eca1f75a86e718bf23d263207aedd5a2763463e9470d9655d81487d5073f24a077edd1798acb2691
+  data.tar.gz: 27a95d5f8911f704c15faf58339c0173ca6cc01a3d7e8caa70743d6f3eadbbb9a44122b2ece7e74feaf4b7b6e7a03b1d7809e22cfa26efb4eb9967972f6846ff

data/CHANGELOG.md

@@ -1,4 +1,5 @@
 ## [Unreleased]
+- Stop accepting new control commands during shutdown so in-flight responses complete reliably.
 
 ## [0.1.0] - 2022-04-03
 

data/Gemfile.lock

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

data/lib/process_bot/control_socket.rb

@@ -79,13 +79,20 @@ class ProcessBot::ControlSocket
 
       if command_type == "graceful" || command_type == "graceful_no_wait" || command_type == "stop"
         begin
+          unless process.accept_control_commands?
+            client.puts(JSON.generate(type: "error", message: "ProcessBot is shutting down", backtrace: Thread.current.backtrace))
+            break
+          end
+
           command_options = if command["options"]
             symbolize_keys(command.fetch("options"))
           else
             {}
           end
 
-          run_command(command_type, command_options, client)
+          process.with_control_command do
+            run_command(command_type, command_options, client)
+          end
         rescue => e # rubocop:disable Style/RescueStandardError
           logger.error e.message
           logger.error e.backtrace

data/lib/process_bot/process.rb

@@ -1,5 +1,6 @@
 require "forwardable"
 require "json"
+require "monitor"
 require "string-cases"
 
 class ProcessBot::Process
@@ -12,11 +13,14 @@ class ProcessBot::Process
   autoload :Handlers, "#{__dir__}/process/handlers"
   autoload :Runner, "#{__dir__}/process/runner"
 
-  attr_reader :current_pid, :current_process_title, :options, :port, :stopped
+  attr_reader :control_command_monitor, :current_pid, :current_process_title, :options, :port, :stopped
 
   def initialize(options)
     @options = options
     @stopped = false
+    @accept_control_commands = true
+    @control_command_monitor = Monitor.new
+    @control_commands_in_flight = 0
 
     options.events.connect(:on_process_started, &method(:on_process_started)) # rubocop:disable Performance/MethodObjectAsBlock
     options.events.connect(:on_socket_opened, &method(:on_socket_opened)) # rubocop:disable Performance/MethodObjectAsBlock
@@ -86,6 +90,8 @@ class ProcessBot::Process
       run
 
       if stopped
+        stop_accepting_control_commands
+        wait_for_control_commands
         break
       else
         logger.logs "Process stopped - starting again after 1 sec"
@@ -127,4 +133,35 @@ class ProcessBot::Process
     @current_process_title = "ProcessBot #{JSON.generate(process_args)}"
     Process.setproctitle(current_process_title)
   end
+
+  def with_control_command
+    control_command_monitor.synchronize do
+      @control_commands_in_flight += 1
+    end
+
+    yield
+  ensure
+    control_command_monitor.synchronize do
+      @control_commands_in_flight -= 1
+    end
+  end
+
+  def accept_control_commands?
+    @accept_control_commands
+  end
+
+  def stop_accepting_control_commands
+    @accept_control_commands = false
+    @control_socket&.stop
+  end
+
+  def wait_for_control_commands
+    sleep 0.1 while control_commands_in_flight.positive?
+  end
+
+  def control_commands_in_flight
+    control_command_monitor.synchronize do
+      @control_commands_in_flight
+    end
+  end
 end

data/lib/process_bot/version.rb

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

metadata

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