Class: PumaWorkerKiller::Reaper

Inherits:
Object
  • Object
show all
Defined in:
lib/puma_worker_killer/reaper.rb

Instance Method Summary collapse

Constructor Details

#initialize(max_ram, master = nil, reaper_status_logs = true, pre_term = nil, on_calculation = nil) ⇒ Reaper

Returns a new instance of Reaper.



5
6
7
8
9
10
11
 
# File 'lib/puma_worker_killer/reaper.rb', line 5

def initialize(max_ram, master = nil, reaper_status_logs = true, pre_term = nil, on_calculation = nil)
  @cluster = PumaWorkerKiller::PumaMemory.new(master)
  @max_ram = max_ram
  @reaper_status_logs = reaper_status_logs
  @pre_term = pre_term
  @on_calculation = on_calculation
end

Instance Method Details

#get_total_memory ⇒ Object

used for tes



14
15
16
 
# File 'lib/puma_worker_killer/reaper.rb', line 14

def get_total_memory
  @cluster.get_total_memory
end

#reap ⇒ Object 



18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
 
# File 'lib/puma_worker_killer/reaper.rb', line 18

def reap
  return false if @cluster.workers_stopped?

  total = get_total_memory
  @on_calculation&.call(total)

  if total > @max_ram
    @cluster.master.log "PumaWorkerKiller: Out of memory. #{@cluster.workers.count} workers consuming total: #{total} mb out of max: #{@max_ram} mb. Sending TERM to pid #{@cluster.largest_worker.pid} consuming #{@cluster.largest_worker_memory} mb."

    # Fetch the largest_worker so that both `@pre_term` and `term_worker` are called with the same worker
    # Avoids a race condition where:
    #   Worker A consume 100 mb memory
    #   Worker B consume 99 mb memory
    #   pre_term gets called with Worker A
    #   A new request comes in, Worker B takes it, and consumes 101 mb memory
    #   term_largest_worker (previously here) gets called and terms Worker B (thus not passing the about-to-be-terminated worker to `@pre_term`)
    largest_worker = @cluster.largest_worker
    @pre_term&.call(largest_worker)
    @cluster.term_worker(largest_worker)

  elsif @reaper_status_logs
    @cluster.master.log "PumaWorkerKiller: Consuming #{total} mb with master and #{@cluster.workers.count} workers."
  end
end