2012-01-16 19:14:47 -05:00
|
|
|
require 'optparse'
|
2012-01-25 16:32:51 -05:00
|
|
|
require 'sidekiq/version'
|
|
|
|
require 'sidekiq/util'
|
|
|
|
require 'sidekiq/client'
|
2012-02-03 13:02:57 -05:00
|
|
|
require 'sidekiq/manager'
|
2012-01-24 01:08:38 -05:00
|
|
|
require 'connection_pool'
|
2012-01-16 19:14:47 -05:00
|
|
|
|
|
|
|
module Sidekiq
|
|
|
|
class CLI
|
2012-01-22 19:01:46 -05:00
|
|
|
include Util
|
|
|
|
|
2012-01-16 19:14:47 -05:00
|
|
|
def initialize
|
|
|
|
parse_options
|
2012-01-16 23:02:58 -05:00
|
|
|
validate!
|
2012-01-23 17:05:03 -05:00
|
|
|
boot_rails
|
2012-01-16 19:14:47 -05:00
|
|
|
end
|
|
|
|
|
2012-01-24 01:08:38 -05:00
|
|
|
FOREVER = 2_000_000_000
|
|
|
|
|
2012-01-16 19:14:47 -05:00
|
|
|
def run
|
2012-01-24 01:08:38 -05:00
|
|
|
::Sidekiq::Client.redis = ConnectionPool.new { Redis.connect(:url => @options[:server]) }
|
2012-02-03 13:02:57 -05:00
|
|
|
manager = Sidekiq::Manager.new(@options[:server], @options)
|
2012-01-16 19:14:47 -05:00
|
|
|
begin
|
|
|
|
log 'Starting processing, hit Ctrl-C to stop'
|
2012-02-03 13:02:57 -05:00
|
|
|
manager.start!
|
2012-02-05 16:22:57 -05:00
|
|
|
# HACK need to determine how to pause main thread while
|
|
|
|
# waiting for signals.
|
2012-01-24 01:08:38 -05:00
|
|
|
sleep FOREVER
|
2012-01-16 19:14:47 -05:00
|
|
|
rescue Interrupt
|
2012-02-05 16:22:57 -05:00
|
|
|
# TODO Need clean shutdown support from Celluloid
|
|
|
|
log 'Shutting down, pausing 5 seconds to let workers finish...'
|
2012-02-03 13:02:57 -05:00
|
|
|
manager.stop!
|
|
|
|
manager.wait(:shutdown)
|
2012-01-16 19:14:47 -05:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2012-01-16 19:18:36 -05:00
|
|
|
private
|
|
|
|
|
2012-01-23 17:05:03 -05:00
|
|
|
def boot_rails
|
2012-02-07 20:16:40 -05:00
|
|
|
ENV['RAILS_ENV'] = @options[:environment] || ENV['RAILS_ENV'] || 'development'
|
2012-01-23 17:05:03 -05:00
|
|
|
require File.expand_path("#{@options[:rails]}/config/environment.rb")
|
2012-01-25 16:32:51 -05:00
|
|
|
::Rails.application.eager_load!
|
2012-01-16 23:02:58 -05:00
|
|
|
end
|
|
|
|
|
|
|
|
def validate!
|
2012-01-22 19:01:46 -05:00
|
|
|
$DEBUG = @options[:verbose]
|
2012-01-23 17:05:03 -05:00
|
|
|
|
|
|
|
if !File.exist?("#{@options[:rails]}/config/boot.rb")
|
|
|
|
log "========== Please point sidekiq to a Rails 3 application =========="
|
|
|
|
log @parser
|
|
|
|
exit(1)
|
|
|
|
end
|
2012-01-16 23:02:58 -05:00
|
|
|
end
|
|
|
|
|
2012-01-16 19:14:47 -05:00
|
|
|
def parse_options(argv=ARGV)
|
|
|
|
@options = {
|
2012-01-16 23:02:58 -05:00
|
|
|
:verbose => false,
|
2012-01-24 17:53:36 -05:00
|
|
|
:queues => ['default'],
|
2012-01-25 16:32:51 -05:00
|
|
|
:processor_count => 25,
|
2012-02-05 15:59:24 -05:00
|
|
|
:server => ENV['REDISTOGO_URL'] || 'redis://localhost:6379/0',
|
2012-01-23 17:05:03 -05:00
|
|
|
:rails => '.',
|
2012-02-07 20:16:40 -05:00
|
|
|
:environment => nil,
|
2012-01-16 19:14:47 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
@parser = OptionParser.new do |o|
|
2012-01-16 23:02:58 -05:00
|
|
|
o.on "-q", "--queue QUEUE,WEIGHT", "Queue to process, with optional weight" do |arg|
|
|
|
|
(q, weight) = arg.split(",")
|
|
|
|
(weight || 1).times do
|
|
|
|
@options[:queues] << q
|
|
|
|
end
|
2012-01-16 19:14:47 -05:00
|
|
|
end
|
|
|
|
|
2012-01-16 23:02:58 -05:00
|
|
|
o.on "-v", "--verbose", "Print more verbose output" do
|
|
|
|
@options[:verbose] = true
|
2012-01-16 19:14:47 -05:00
|
|
|
end
|
|
|
|
|
2012-01-26 00:12:18 -05:00
|
|
|
o.on "-s", "--server LOCATION", "Where to find Redis" do |arg|
|
2012-01-16 19:14:47 -05:00
|
|
|
@options[:server] = arg
|
|
|
|
end
|
|
|
|
|
2012-01-24 01:08:38 -05:00
|
|
|
o.on '-e', '--environment ENV', "Rails application environment" do |arg|
|
|
|
|
@options[:environment] = arg
|
|
|
|
end
|
|
|
|
|
2012-01-26 00:12:18 -05:00
|
|
|
o.on '-r', '--rails PATH', "Location of Rails application with workers" do |arg|
|
2012-01-23 17:05:03 -05:00
|
|
|
@options[:rails] = arg
|
|
|
|
end
|
|
|
|
|
2012-01-25 16:32:51 -05:00
|
|
|
o.on '-c', '--concurrency INT', "processor threads to use" do |arg|
|
|
|
|
@options[:processor_count] = arg.to_i
|
2012-01-16 19:14:47 -05:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2012-01-24 17:53:36 -05:00
|
|
|
@parser.banner = "sidekiq [options]"
|
2012-01-16 19:14:47 -05:00
|
|
|
@parser.on_tail "-h", "--help", "Show help" do
|
|
|
|
log @parser
|
|
|
|
exit 1
|
|
|
|
end
|
|
|
|
@parser.parse!(argv)
|
|
|
|
end
|
|
|
|
|
|
|
|
end
|
|
|
|
end
|