1
0
Fork 0
mirror of https://github.com/puma/puma.git synced 2022-11-09 13:48:40 -05:00
puma--puma/bin/mongrel_rails

225 lines
5.9 KiB
Text
Raw Normal View History

require 'rubygems'
require 'mongrel'
require 'cgi'
require 'daemons/daemonize'
require 'mongrel/command'
class CGIFixed < ::CGI
public :env_table
attr_reader :options
def initialize(params, data, out, *args)
@env_table = params
@args = *args
@input = StringIO.new(data)
@out = out
@options = {}
super(*args)
end
def header(options = "text/html")
if options.class == Hash
# passing in a header so need to keep the status around and other options
@options = options
end
super(options)
end
def status
s = @options["Status"] || @options["status"]
s[0 .. s.index(' ')] || "200"
end
def args
@args
end
def env_table
@env_table
end
def stdinput
@input
end
def stdoutput
@out
end
end
class RailsHandler < Mongrel::HttpHandler
def initialize(dir, mime_map = {})
@files = Mongrel::DirHandler.new(dir,false)
@guard = Mutex.new
# register the requested mime types
mime_map.each {|k,v| Mongrel::DirHandler::add_mime_type(k,v) }
end
def process(request, response)
# not static, need to talk to rails
return if response.socket.closed?
if @files.can_serve(request.params["PATH_INFO"])
@files.process(request,response)
else
cgi = CGIFixed.new(request.params, request.body, response.socket)
begin
@guard.synchronize do
# Rails is not thread safe so must be run entirely within synchronize
Dispatcher.dispatch(cgi, ActionController::CgiRequest::DEFAULT_SESSION_OPTIONS, response.body)
end
response.status = cgi.status
response.send_status
response.send_body
rescue Object => rails_error
STDERR.puts "calling Dispatcher.dispatch #{rails_error}"
STDERR.puts rails_error.backtrace.join("\n")
end
end
end
end
class StartCommand < Mongrel::Command::Command
def configure
options [
["-e", "--environment ENV", "Rails environment to run as", :@environment, ENV['RAILS_ENV'] || "development"],
["-d", "--daemonize", "Whether to run in the background or not", :@daemon, false],
['-p', '--port PORT', "Which port to bind to", :@port, 3000],
['-a', '--address ADDR', "Address to bind to", :@address, "0.0.0.0"],
['-l', '--log FILE', "Where to write log messages", :@log_file, "log/mongrel.log"],
['-P', '--pid FILE', "Where to write the PID", :@pid_file, "log/mongrel.pid"],
['-n', '--num-procs INT', "Number of processor threads to use", :@num_procs, 20],
['-t', '--timeout SECONDS', "Timeout all requests after SECONDS time", :@timeout, 120],
['-m', '--mime PATH', "A YAML file that lists additional MIME types", :@mime_map, nil],
['-c', '--chdir PATH', "Change to dir before starting (will be expanded)", :@cwd, Dir.pwd],
['-r', '--root PATH', "Set the document root (default 'public')", :@docroot, "public"],
]
end
def validate
@cwd = File.expand_path(@cwd)
valid_dir? @cwd, "Invalid path to change to during daemon mode: #@cwd"
# change there to start, then we'll have to come back after daemonize
Dir.chdir(@cwd)
valid_dir? File.dirname(@log_file), "Path to log file not valid: #@log_file"
valid_dir? File.dirname(@pid_file), "Path to pid file not valid: #@pid_file"
valid_dir? @docroot, "Path to docroot not valid: #@docroot"
valid_exists? @mime_map, "MIME mapping file does not exist: #@mime_map" if @mime_map
return @valid
end
def daemonize
# save this for later since daemonize will hose it
if @daemon
puts "Started Mongrel server in #@environment mode at #@address:#@port"
Daemonize.daemonize(log_file=File.join(@cwd, @log_file))
# change back to the original starting directory
Dir.chdir(@cwd)
open(@pid_file,"w") {|f| f.write(Process.pid) }
else
puts "Running Mongrel server in #@environment mode at #@address:#@port"
end
end
def load_mime_map
mime = {}
# configure any requested mime map
if @mime_map
puts "Loading additional MIME types from #@mime_map"
mime.merge!(YAML.load_file(@mime_map))
# check all the mime types to make sure they are the right format
mime.each {|k,v| puts "WARNING: MIME type #{k} must start with '.'" if k.index(".") != 0 }
end
return mime
end
def configure_rails
ENV['RAILS_ENV'] = @environment
require 'config/environment'
# configure the rails handler
rails = RailsHandler.new(@docroot, load_mime_map)
return rails
end
def start_mongrel(rails)
# start up mongrel with the right configurations
server = Mongrel::HttpServer.new(@address, @port, @num_procs.to_i, @timeout.to_i)
server.register("/", rails)
server.run
begin
puts "Server ready."
server.acceptor.join
rescue Interrupt
puts "Interrupted."
end
end
def run
daemonize
rails = configure_rails
start_mongrel(rails)
end
end
class StopCommand < Mongrel::Command::Command
def configure
options [
['-c', '--chdir PATH', "Change to dir before starting (will be expanded)", :@cwd, Dir.pwd],
['-P', '--pid FILE', "Where to write the PID", :@pid_file, "log/mongrel.pid"]
]
end
def validate
@cwd = File.expand_path(@cwd)
valid_dir? @cwd, "Invalid path to change to during daemon mode: #@cwd"
@pid_file = File.join(@cwd,@pid_file)
valid_exists? @pid_file, "PID file #@pid_file does not exist. Not running?"
return @valid
end
def run
pid = open(@pid_file).read.to_i
print "Stopping Mongrel at PID #{pid}..."
begin
Process.kill("INT", pid)
rescue Errno::ESRCH
puts "Process does not exist. Not running."
end
File.unlink(@pid_file)
puts "Done."
end
end
Mongrel::Command::Registry.instance.run ARGV