worker_killer 1.0.4.189871 → 1.0.5.213889

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1481f9bea1f77d274c4155a94bd83c6527ab72eb53fe6958b8036985d714720c
-  data.tar.gz: 155090caa35e210c9e130f395b00d4b83eab0ec7dcc3aba1a4419e783410c442
+  metadata.gz: 1d0558051a55e944e97bb26729239a97d7f150fb37885af8761246f78db4e280
+  data.tar.gz: 417bb8c4f0c4818d4e2a19ad4f2cb8cc5e3ca82621b925c81c2e4b33102803fb
 SHA512:
-  metadata.gz: 3b3d7dbef3dc17f218341dace5fca0a1f4c92263ef104b1c1d7f9d1a5f659650e21578f00844f55cd073228117b17744cb5526c388936e90a862e215d9c9f497
-  data.tar.gz: 7cd8957b9e57ee14f6ae13ff03dbec12369fc45e82f97d4f93b3f378606460254110c37209f7afb53cd2d053a81ccc2a021089386e15dd83bee7a174783e0683
+  metadata.gz: 855f6852cb2c7c8115388038f07d234958a6183c008d21b28ad8867a34a53dcfd713f34645c1b0f435fafc67f5f7ad0c1272b7b4f94f2d91b170ca91c0f97133
+  data.tar.gz: 8699bc63da513186e47cb1ef626f732e61cfe7a603231370223cbcea5ec3b6d8897a314ed5fe8dee9a23765822b4c20aa43ddc6fda7b02bb775f01aa1f838a7e

data/README.md

@@ -17,6 +17,7 @@ Features:
 
 - generic middleware implementation
 - Phusion Passenger support(through `passenger-config detach-process <PID>`)
+- Puma phased-restart support(through `pumactl phased-restart`)
 - DelayedJob support
 - custom reaction hook
 
@@ -42,14 +43,14 @@ Add these lines to your `config.ru` or `application.rb`. (These lines should be
 
   # Max requests per worker
   middleware.insert_before(
-    Rack::Sendfile,
+    Rack::Runtime,
     WorkerKiller::Middleware::RequestsLimiter, killer: killer, min: 3072, max: 4096
   )
 
   # Max memory size (RSS) per worker
   middleware.insert_before(
-    Rack::Sendfile,
-    WorkerKiller::Middleware::OOMLimiter, killer: killer, min: 500 * (1024**2), max: 600 * (1024**2)
+    Rack::Runtime,
+    WorkerKiller::Middleware::OOMLimiter, killer: killer, min: nil, max: 0.5, check_cycle: 16
   )
 ```
 
@@ -74,6 +75,39 @@ Add these lines to your `initializers/delayed_job.rb` or `application.rb`.
   end
 ```
 
+## Puma Web-server
+
+Add these lines to your `puma.rb` AND `application.rb`.
+
+```ruby
+  # puma.rb
+
+  on_worker_boot do |*args|
+    require 'worker_killer/killer'
+    require 'active_support/notifications'
+
+    ::ActiveSupport::Notifications.subscribe 'initialize.custom.rails' do |*eargs|
+      event = ActiveSupport::Notifications::Event.new(*eargs)
+      config = event.payload[:config]
+
+      killer = ::WorkerKiller::Killer::Puma.new
+      config.middleware.insert_before(
+        ::Rack::Runtime,
+        ::WorkerKiller::Middleware::RequestsLimiter, killer: killer, min: 3072, max: 4096
+      )
+
+      config.middleware.insert_before(
+        ::Rack::Runtime,
+        ::WorkerKiller::Middleware::OOMLimiter, killer: killer, min: nil, max: 0.5, verbose: true, check_cycle: 16
+      )
+    end
+  end
+
+  # application.rb
+
+  ActiveSupport::Notifications.instrument 'initialize.custom.rails', config: config
+```
+
 This gem provides two modules: WorkerKiller::CountLimiter and WorkerKiller::MemoryLimiter, some Rack integration and DelayedJob plugin.
 
 ### WorkerKiller::Middleware::RequestsLimiter and WorkerKiller::DelayedJobPlugin::JobsLimiter

data/lib/worker_killer/killer/puma.rb

@@ -0,0 +1,33 @@
+module ::WorkerKiller
+  module Killer
+    class Puma < ::WorkerKiller::Killer::Base
+
+      def initialize **kwrags
+        super
+      end
+
+      def do_kill(sig, pid, alive_sec, **_params)
+        cmd = 'pumactl phased-restart'
+
+        if sig == :KILL
+          logger.error "#{self} force to kill self (pid: #{pid}) alive: #{alive_sec} sec (trial #{kill_attempts})"
+          Process.kill sig, pid
+          return
+        end
+
+        return if @already_detached
+
+        logger.warn "#{self} run #{cmd.inspect} (pid: #{pid}) alive: #{alive_sec} sec (trial #{kill_attempts})"
+        @already_detached = true
+
+        Thread.new(cmd) do |command|
+          unless Kernel.system(command)
+            logger.warn "#{self} run #{command.inspect} failed: #{$?.inspect}"
+          end
+        end
+      end
+
+    end
+  end
+end
+

data/lib/worker_killer/killer.rb

@@ -38,5 +38,6 @@ end
 
 require_relative 'killer/signal'
 require_relative 'killer/passenger'
+require_relative 'killer/puma'
 require_relative 'killer/delayed_job'
 

data/lib/worker_killer/version.rb

@@ -2,7 +2,7 @@
 
 module WorkerKiller
 
-  VERSION = '1.0.4'
+  VERSION = '1.0.5'
 
 end
 

data/spec/killer/puma_spec.rb

@@ -0,0 +1,33 @@
+RSpec.describe WorkerKiller::Killer::Puma do
+  let(:config) do
+    WorkerKiller::Configuration.new.tap do |c|
+      c.quit_attempts = 2
+      c.term_attempts = 2
+    end
+  end
+
+  let(:killer){ described_class.new }
+
+  describe '#kill' do
+    around do |example|
+      prev = WorkerKiller.configuration
+      WorkerKiller.configuration = config
+      example.run
+    ensure
+      WorkerKiller.configuration = prev
+    end
+
+    it 'expect right signal order' do
+      expect(Kernel).to receive(:system).with('pumactl phased-restart').and_return(true)
+      expect(Process).to receive(:kill).with(:KILL, anything).exactly(5).times
+
+      thread = killer.kill(Time.now)
+      thread.join
+
+      1.times { killer.kill(Time.now) } # 1 QUIT
+      2.times { killer.kill(Time.now) } # 1 TERM
+      5.times { killer.kill(Time.now) } # 5 KILL
+    end
+  end
+end
+

data/spec/killer_spec.rb

@@ -19,7 +19,7 @@ RSpec.describe WorkerKiller::Killer::Base do
       end
 
       it 'expect right signal order' do
-        if RUBY_VERSION >= "3.0.0"
+        if RUBY_VERSION >= "2.7.0"
           expect(killer).to receive(:do_kill).with(:QUIT, anything, anything).exactly(2).times
           expect(killer).to receive(:do_kill).with(:TERM, anything, anything).exactly(2).times
           expect(killer).to receive(:do_kill).with(:KILL, anything, anything).exactly(5).times

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: worker_killer
 version: !ruby/object:Gem::Version
-  version: 1.0.4.189871
+  version: 1.0.5.213889
 platform: ruby
 authors:
 - Samoilenko Yuri
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-10-05 00:00:00.000000000 Z
+date: 2024-02-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: get_process_mem
@@ -131,6 +131,7 @@ files:
 - lib/worker_killer/killer.rb
 - lib/worker_killer/killer/delayed_job.rb
 - lib/worker_killer/killer/passenger.rb
+- lib/worker_killer/killer/puma.rb
 - lib/worker_killer/killer/signal.rb
 - lib/worker_killer/memory_limiter.rb
 - lib/worker_killer/middleware.rb
@@ -140,6 +141,7 @@ files:
 - spec/delayed_job_plugin/oom_limiter_spec.rb
 - spec/killer/delayed_job_spec.rb
 - spec/killer/passenger_spec.rb
+- spec/killer/puma_spec.rb
 - spec/killer/signal_spec.rb
 - spec/killer_spec.rb
 - spec/memory_limiter_spec.rb
@@ -171,15 +173,16 @@ signing_key:
 specification_version: 4
 summary: Kill any workers by memory and request counts or take custom reaction
 test_files:
-- spec/middleware_spec.rb
+- spec/killer/signal_spec.rb
+- spec/killer/puma_spec.rb
+- spec/killer/delayed_job_spec.rb
+- spec/killer/passenger_spec.rb
 - spec/killer_spec.rb
+- spec/count_limiter_spec.rb
+- spec/worker_killer_spec.rb
+- spec/support/logger.rb
+- spec/middleware_spec.rb
 - spec/spec_helper.rb
 - spec/delayed_job_plugin/oom_limiter_spec.rb
 - spec/delayed_job_plugin/jobs_limiter_spec.rb
-- spec/killer/passenger_spec.rb
-- spec/killer/signal_spec.rb
-- spec/killer/delayed_job_spec.rb
 - spec/memory_limiter_spec.rb
-- spec/support/logger.rb
-- spec/worker_killer_spec.rb
-- spec/count_limiter_spec.rb