mirror of
https://github.com/puma/puma.git
synced 2022-11-09 13:48:40 -05:00
130 lines
2.8 KiB
Ruby
130 lines
2.8 KiB
Ruby
module Puma
|
|
class Runner
|
|
def initialize(cli)
|
|
@cli = cli
|
|
@options = cli.options
|
|
@app = nil
|
|
@control = nil
|
|
end
|
|
|
|
def daemon?
|
|
@options[:daemon]
|
|
end
|
|
|
|
def development?
|
|
@options[:environment] == "development"
|
|
end
|
|
|
|
def log(str)
|
|
@cli.log str
|
|
end
|
|
|
|
def error(str)
|
|
@cli.error str
|
|
end
|
|
|
|
def before_restart
|
|
@control.stop(true) if @control
|
|
end
|
|
|
|
def start_control
|
|
str = @options[:control_url]
|
|
return unless str
|
|
|
|
require 'puma/app/status'
|
|
|
|
uri = URI.parse str
|
|
|
|
app = Puma::App::Status.new @cli
|
|
|
|
if token = @options[:control_auth_token]
|
|
app.auth_token = token unless token.empty? or token == :none
|
|
end
|
|
|
|
control = Puma::Server.new app, @cli.events
|
|
control.min_threads = 0
|
|
control.max_threads = 1
|
|
|
|
case uri.scheme
|
|
when "tcp"
|
|
log "* Starting control server on #{str}"
|
|
control.add_tcp_listener uri.host, uri.port
|
|
when "unix"
|
|
log "* Starting control server on #{str}"
|
|
path = "#{uri.host}#{uri.path}"
|
|
|
|
control.add_unix_listener path
|
|
else
|
|
error "Invalid control URI: #{str}"
|
|
end
|
|
|
|
control.run
|
|
@control = control
|
|
end
|
|
|
|
def output_header(mode)
|
|
min_t = @options[:min_threads]
|
|
max_t = @options[:max_threads]
|
|
|
|
log "Puma starting in #{mode} mode..."
|
|
log "* Version #{Puma::Const::PUMA_VERSION}, codename: #{Puma::Const::CODE_NAME}"
|
|
log "* Min threads: #{min_t}, max threads: #{max_t}"
|
|
log "* Environment: #{ENV['RACK_ENV']}"
|
|
end
|
|
|
|
def redirect_io
|
|
stdout = @options[:redirect_stdout]
|
|
stderr = @options[:redirect_stderr]
|
|
append = @options[:redirect_append]
|
|
|
|
if stdout
|
|
STDOUT.reopen stdout, (append ? "a" : "w")
|
|
STDOUT.sync = true
|
|
STDOUT.puts "=== puma startup: #{Time.now} ==="
|
|
end
|
|
|
|
if stderr
|
|
STDERR.reopen stderr, (append ? "a" : "w")
|
|
STDERR.sync = true
|
|
STDERR.puts "=== puma startup: #{Time.now} ==="
|
|
end
|
|
end
|
|
|
|
def load_and_bind
|
|
unless @cli.config.app_configured?
|
|
error "No application configured, nothing to run"
|
|
exit 1
|
|
end
|
|
|
|
# Load the app before we daemonize.
|
|
begin
|
|
@app = @cli.config.app
|
|
rescue Exception => e
|
|
log "! Unable to load application"
|
|
raise e
|
|
end
|
|
|
|
@cli.binder.parse @options[:binds], self
|
|
end
|
|
|
|
def app
|
|
@app ||= @cli.config.app
|
|
end
|
|
|
|
def start_server
|
|
min_t = @options[:min_threads]
|
|
max_t = @options[:max_threads]
|
|
|
|
server = Puma::Server.new app, @cli.events
|
|
server.min_threads = min_t
|
|
server.max_threads = max_t
|
|
server.inherit_binder @cli.binder
|
|
|
|
unless development?
|
|
server.leak_stack_on_error = false
|
|
end
|
|
|
|
server
|
|
end
|
|
end
|
|
end
|