Add rolling_pre_term config

lib/puma_worker_killer.rb

@@ -3,13 +3,16 @@
 module PumaWorkerKiller
   extend self
 
-  attr_accessor :ram, :frequency, :percent_usage, :rolling_restart_frequency, :reaper_status_logs, :pre_term, :on_calculation
+  attr_accessor :ram, :frequency, :percent_usage, :rolling_restart_frequency,
+                :reaper_status_logs, :pre_term, :rolling_pre_term, :on_calculation
+
   self.ram           = 512  # mb
   self.frequency     = 10   # seconds
   self.percent_usage = 0.99 # percent of RAM to use
   self.rolling_restart_frequency = 6 * 3600 # 6 hours in seconds
   self.reaper_status_logs = true
   self.pre_term = nil
+  self.rolling_pre_term = nil
   self.on_calculation = nil
 
   def config
@@ -27,7 +30,7 @@ def start(frequency = self.frequency, reaper = self.reaper)
 
   def enable_rolling_restart(frequency = self.rolling_restart_frequency)
     frequency = frequency + rand(0..10.0) # so all workers don't restart at the exact same time across multiple machines
-    AutoReap.new(frequency, RollingRestart.new).start
+    AutoReap.new(frequency, RollingRestart.new(nil, self.rolling_pre_term)).start
   end
 end
 

lib/puma_worker_killer/rolling_restart.rb

@@ -1,7 +1,8 @@
 module PumaWorkerKiller
   class RollingRestart
-    def initialize(master = nil)
+    def initialize(master = nil, rolling_pre_term = nil)
       @cluster = PumaWorkerKiller::PumaMemory.new(master)
+      @rolling_pre_term = rolling_pre_term
     end
 
     # used for tes
@@ -16,6 +17,7 @@ def reap(wait_between_worker_kill = 60) # seconds
 
       @cluster.workers.each do |worker, ram|
         @cluster.master.log "PumaWorkerKiller: Rolling Restart. #{@cluster.workers.count} workers consuming total: #{ total_memory } mb. Sending TERM to pid #{worker.pid}."
+        @rolling_pre_term.call(worker) unless @rolling_pre_term.nil?
         worker.term
         sleep wait_between_worker_kill
       end