Contents

require File.dirname(__FILE__) + '/daemoned'

module SweatShop
  class Sweatd
    include Daemoned
    queues     = []
    groups     = []
    rails_root = nil
    start_cmd  = "ruby #{__FILE__} start #{ARGV.reject{|a| a == 'start'}.join(' ')}"

    arg '--workers=Worker,Worker', 'Workers to service (Default is all)' do |value|
      queues = value.split(',')
    end

    arg '--groups=GROUP,GROUP', 'Groups of queues to service' do |value|
      groups = value.split(',').collect{|g| g.to_sym}
    end

    arg '--worker-file=WORKERFILE', 'Worker file to load'  do |value|
      require value
    end

    arg '--worker-dir=WORKERDIR', 'Directory containing workers'  do |value|
      Dir.glob(value + '*.rb').each{|worker| require worker}
    end

    arg '--rails=DIR', 'Pass in RAILS_ROOT to run this daemon in a rails environment' do |value|
      rails_root = value
    end

    sig(:term, :int) do
      puts "Shutting down sweatd..."
      SweatShop.stop
    end

    sig(:hup) do
      puts "Received HUP"
      SweatShop.stop
      remove_pid!
      puts "Restarting sweatd with #{start_cmd}..."
      `#{start_cmd}`        
    end
    
    before do
      if rails_root
        puts "Loading Rails..."
        require rails_root + '/config/environment' 
      end
      require File.dirname(__FILE__) + '/../sweat_shop'
    end

    daemonize(:kill_timeout => 20) do
      workers = []

      if groups.any?
        workers += SweatShop.workers_in_group(groups)
      end

      if queues.any?
        workers += queues.collect{|q| Object.module_eval(q)}
      end

      if workers.any?
        worker_str = workers.join(',')
        puts "Starting #{worker_str}..." 
        $0 = "Sweatd: #{worker_str}"
        SweatShop.do_tasks(workers)
      else
        puts "Starting all workers..." 
        $0 = 'Sweatd: all'
        SweatShop.do_all_tasks
      end
    end

  end
end

Version data entries

11 entries across 11 versions & 6 rubygems

Version	Path
Bira-sweat_shop-1.0.0	lib/sweat_shop/sweatd.rb
famoseagle-sweat_shop-1.0.0	lib/sweat_shop/sweatd.rb
famoseagle-sweat_shop-1.1.0	lib/sweat_shop/sweatd.rb
famoseagle-sweat_shop-1.2.0	lib/sweat_shop/sweatd.rb
famoseagle-sweat_shop-1.3.0	lib/sweat_shop/sweatd.rb
famoseagle-sweat_shop-1.3.1	lib/sweat_shop/sweatd.rb
netinlet-sweat_shop-1.1.0	lib/sweat_shop/sweatd.rb
netinlet-sweat_shop-1.1.1	lib/sweat_shop/sweatd.rb
sa-sweat_shop-1.4.3	lib/sweat_shop/sweatd.rb
sweatshop-1.4.0	lib/sweat_shop/sweatd.rb
sweat_shop-1.4.0	lib/sweat_shop/sweatd.rb