From 709ef400130a3a0eeb47271f4357b90c19a92792 Mon Sep 17 00:00:00 2001 From: zedshaw Date: Tue, 14 Feb 2006 12:19:02 +0000 Subject: Added requested features for running RoR applications. git-svn-id: svn+ssh://rubyforge.org/var/svn/mongrel/trunk@36 19e92222-5c0b-0410-8929-a290d50e31e9 --- bin/mongrel_rails | 103 +++++++++++++++++++++++++++++++++++++++++------------- 1 file changed, 78 insertions(+), 25 deletions(-) (limited to 'bin') diff --git a/bin/mongrel_rails b/bin/mongrel_rails index e94d855..f36c48f 100644 --- a/bin/mongrel_rails +++ b/bin/mongrel_rails @@ -51,9 +51,13 @@ end class RailsHandler < Mongrel::HttpHandler - def initialize(dir) + + 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) @@ -88,50 +92,94 @@ class StartCommand < Mongrel::Command::Command def configure options [ - ["-e", "--environment ENV", "Rails environment to run as", :@environment, "production"], - ["-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"] + ["-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 - valid? ["production","development"].include?(@environment), "Only valid environments are 'production' or 'development'" + @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 run - cwd = Dir.pwd - + def daemonize + # save this for later since daemonize will hose it if @daemon - puts "Running as Daemon at #@address:#@port" - Daemonize.daemonize(log_file=@log_file) - open(File.join(cwd,@pid_file),"w") {|f| f.write(Process.pid) } + 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) + Dir.chdir(@cwd) + + open(@pid_file,"w") {|f| f.write(Process.pid) } else - puts "Running at #@address:#@port" + puts "Running Mongrel server in #@environment mode at #@address:#@port" end + end - ENV['RAILS_ENV'] = @environment + 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' - h = Mongrel::HttpServer.new(@address, @port) - h.register("/", RailsHandler.new(File.join(cwd,"public"))) - h.run + + # 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 - h.acceptor.join + puts "Server ready." + server.acceptor.join rescue Interrupt puts "Interrupted." end end + + def run + daemonize + rails = configure_rails + start_mongrel(rails) + end end @@ -140,13 +188,18 @@ class StopCommand < Mongrel::Command::Command def configure options [ - ["-d", "--daemonize", "Whether to run in the background or not", :@daemon, false], - ['-P', '--pid FILE', "Where to write the PID", :@pid_file, "log/mongrel.pid"] + ['-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 -- cgit v1.2.3-24-ge0c7