rosa-build/lib/recipes/resque.rb

88 lines
2.3 KiB
Ruby

Capistrano::Configuration.instance(:must_exist).load do
namespace :resque do
task :start do
start_workers
end
task :stop do
stop_workers
end
task :restart do
stop_workers
start_workers
end
def rails_env
fetch(:rails_env, false) ? "RAILS_ENV=#{fetch(:rails_env)}" : ''
end
def stop_workers
# ps = 'ps aux | grep resque | grep -v grep'
# run "#{ps} && kill -QUIT `#{ps} | awk '{ print $2 }'` || echo 'Workers already stopped!'"
run "cd #{fetch :current_path} && #{rails_env} bundle exec rake resque:stop_workers"
end
def start_workers
queue = [
:publish_observer,
:rpm_worker_observer,
:iso_worker_observer,
:fork_import,
:hook,
:clone_build,
:middle,
:notification
].join(',')
run "cd #{fetch :current_path} && COUNT=#{workers_count - 1} QUEUE=#{queue} INTERVAL=0.1 #{rails_env} BACKGROUND=yes bundle exec rake resque:workers"
run "cd #{fetch :current_path} && COUNT=1 QUEUE=low #{rails_env} BACKGROUND=yes bundle exec rake resque:workers"
end
def remote_file_exists?(full_path)
'true' == capture("if [ -e #{full_path} ]; then echo 'true'; fi").strip
end
namespace :scheduler do
desc "See current scheduler status"
task :status do
pid = "#{fetch :current_path}/tmp/pids/scheduler.pid"
if remote_file_exists?(pid)
info capture(:ps, "-f -p $(cat #{pid}) | sed -n 2p")
end
end
desc "Starts resque scheduler with default configs"
task :start do
start_scheduler
end
desc "Stops resque scheduler"
task :stop do
stop_scheduler
end
task :restart do
stop_scheduler
start_scheduler
end
def start_scheduler
pid = "#{fetch :current_path}/tmp/pids/scheduler.pid"
run "cd #{fetch :current_path} && #{rails_env} PIDFILE=#{pid} BACKGROUND=yes VERBOSE=1 MUTE=1 RESQUE_SCHEDULER_INTERVAL=0.5 bundle exec rake resque:scheduler"
end
def stop_scheduler
pid = "#{fetch :current_path}/tmp/pids/scheduler.pid"
if remote_file_exists?(pid)
run "cd #{fetch :current_path} && kill -s QUIT $(cat #{pid}); rm #{pid}"
end
end
end
end
end