mirror of
https://github.com/puma/puma.git
synced 2022-11-09 13:48:40 -05:00
213 lines
4.8 KiB
Ruby
213 lines
4.8 KiB
Ruby
module Puma
|
|
class Configuration
|
|
DefaultRackup = "config.ru"
|
|
|
|
DefaultTCPHost = "0.0.0.0"
|
|
DefaultTCPPort = 9292
|
|
|
|
def initialize(options)
|
|
@options = options
|
|
@options[:binds] ||= []
|
|
@options[:on_restart] ||= []
|
|
end
|
|
|
|
attr_reader :options
|
|
|
|
def initialize_copy(other)
|
|
@options = @options.dup
|
|
end
|
|
|
|
def load
|
|
if path = @options[:config_file]
|
|
DSL.new(@options)._load_from path
|
|
end
|
|
|
|
# Rakeup default option support
|
|
if host = @options[:Host]
|
|
port = @options[:Port] || DefaultTCPPort
|
|
|
|
@options[:binds] << "tcp://#{host}:#{port}"
|
|
end
|
|
|
|
if @options[:binds].empty?
|
|
@options[:binds] << "tcp://#{DefaultTCPHost}:#{DefaultTCPPort}"
|
|
end
|
|
|
|
if @options[:control_url] == "auto"
|
|
path = Configuration.temp_path
|
|
@options[:control_url] = "unix://#{path}"
|
|
@options[:control_url_temp] = path
|
|
end
|
|
|
|
unless @options[:control_auth_token]
|
|
setup_random_token
|
|
end
|
|
end
|
|
|
|
# Load the specified rackup file, pull an options from
|
|
# the rackup file, and set @app.
|
|
#
|
|
def app
|
|
if app = @options[:app]
|
|
return app
|
|
end
|
|
|
|
path = @options[:rackup] || DefaultRackup
|
|
|
|
unless File.exists?(path)
|
|
raise "Missing rackup file '#{path}'"
|
|
end
|
|
|
|
app, options = Rack::Builder.parse_file path
|
|
@options.merge! options
|
|
|
|
options.each do |key,val|
|
|
if key.to_s[0,4] == "bind"
|
|
@options[:binds] << val
|
|
end
|
|
end
|
|
|
|
unless @options[:quiet]
|
|
app = Rack::CommonLogger.new(app, STDOUT)
|
|
end
|
|
|
|
return app
|
|
end
|
|
|
|
def setup_random_token
|
|
begin
|
|
require 'openssl'
|
|
rescue LoadError
|
|
end
|
|
|
|
count = 16
|
|
|
|
bytes = nil
|
|
|
|
if defined? OpenSSL::Random
|
|
bytes = OpenSSL::Random.random_bytes(count)
|
|
elsif File.exists?("/dev/urandom")
|
|
File.open("/dev/urandom") do |f|
|
|
bytes = f.read(count)
|
|
end
|
|
end
|
|
|
|
if bytes
|
|
token = ""
|
|
bytes.each_byte { |b| token << b.to_s(16) }
|
|
else
|
|
token = (0..count).to_a.map { rand(255).to_s(16) }.join
|
|
end
|
|
|
|
@options[:control_auth_token] = token
|
|
end
|
|
|
|
def self.temp_path
|
|
require 'tmpdir'
|
|
|
|
t = (Time.now.to_f * 1000).to_i
|
|
"#{Dir.tmpdir}/puma-status-#{t}-#{$$}"
|
|
end
|
|
|
|
# The methods that are available for use inside the config file.
|
|
#
|
|
class DSL
|
|
def initialize(options)
|
|
@options = options
|
|
end
|
|
|
|
def _load_from(path)
|
|
instance_eval File.read(path), path, 1
|
|
end
|
|
|
|
# Use +obj+ or +block+ as the Rack app. This allows a config file to
|
|
# be the app itself.
|
|
#
|
|
def app(obj=nil, &block)
|
|
obj ||= block
|
|
|
|
raise "Provide either a #call'able or a block" unless obj
|
|
|
|
@options[:app] = obj
|
|
end
|
|
|
|
# Start the Puma control rack app on +url+. This app can be communicated
|
|
# with to control the main server.
|
|
#
|
|
def activate_control_app(url="auto", opts=nil)
|
|
@options[:control_url] = url
|
|
|
|
if opts
|
|
if tok = opts[:auth_token]
|
|
@options[:control_auth_token] = tok
|
|
end
|
|
|
|
if opts[:no_token]
|
|
@options[:control_auth_token] = :none
|
|
end
|
|
end
|
|
end
|
|
|
|
# Bind the server to +url+. tcp:// and unix:// are the only accepted
|
|
# protocols.
|
|
#
|
|
def bind(url)
|
|
@options[:binds] << url
|
|
end
|
|
|
|
# Code to run before doing a restart. This code should
|
|
# close logfiles, database connections, etc.
|
|
#
|
|
# This can be called multiple times to add code each time.
|
|
#
|
|
def on_restart(&blk)
|
|
@options[:on_restart] << blk
|
|
end
|
|
|
|
# Store the pid of the server in the file at +path+.
|
|
def pidfile(path)
|
|
@options[:pidfile] = path
|
|
end
|
|
|
|
# Disable request logging.
|
|
#
|
|
def quiet
|
|
@options[:quiet] = true
|
|
end
|
|
|
|
# Load +path+ as a rackup file.
|
|
#
|
|
def rackup(path)
|
|
@options[:rackup] = path.to_s
|
|
end
|
|
|
|
# Configure +min+ to be the minimum number of threads to use to answer
|
|
# requests and +max+ the maximum.
|
|
#
|
|
def threads(min, max)
|
|
if min > max
|
|
raise "The minimum number of threads must be less than the max"
|
|
end
|
|
|
|
@options[:min_threads] = min
|
|
@options[:max_threads] = max
|
|
end
|
|
|
|
def ssl_bind(host, port, opts)
|
|
o = [
|
|
"cert=#{opts[:cert]}",
|
|
"key=#{opts[:key]}"
|
|
]
|
|
|
|
@options[:binds] << "ssl://#{host}:#{port}?#{o.join('&')}"
|
|
end
|
|
|
|
# Use +path+ as the file to store the server info state. This is
|
|
# used by pumactl to query and control the server.
|
|
#
|
|
def state_path(path)
|
|
@options[:state] = path.to_s
|
|
end
|
|
end
|
|
end
|
|
end
|