blob: 684f04db86abc364b78af18b1dd7f02f32a79f61 (
plain) (
tree)
|
|
#!/usr/local/bin/ruby
# to allow unit testing
if !defined?(RAILS_ROOT)
require File.dirname(__FILE__) + "/../config/environment"
end
require 'dispatcher'
require 'fcgi'
require 'logger'
class RailsFCGIHandler
attr_reader :please_exit_at_your_earliest_convenience
attr_reader :i_am_currently_processing_a_request
def initialize(log_file_path = "#{RAILS_ROOT}/log/fastcgi.crash.log")
@please_exit_at_your_earliest_convenience = false
@i_am_currently_processing_a_request = false
trap_handler = method(:trap_handler).to_proc
trap("HUP", trap_handler)
trap("USR1", trap_handler)
# initialize to 11 seconds from now to minimize special cases
@last_error_on = Time.now - 11
@log_file_path = log_file_path
dispatcher_log(:info, "fcgi #{$$} starting")
end
def process!
FCGI.each_cgi do |cgi|
process_request(cgi)
break if please_exit_at_your_earliest_convenience
end
dispatcher_log(:info, "fcgi #{$$} terminated gracefully")
rescue SystemExit => exit_error
dispatcher_log(:info, "fcgi #{$$} terminated by explicit exit")
rescue Object => fcgi_error
# retry on errors that would otherwise have terminated the FCGI process,
# but only if they occur more than 10 seconds apart.
if !(SignalException === fcgi_error) && Time.now - @last_error_on > 10
@last_error_on = Time.now
dispatcher_error(fcgi_error,
"FCGI process #{$$} almost killed by this error\n")
retry
else
dispatcher_error(fcgi_error, "FCGI process #{$$} killed by this error\n")
end
end
private
def logger
@logger ||= Logger.new(@log_file_path)
end
def dispatcher_log(level, msg)
logger.send(level, msg)
rescue Object => log_error
STDERR << "Couldn't write to #{@log_file_path.inspect}: #{msg}\n"
STDERR << " #{log_error.class}: #{log_error.message}\n"
end
def dispatcher_error(e,msg="")
error_message =
"[#{Time.now}] Dispatcher failed to catch: #{e} (#{e.class})\n" +
" #{e.backtrace.join("\n ")}\n#{msg}"
dispatcher_log(:error, error_message)
end
def trap_handler(signal)
if i_am_currently_processing_a_request
dispatcher_log(:info, "asking #{$$} to terminate ASAP")
@please_exit_at_your_earliest_convenience = true
else
dispatcher_log(:info, "telling #{$$} to terminate NOW")
exit
end
end
def process_request(cgi)
@i_am_currently_processing_a_request = true
Dispatcher.dispatch(cgi)
rescue Object => e
raise if SignalException === e
dispatcher_error(e)
ensure
@i_am_currently_processing_a_request = false
end
end
if __FILE__ == $0
handler = RailsFCGIHandler.new
handler.process!
end
|