mirror of
https://github.com/puma/puma.git
synced 2022-11-09 13:48:40 -05:00
85dfe8edcf
When options are passed to the Puma rack handler it is unknown if the options were set via a framework as a default or via a user. Puma currently has 3 different sources of configuration, the user via command line, the config files, and defaults. Rails 5.1+ will record the values actually specified by the user versus the values specified by the frameworks. It passes these values to the Rack handler and now it's up to Puma to do something with that information. When only framework defaults are passed it will set ``` options[:user_supplied_options] = [] ``` When one or more options are specified by the user such as `:Port` then those keys will be in the array. In that example it will look like this ``` options[:user_supplied_options] = [:Port] ``` This change is 100% backwards compatible. If the framework is older and does not pass this information then the `user_supplied_options` will not be set, in that case we assume all values are user supplied. Internally we accomplish this separation by replacing `LeveledOptions` which was a generic way of specifying options with different priorities with a more explicit `UserFileDefaultOptions` this assumes only 3 levels of options and it will use them in the order supplied (user config wins over file based config wins over defaults). Now instead of using 1 dsl to set all values, we use 3. A user dsl, a file dsl and a Configuration.new` will return all 3 DSLs to the block. It's up to the person using the block to use the correct dsl corresponding to the source of data they are getting.
104 lines
3.1 KiB
Ruby
104 lines
3.1 KiB
Ruby
require 'rack/handler'
|
|
|
|
module Rack
|
|
module Handler
|
|
module Puma
|
|
DEFAULT_OPTIONS = {
|
|
:Verbose => false,
|
|
:Silent => false
|
|
}
|
|
|
|
def self.config(app, options = {})
|
|
require 'puma/configuration'
|
|
require 'puma/events'
|
|
require 'puma/launcher'
|
|
|
|
default_options = DEFAULT_OPTIONS.dup
|
|
|
|
# Libraries pass in values such as :Port and there is no way to determine
|
|
# if it is a default provided by the library or a special value provided
|
|
# by the user. A special key `user_supplied_options` can be passed. This
|
|
# contains an array of all explicitly defined user options. We then
|
|
# know that all other values are defaults
|
|
if user_supplied_options = options.delete(:user_supplied_options)
|
|
(options.keys - user_supplied_options).each do |k, v|
|
|
default_options[k] = options.delete(k)
|
|
end
|
|
end
|
|
|
|
conf = ::Puma::Configuration.new(options, default_options) do |user_config, file_config, default_config|
|
|
user_config.quiet
|
|
|
|
if options.delete(:Verbose)
|
|
app = Rack::CommonLogger.new(app, STDOUT)
|
|
end
|
|
|
|
if options[:environment]
|
|
user_config.environment options[:environment]
|
|
end
|
|
|
|
if options[:Threads]
|
|
min, max = options.delete(:Threads).split(':', 2)
|
|
user_config.threads min, max
|
|
end
|
|
|
|
self.set_host_port_to_config(options[:Host], options[:Port], user_config)
|
|
self.set_host_port_to_config(default_options[:Host], default_options[:Port], default_config)
|
|
|
|
user_config.app app
|
|
end
|
|
conf
|
|
end
|
|
|
|
|
|
|
|
def self.run(app, options = {})
|
|
conf = self.config(app, options)
|
|
|
|
events = options.delete(:Silent) ? ::Puma::Events.strings : ::Puma::Events.stdio
|
|
|
|
launcher = ::Puma::Launcher.new(conf, :events => events)
|
|
|
|
yield launcher if block_given?
|
|
begin
|
|
launcher.run
|
|
rescue Interrupt
|
|
puts "* Gracefully stopping, waiting for requests to finish"
|
|
launcher.stop
|
|
puts "* Goodbye!"
|
|
end
|
|
end
|
|
|
|
def self.valid_options
|
|
{
|
|
"Host=HOST" => "Hostname to listen on (default: localhost)",
|
|
"Port=PORT" => "Port to listen on (default: 8080)",
|
|
"Threads=MIN:MAX" => "min:max threads to use (default 0:16)",
|
|
"Verbose" => "Don't report each request (default: false)"
|
|
}
|
|
end
|
|
private
|
|
def self.set_host_port_to_config(host, port, config)
|
|
if host && (host[0,1] == '.' || host[0,1] == '/')
|
|
config.bind "unix://#{host}"
|
|
elsif host && host =~ /^ssl:\/\//
|
|
uri = URI.parse(host)
|
|
uri.port ||= port || ::Puma::Configuration::DefaultTCPPort
|
|
config.bind uri.to_s
|
|
else
|
|
|
|
if host
|
|
port ||= ::Puma::Configuration::DefaultTCPPort
|
|
end
|
|
|
|
if port
|
|
host ||= ::Puma::Configuration::DefaultTCPHost
|
|
config.port port, host
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
register :puma, Puma
|
|
end
|
|
end
|