Contents

require File.dirname(__FILE__) + '/../sweat_shop'
require 'i_can_daemonize'

module SweatShop
  class Sweatd
    include ICanDaemonize
    queues     = []
    groups     = []
    rails_root = nil

    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
    
    before do
      if rails_root
        puts "Loading Rails..."
        require rails_root + '/config/environment' 
      end
    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

7 entries across 7 versions & 1 rubygems

Version	Path
famoseagle-sweat_shop-0.3.0	lib/sweat_shop/sweatd.rb
famoseagle-sweat_shop-0.5.0	lib/sweat_shop/sweatd.rb
famoseagle-sweat_shop-0.6.0	lib/sweat_shop/sweatd.rb
famoseagle-sweat_shop-0.7.0	lib/sweat_shop/sweatd.rb
famoseagle-sweat_shop-0.8.0	lib/sweat_shop/sweatd.rb
famoseagle-sweat_shop-0.8.1	lib/sweat_shop/sweatd.rb
famoseagle-sweat_shop-0.8.2	lib/sweat_shop/sweatd.rb