mirror of
https://github.com/puma/puma.git
synced 2022-11-09 13:48:40 -05:00
131 lines
2.8 KiB
Ruby
131 lines
2.8 KiB
Ruby
require 'optparse'
|
|
require 'uri'
|
|
|
|
require 'puma/server'
|
|
require 'puma/const'
|
|
|
|
module Puma
|
|
class CLI
|
|
DefaultTCPHost = "0.0.0.0"
|
|
DefaultTCPPort = 9292
|
|
|
|
def initialize(argv, stdout=STDOUT, stderr=STDERR)
|
|
@argv = argv
|
|
@stdout = stdout
|
|
@stderr = stderr
|
|
|
|
@events = Events.new @stdout, @stderr
|
|
|
|
setup_options
|
|
end
|
|
|
|
def log(str)
|
|
@stdout.puts str
|
|
end
|
|
|
|
def error(str)
|
|
@stderr.puts "ERROR: #{str}"
|
|
exit 1
|
|
end
|
|
|
|
def setup_options
|
|
@options = {
|
|
:min_threads => 0,
|
|
:max_threads => 16
|
|
}
|
|
|
|
@binds = []
|
|
|
|
@parser = OptionParser.new do |o|
|
|
o.on '-t', '--threads INT', "min:max threads to use (default 0:16)" do |arg|
|
|
min, max = arg.split(":")
|
|
if max
|
|
@options[:min_threads] = min.to_i
|
|
@options[:max_threads] = max.to_i
|
|
else
|
|
@options[:min_threads] = 0
|
|
@options[:max_threads] = arg.to_i
|
|
end
|
|
end
|
|
|
|
o.on "-b", "--bind URI", "URI to bind to (tcp:// and unix:// only)" do |arg|
|
|
@binds << arg
|
|
end
|
|
end
|
|
|
|
@parser.banner = "puma <options> <rackup file>"
|
|
|
|
@parser.on_tail "-h", "--help", "Show help" do
|
|
log @parser
|
|
exit 1
|
|
end
|
|
end
|
|
|
|
def load_rackup
|
|
@app, options = Rack::Builder.parse_file @rackup
|
|
@options.merge! options
|
|
|
|
options.each do |key,val|
|
|
if key.to_s[0,4] == "bind"
|
|
@binds << val
|
|
end
|
|
end
|
|
end
|
|
|
|
def run
|
|
@parser.parse! @argv
|
|
|
|
@rackup = ARGV.shift || "config.ru"
|
|
|
|
unless File.exists?(@rackup)
|
|
raise "Missing rackup file '#{@rackup}'"
|
|
end
|
|
|
|
load_rackup
|
|
|
|
if @binds.empty?
|
|
@options[:Host] ||= DefaultTCPHost
|
|
@options[:Port] ||= DefaultTCPPort
|
|
end
|
|
|
|
min_t = @options[:min_threads]
|
|
max_t = @options[:max_threads]
|
|
|
|
server = Puma::Server.new @app, @events
|
|
server.min_threads = min_t
|
|
server.max_threads = max_t
|
|
|
|
log "Puma #{Puma::Const::PUMA_VERSION} starting..."
|
|
log "* Min threads: #{min_t}, max threads: #{max_t}"
|
|
|
|
if @options[:Host]
|
|
log "* Listening on tcp://#{@options[:Host]}:#{@options[:Port]}"
|
|
server.add_tcp_listener @options[:Host], @options[:Port]
|
|
end
|
|
|
|
@binds.each do |str|
|
|
uri = URI.parse str
|
|
case uri.scheme
|
|
when "tcp"
|
|
log "* Listening on #{str}"
|
|
server.add_tcp_listener uri.host, uri.port
|
|
when "unix"
|
|
log "* Listening on #{str}"
|
|
path = "#{uri.host}#{uri.path}"
|
|
|
|
server.add_unix_listener path
|
|
else
|
|
error "Invalid URI: #{str}"
|
|
end
|
|
end
|
|
|
|
if server.attempt_bonjour(@rackup)
|
|
log "* Announced services via bonjour"
|
|
end
|
|
|
|
log "Use Ctrl-C to stop"
|
|
|
|
server.run.join
|
|
end
|
|
end
|
|
end
|