1
0
Fork 0
mirror of https://github.com/puma/puma.git synced 2022-11-09 13:48:40 -05:00
puma--puma/lib/puma/launcher.rb

546 lines
15 KiB
Ruby
Raw Normal View History

# frozen_string_literal: true
require 'puma/events'
require 'puma/detect'
require 'puma/cluster'
require 'puma/single'
require 'puma/const'
require 'puma/binder'
2016-02-03 15:12:17 -05:00
module Puma
2016-04-07 14:22:15 -04:00
# Puma::Launcher is the single entry point for starting a Puma server based on user
2016-02-03 18:21:25 -05:00
# configuration. It is responsible for taking user supplied arguments and resolving them
# with configuration in `config/puma.rb` or `config/puma/<env>.rb`.
#
# It is responsible for either launching a cluster of Puma workers or a single
# puma server.
class Launcher
2016-02-04 10:49:01 -05:00
KEYS_NOT_TO_PERSIST_IN_STATE = [
:logger, :lowlevel_error_handler,
:before_worker_shutdown, :before_worker_boot, :before_worker_fork,
:after_worker_boot, :before_fork, :on_restart
]
2016-02-03 18:21:25 -05:00
# Returns an instance of Launcher
#
# +conf+ A Puma::Configuration object indicating how to run the server.
2016-02-03 18:21:25 -05:00
#
# +launcher_args+ A Hash that currently has one required key `:events`,
# this is expected to hold an object similar to an `Puma::Events.stdio`,
# this object will be responsible for broadcasting Puma's internal state
# to a logging destination. An optional key `:argv` can be supplied,
# this should be an array of strings, these arguments are re-used when
# restarting the puma server.
2016-02-03 18:21:25 -05:00
#
# Examples:
#
# conf = Puma::Configuration.new do |user_config|
# user_config.threads 1, 10
# user_config.app do |env|
# [200, {}, ["hello world"]]
# end
# end
# Puma::Launcher.new(conf, events: Puma::Events.stdio).run
def initialize(conf, launcher_args={})
@runner = nil
@events = launcher_args[:events] || Events::DEFAULT
2016-02-03 18:22:40 -05:00
@argv = launcher_args[:argv] || []
@original_argv = @argv.dup
@config = conf
@binder = Binder.new(@events, conf)
@binder.create_inherited_fds(ENV).each { |k| ENV.delete k }
@binder.create_activated_fds(ENV).each { |k| ENV.delete k }
@environment = conf.environment
# Advertise the Configuration
Puma.cli_config = @config if defined?(Puma.cli_config)
@config.load
if @config.options[:bind_to_activated_sockets]
@config.options[:binds] = @binder.synthesize_binds_from_activated_fs(
@config.options[:binds],
@config.options[:bind_to_activated_sockets] == 'only'
)
end
@options = @config.options
2017-03-03 15:41:42 -05:00
@config.clamp
@events.formatter = Events::PidFormatter.new if clustered?
@events.formatter = options[:log_formatter] if @options[:log_formatter]
generate_restart_data
if clustered? && !Process.respond_to?(:fork)
unsupported "worker mode not supported on #{RUBY_ENGINE} on this platform"
end
Dir.chdir(@restart_dir)
prune_bundler if prune_bundler?
@environment = @options[:environment] if @options[:environment]
set_rack_environment
if clustered?
@options[:logger] = @events
@runner = Cluster.new(self, @events)
else
@runner = Single.new(self, @events)
end
Puma.stats_object = @runner
@status = :run
log_config if ENV['PUMA_LOG_CONFIG']
end
attr_reader :binder, :events, :config, :options, :restart_dir
2016-02-03 15:12:17 -05:00
# Return stats about the server
def stats
@runner.stats
end
# Write a state file that can be used by pumactl to control
# the server
def write_state
write_pid
path = @options[:state]
permission = @options[:state_permission]
return unless path
2016-09-08 11:46:24 -04:00
require 'puma/state_file'
sf = StateFile.new
sf.pid = Process.pid
sf.control_url = @options[:control_url]
sf.control_auth_token = @options[:control_auth_token]
2020-05-23 13:27:51 -04:00
sf.running_from = File.expand_path('.')
sf.save path, permission
end
# Delete the configured pidfile
def delete_pidfile
path = @options[:pidfile]
File.unlink(path) if path && File.exist?(path)
end
# Begin async shutdown of the server
def halt
@status = :halt
@runner.halt
end
# Begin async shutdown of the server gracefully
def stop
@status = :stop
@runner.stop
end
# Begin async restart of the server
def restart
@status = :restart
@runner.restart
end
# Begin a phased restart if supported
def phased_restart
unless @runner.respond_to?(:phased_restart) and @runner.phased_restart
log "* phased-restart called but not available, restarting normally."
return restart
end
true
end
# Run the server. This blocks until the server is stopped
def run
previous_env =
if defined?(Bundler)
env = Bundler::ORIGINAL_ENV.dup
# add -rbundler/setup so we load from Gemfile when restarting
bundle = "-rbundler/setup"
env["RUBYOPT"] = [env["RUBYOPT"], bundle].join(" ").lstrip unless env["RUBYOPT"].to_s.include?(bundle)
env
else
ENV.to_h
end
@config.clamp
@config.plugins.fire_starts self
setup_signals
set_process_title
integrate_with_systemd
@runner.run
case @status
when :halt
log "* Stopping immediately!"
@runner.stop_control
when :run, :stop
graceful_stop
when :restart
log "* Restarting..."
ENV.replace(previous_env)
@runner.stop_control
restart!
when :exit
# nothing
end
close_binder_listeners unless @status == :restart
end
# Return all tcp ports the launcher may be using, TCP or SSL
# @!attribute [r] connected_ports
# @version 5.0.0
def connected_ports
@binder.connected_ports
end
# @!attribute [r] restart_args
2016-03-05 19:28:39 -05:00
def restart_args
cmd = @options[:restart_cmd]
if cmd
cmd.split(' ') + @original_argv
else
@restart_argv
end
end
def close_binder_listeners
@runner.close_control_listeners
@binder.close_listeners
unless @status == :restart
log "=== puma shutdown: #{Time.now} ==="
log "- Goodbye!"
end
end
# @!attribute [r] thread_status
# @version 5.0.0
def thread_status
Thread.list.each do |thread|
name = "Thread: TID-#{thread.object_id.to_s(36)}"
name += " #{thread['label']}" if thread['label']
name += " #{thread.name}" if thread.respond_to?(:name) && thread.name
backtrace = thread.backtrace || ["<no backtrace available>"]
yield name, backtrace
end
end
private
2019-09-20 05:51:57 -04:00
# If configured, write the pid of the current process out
# to a file.
def write_pid
path = @options[:pidfile]
return unless path
cur_pid = Process.pid
File.write path, cur_pid, mode: 'wb:UTF-8'
2019-09-20 05:51:57 -04:00
at_exit do
delete_pidfile if cur_pid == Process.pid
2019-09-20 05:51:57 -04:00
end
end
def reload_worker_directory
@runner.reload_worker_directory if @runner.respond_to?(:reload_worker_directory)
end
def restart!
@events.fire_on_restart!
@config.run_hooks :on_restart, self, @events
if Puma.jruby?
close_binder_listeners
require 'puma/jruby_restart'
JRubyRestart.chdir_exec(@restart_dir, restart_args)
elsif Puma.windows?
close_binder_listeners
argv = restart_args
Dir.chdir(@restart_dir)
Kernel.exec(*argv)
else
argv = restart_args
Dir.chdir(@restart_dir)
ENV.update(@binder.redirects_for_restart_env)
argv += [@binder.redirects_for_restart]
Kernel.exec(*argv)
end
end
# @!attribute [r] files_to_require_after_prune
def files_to_require_after_prune
puma = spec_for_gem("puma")
require_paths_for_gem(puma) + extra_runtime_deps_directories
end
# @!attribute [r] extra_runtime_deps_directories
def extra_runtime_deps_directories
Array(@options[:extra_runtime_dependencies]).map do |d_name|
if (spec = spec_for_gem(d_name))
require_paths_for_gem(spec)
else
log "* Could not load extra dependency: #{d_name}"
nil
end
end.flatten.compact
end
# @!attribute [r] puma_wild_location
def puma_wild_location
puma = spec_for_gem("puma")
dirs = require_paths_for_gem(puma)
puma_lib_dir = dirs.detect { |x| File.exist? File.join(x, '../bin/puma-wild') }
File.expand_path(File.join(puma_lib_dir, "../bin/puma-wild"))
end
def prune_bundler
return if ENV['PUMA_BUNDLER_PRUNED']
return unless defined?(Bundler)
require_rubygems_min_version!(Gem::Version.new("2.2"), "prune_bundler")
unless puma_wild_location
log "! Unable to prune Bundler environment, continuing"
return
end
dirs = files_to_require_after_prune
2020-02-13 17:55:35 -05:00
log '* Pruning Bundler environment'
home = ENV['GEM_HOME']
bundle_gemfile = Bundler.original_env['BUNDLE_GEMFILE']
bundle_app_config = Bundler.original_env['BUNDLE_APP_CONFIG']
with_unbundled_env do
ENV['GEM_HOME'] = home
ENV['BUNDLE_GEMFILE'] = bundle_gemfile
ENV['PUMA_BUNDLER_PRUNED'] = '1'
ENV["BUNDLE_APP_CONFIG"] = bundle_app_config
args = [Gem.ruby, puma_wild_location, '-I', dirs.join(':')] + @original_argv
# Ruby 2.0+ defaults to true which breaks socket activation
2019-06-27 18:32:01 -04:00
args += [{:close_others => false}]
Kernel.exec(*args)
end
end
#
# Puma's systemd integration allows Puma to inform systemd:
# 1. when it has successfully started
# 2. when it is starting shutdown
# 3. periodically for a liveness check with a watchdog thread
#
def integrate_with_systemd
return unless ENV["NOTIFY_SOCKET"]
begin
require 'puma/systemd'
rescue LoadError
log "Systemd integration failed. It looks like you're trying to use systemd notify but don't have sd_notify gem installed"
return
end
log "* Enabling systemd notification integration"
systemd = Systemd.new(@events)
systemd.hook_events
systemd.start_watchdog
end
def spec_for_gem(gem_name)
Bundler.rubygems.loaded_specs(gem_name)
end
def require_paths_for_gem(gem_spec)
gem_spec.full_require_paths
end
def log(str)
@events.log str
end
def clustered?
(@options[:workers] || 0) > 0
end
def unsupported(str)
@events.error(str)
raise UnsupportedOption
end
def graceful_stop
@events.fire_on_stopped!
@runner.stop_blocked
end
def set_process_title
Process.respond_to?(:setproctitle) ? Process.setproctitle(title) : $0 = title
end
# @!attribute [r] title
def title
buffer = "puma #{Puma::Const::VERSION} (#{@options[:binds].join(',')})"
buffer += " [#{@options[:tag]}]" if @options[:tag] && !@options[:tag].empty?
buffer
end
def set_rack_environment
@options[:environment] = environment
ENV['RACK_ENV'] = environment
end
# @!attribute [r] environment
def environment
@environment
end
def prune_bundler?
@options[:prune_bundler] && clustered? && !@options[:preload_app]
end
def generate_restart_data
if dir = @options[:directory]
@restart_dir = dir
elsif Puma.windows?
# I guess the value of PWD is garbage on windows so don't bother
# using it.
@restart_dir = Dir.pwd
# Use the same trick as unicorn, namely favor PWD because
# it will contain an unresolved symlink, useful for when
# the pwd is /data/releases/current.
elsif dir = ENV['PWD']
s_env = File.stat(dir)
s_pwd = File.stat(Dir.pwd)
if s_env.ino == s_pwd.ino and (Puma.jruby? or s_env.dev == s_pwd.dev)
@restart_dir = dir
end
end
@restart_dir ||= Dir.pwd
# if $0 is a file in the current directory, then restart
# it the same, otherwise add -S on there because it was
# picked up in PATH.
#
if File.exist?($0)
arg0 = [Gem.ruby, $0]
else
arg0 = [Gem.ruby, "-S", $0]
end
# Detect and reinject -Ilib from the command line, used for testing without bundler
# cruby has an expanded path, jruby has just "lib"
lib = File.expand_path "lib"
arg0[1,0] = ["-I", lib] if [lib, "lib"].include?($LOAD_PATH[0])
if defined? Puma::WILD_ARGS
@restart_argv = arg0 + Puma::WILD_ARGS + @original_argv
else
@restart_argv = arg0 + @original_argv
end
end
def setup_signals
begin
Signal.trap "SIGUSR2" do
restart
end
rescue Exception
log "*** SIGUSR2 not implemented, signal based restart unavailable!"
end
2016-03-06 00:44:37 -05:00
unless Puma.jruby?
begin
Signal.trap "SIGUSR1" do
phased_restart
end
rescue Exception
log "*** SIGUSR1 not implemented, signal based restart unavailable!"
end
end
begin
Signal.trap "SIGTERM" do
graceful_stop
raise(SignalException, "SIGTERM") if @options[:raise_exception_on_sigterm]
end
rescue Exception
log "*** SIGTERM not implemented, signal based gracefully stopping unavailable!"
end
begin
Signal.trap "SIGINT" do
stop
end
rescue Exception
log "*** SIGINT not implemented, signal based gracefully stopping unavailable!"
end
begin
Signal.trap "SIGHUP" do
if @runner.redirected_io?
@runner.redirect_io
else
stop
end
end
rescue Exception
log "*** SIGHUP not implemented, signal based logs reopening unavailable!"
end
begin
2020-03-14 15:29:05 -04:00
unless Puma.jruby? # INFO in use by JVM already
Signal.trap "SIGINFO" do
thread_status do |name, backtrace|
@events.log name
@events.log backtrace.map { |bt| " #{bt}" }
end
end
end
rescue Exception
# Not going to log this one, as SIGINFO is *BSD only and would be pretty annoying
# to see this constantly on Linux.
end
end
def require_rubygems_min_version!(min_version, feature)
return if min_version <= Gem::Version.new(Gem::VERSION)
raise "#{feature} is not supported on your version of RubyGems. " \
"You must have RubyGems #{min_version}+ to use this feature."
end
# @version 5.0.0
def with_unbundled_env
bundler_ver = Gem::Version.new(Bundler::VERSION)
if bundler_ver < Gem::Version.new('2.1.0')
Bundler.with_clean_env { yield }
else
Bundler.with_unbundled_env { yield }
end
end
def log_config
log "Configuration:"
@config.final_options
.each { |config_key, value| log "- #{config_key}: #{value}" }
log "\n"
end
end
end