Module: Unicorn::WorkerKiller::Oom
- Defined in:
- lib/unicorn/worker_killer.rb
Class Method Summary collapse
-
.new(app, memory_limit_min = (1024**3), memory_limit_max = (2*(1024**3)), check_cycle = 16) ⇒ Object
Killing the process must be occurred at the outside of the request.
Instance Method Summary collapse
Class Method Details
.new(app, memory_limit_min = (1024**3), memory_limit_max = (2*(1024**3)), check_cycle = 16) ⇒ Object
Killing the process must be occurred at the outside of the request. We’re using similar techniques used by OobGC, to ensure actual killing doesn’t affect the request.
34 35 36 37 38 39 40 41 42 43 |
# File 'lib/unicorn/worker_killer.rb', line 34 def self.new(app, memory_limit_min = (1024**3), memory_limit_max = (2*(1024**3)), check_cycle = 16) ObjectSpace.each_object(Unicorn::HttpServer) do |s| s.extend(self) s.instance_variable_set(:@_worker_memory_limit_min, memory_limit_min) s.instance_variable_set(:@_worker_memory_limit_max, memory_limit_max) s.instance_variable_set(:@_worker_check_cycle, check_cycle) s.instance_variable_set(:@_worker_check_count, 0) end app # pretend to be Rack middleware since it was in the past end |
Instance Method Details
#process_client(client) ⇒ Object
49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
# File 'lib/unicorn/worker_killer.rb', line 49 def process_client(client) super(client) # Unicorn::HttpServer#process_client return if @_worker_memory_limit_min == 0 && @_worker_memory_limit_max == 0 @_worker_process_start ||= Time.now @_worker_memory_limit ||= @_worker_memory_limit_min + randomize(@_worker_memory_limit_max - @_worker_memory_limit_min + 1) @_worker_check_count += 1 if @_worker_check_count % @_worker_check_cycle == 0 rss = _worker_rss() if rss > @_worker_memory_limit logger.warn "#{self}: worker (pid: #{Process.pid}) exceeds memory limit (#{rss} bytes > #{@_worker_memory_limit} bytes)" Unicorn::WorkerKiller.kill_self(logger, @_worker_process_start) end @_worker_check_count = 0 end end |
#randomize(integer) ⇒ Object
45 46 47 |
# File 'lib/unicorn/worker_killer.rb', line 45 def randomize(integer) RUBY_VERSION > "1.9" ? Random.rand(integer) : rand(integer) end |