mirror of
https://github.com/puma/puma.git
synced 2022-11-09 13:48:40 -05:00
60 lines
1.3 KiB
Ruby
60 lines
1.3 KiB
Ruby
require 'ffi'
|
|
|
|
module Puma
|
|
module JRubyRestart
|
|
extend FFI::Library
|
|
ffi_lib 'c'
|
|
|
|
attach_function :execlp, [:string, :varargs], :int
|
|
attach_function :chdir, [:string], :int
|
|
attach_function :fork, [], :int
|
|
attach_function :exit, [:int], :void
|
|
attach_function :setsid, [], :int
|
|
|
|
def self.chdir_exec(dir, argv)
|
|
chdir(dir)
|
|
cmd = argv.first
|
|
argv = ([:string] * argv.size).zip(argv).flatten
|
|
argv << :string
|
|
argv << nil
|
|
execlp(cmd, *argv)
|
|
raise SystemCallError.new(FFI.errno)
|
|
end
|
|
|
|
def self.daemon?
|
|
ENV.key? 'PUMA_DAEMON_RESTART'
|
|
end
|
|
|
|
def self.daemon_init
|
|
return false unless ENV.key? 'PUMA_DAEMON_RESTART'
|
|
|
|
master = ENV['PUMA_DAEMON_RESTART']
|
|
Process.kill "SIGUSR2", master.to_i
|
|
|
|
setsid
|
|
|
|
null = File.open "/dev/null", "w+"
|
|
STDIN.reopen null
|
|
STDOUT.reopen null
|
|
STDERR.reopen null
|
|
|
|
true
|
|
end
|
|
|
|
def self.daemon_start(dir, argv)
|
|
ENV['PUMA_DAEMON_RESTART'] = Process.pid.to_s
|
|
|
|
cmd = argv.first
|
|
argv = ([:string] * argv.size).zip(argv).flatten
|
|
argv << :string
|
|
argv << nil
|
|
|
|
chdir(dir)
|
|
ret = fork
|
|
return ret if ret != 0
|
|
execlp(cmd, *argv)
|
|
raise SystemCallError.new(FFI.errno)
|
|
end
|
|
end
|
|
end
|
|
|