Class: MicroQ::CLI
- Inherits:
-
Object
- Object
- MicroQ::CLI
- Defined in:
- lib/micro_q/cli.rb
Class Method Summary collapse
Instance Method Summary collapse
Class Method Details
.run ⇒ Object
6 7 8 9 10 11 |
# File 'lib/micro_q/cli.rb', line 6 def self.run @cli ||= new @cli.parse @cli.verify! @cli.setup end |
Instance Method Details
#parse ⇒ Object
13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
# File 'lib/micro_q/cli.rb', line 13 def parse opts = Slop.parse do 'Usage: microq [options]' on 'h', 'This menu' on 'help', 'This menu' on 't=', 'The queue type to process [redis, sqs]' on 'type=', 'The queue type to process [redis, sqs]' on 'r=', 'The path to the rails application' on 'require=', 'The path to the rails application' on 'w=', 'The number of worker threads' on 'workers=', 'The number of worker threads' end (puts usage; exit) if opts[:help] || opts[:h] @workers = opts[:workers] || opts[:w] @require = opts[:require] || opts[:r] || 'config/environment.rb' @mode = (opts[:type] || opts[:t] || 'sqs'.tap { puts 'Defaulting to sqs type' }).downcase end |
#setup ⇒ Object
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
# File 'lib/micro_q/cli.rb', line 43 def setup puts 'Requiring rails application...' require 'rails' if File.directory?(@require) require File.("#{@require}/config/environment.rb") else require @require end Rails.application.eager_load! aws_keys = MicroQ.config.aws.try(:keys) || [] queue = if @mode == 'sqs' raise 'SQS mode requires an aws :key and :secret see https://github.com/bnorton/micro_q/wiki/Named-Queues' unless aws_keys.include?(:key) && aws_keys.include?(:secret) MicroQ::Queue::Sqs elsif @mode == 'redis' puts 'Using redis config' + MicroQ.config.redis.map {|(k,v)| "#{k}=#{v}" }.join(' ') MicroQ::Queue::Redis else raise 'Only Redis and SQS mode are supported via the command line.' end MicroQ.configure do |config| config.queue = queue # set workers after assigning the queue since this sets workers=0 internally for the sqs queue config.workers = @workers.to_i if @workers config['worker_mode?'] = true end puts "Running micro_q in #{@mode.upcase} mode with #{MicroQ.config.workers} workers..." puts 'ctl+c to shutdown...' MicroQ.start sleep rescue Interrupt puts 'Exiting via interrupt' exit(1) end |
#usage ⇒ Object
86 87 88 89 90 91 92 93 94 95 |
# File 'lib/micro_q/cli.rb', line 86 def usage <<-USAGE Run microq with messages enqueued to and dequeued from SQS or Redis. Usage: microq [options] -h, --help This menu. -t, --type [redis, sqs] The type of queue being utilized (either Redis or SQS) -r, --require The path the rails app's config/environment.rb file. -w, --workers The number of workers for this process. USAGE end |
#verify! ⇒ Object
34 35 36 37 38 39 40 41 |
# File 'lib/micro_q/cli.rb', line 34 def verify! unless File.exist?(@require) || File.exist?("#{@require}/config/application.rb") puts 'Typically the -r option is simply the base path of the application' puts 'This could be `pwd` or something special in production...' puts "We tried both File.exist?(opts[:require]) and File.exist?(opts[:require] + 'config/application.rb')" raise "MicroQ requires a valid rails application path via the -r option\n" end end |