X-Git-Url: https://git.openstreetmap.org/rails.git/blobdiff_plain/56ddd1a21fabf95f7801a3a994ec22a9029c53c8..ef7f3d800cbdd49b692df10d312e5fd880e2e938:/config/initializers/memory_limits.rb diff --git a/config/initializers/memory_limits.rb b/config/initializers/memory_limits.rb index ce518269f..199bda5c6 100644 --- a/config/initializers/memory_limits.rb +++ b/config/initializers/memory_limits.rb @@ -1,39 +1,41 @@ # Setup any specified hard limit on the virtual size of the process -if defined?(HARD_MEMORY_LIMIT) and defined?(PhusionPassenger) and Process.const_defined?(:RLIMIT_AS) - Process.setrlimit Process::RLIMIT_AS, HARD_MEMORY_LIMIT*1024*1024, Process::RLIM_INFINITY +if defined?(HARD_MEMORY_LIMIT) && defined?(PhusionPassenger) && Process.const_defined?(:RLIMIT_AS) + Process.setrlimit Process::RLIMIT_AS, HARD_MEMORY_LIMIT * 1024 * 1024, Process::RLIM_INFINITY end # If we're running under passenger and a soft memory limit is # configured then setup some rack middleware to police the limit -if defined?(SOFT_MEMORY_LIMIT) and defined?(PhusionPassenger) +if defined?(SOFT_MEMORY_LIMIT) && defined?(PhusionPassenger) # Define some rack middleware to police the soft memory limit class MemoryLimit def initialize(app) @app = app end - + def call(env) # Process this requst status, headers, body = @app.call(env) - + # Restart if we've hit our memory limit if resident_size > SOFT_MEMORY_LIMIT Process.kill("USR1", Process.pid) end - + # Return the result of this request [status, headers, body] end - private + + private + def resident_size # Read statm to get process sizes. Format is # Size RSS Shared Text Lib Data fields = File.open("/proc/self/statm") do |file| fields = file.gets.split(" ") end - + # Return resident size in megabytes - return fields[1].to_i / 256 + fields[1].to_i / 256 end end