1
0
Fork 0
mirror of https://github.com/mperham/sidekiq.git synced 2022-11-09 13:52:34 -05:00
mperham--sidekiq/bin/sidekiqctl
2014-02-18 11:23:15 -08:00

78 lines
1.3 KiB
Ruby
Executable file

#!/usr/bin/env ruby
require 'fileutils'
class Sidekiqctl
attr_reader :stage, :pidfile, :timeout
def initialize(stage, pidfile, timeout)
@stage = stage
@pidfile = pidfile
@timeout = timeout
done('No pidfile given', :error) if !pidfile
done("Pidfile #{pidfile} does not exist", :warn) if !File.exist?(pidfile)
done('Invalid pidfile content', :error) if pid == 0
fetch_process
begin
send(stage)
rescue NoMethodError
done 'Invalid control command', :error
end
end
def fetch_process
Process.getpgid(pid)
rescue Errno::ESRCH
done "Process doesn't exist", :error
end
def done(msg, error = nil)
puts msg
exit(exit_signal(error))
end
def exit_signal(error)
(error == :error) ? 1 : 0
end
def pid
@pid ||= File.read(pidfile).to_i
end
def quiet
`kill -USR1 #{pid}`
end
def stop
`kill -TERM #{pid}`
timeout.times do
begin
Process.getpgid(pid)
rescue Errno::ESRCH
FileUtils.rm_f pidfile
done 'Sidekiq shut down gracefully.'
end
sleep 1
end
`kill -9 #{pid}`
FileUtils.rm_f pidfile
done 'Sidekiq shut down forcefully.'
end
def shutdown
quiet
stop
end
end
stage = ARGV[0]
pidfile = ARGV[1]
timeout = ARGV[2].to_i
timeout = 10 if timeout == 0
Sidekiqctl.new(stage, pidfile, timeout)