Safely stop Sidekiq process

stop_sidekiq.rake

require 'sidekiq/api'

namespace :sidekiq do
  task 'stop' do
    ps = Sidekiq::ProcessSet.new
    abort 'Sidekiq process not running' if ps.count == 0

    ps.first.quiet!

    puts "SIGUSR1 sent\nWaiting 10sec for status update"
    sleep 10

    while (running_tasks = ps.first['busy']) > 0
      puts "Waiting for tasks to finish. Num tasks: #{running_tasks}"
      sleep 5
    end

    ps.first.stop!
    puts 'SIGTERM sent'
  end
end