sweatshop 1.4.0

data/script/sweatshop

@@ -0,0 +1,17 @@
+#!/usr/bin/env ruby
+
+if ARGV.include?('-d')
+  ARGV.delete('-d')
+  ARGV.unshift('start') unless ARGV.detect{|a| ['start', 'restart'].include?(a)}
+
+  root = File.expand_path(File.dirname(__FILE__) + '/..')
+  ARGV << "--rails=#{root}"
+
+  system("ruby #{root}/vendor/gems/sweat_shop/lib/sweat_shop/sweatd.rb #{ARGV.join(' ')}")
+  exit $?.exitstatus
+else
+  puts "Loading Rails..."
+  require File.dirname(__FILE__) + '/../config/environment'
+  puts "Listening for new tasks..."
+  SweatShop.do_all_tasks
+end

data/test/hello_worker.rb

@@ -0,0 +1,13 @@
+require File.dirname(__FILE__) + '/../lib/sweat_shop'
+class HelloWorker < SweatShop::Worker
+  TEST_FILE = File.dirname(__FILE__) + '/test.txt' unless defined?(TEST_FILE)
+
+  def hello(name)
+    puts name
+    "Hi, #{name}"
+  end
+
+  after_task do |task|
+    File.open(TEST_FILE, 'w'){|f| f << task[:result]}
+  end
+end

data/test/test_functional_worker.rb

@@ -0,0 +1,72 @@
+require File.dirname(__FILE__) + '/../lib/sweat_shop'
+require File.dirname(__FILE__) + '/test_helper'
+require File.dirname(__FILE__) + '/hello_worker'
+
+class WorkerTest < Test::Unit::TestCase
+
+  def setup
+    File.delete(HelloWorker::TEST_FILE) if File.exist?(HelloWorker::TEST_FILE)
+  end
+
+  def teardown
+    SweatShop.instance_variable_set("@config", nil)
+    SweatShop.instance_variable_set("@queues", nil)
+    File.delete(HelloWorker::TEST_FILE) if File.exist?(HelloWorker::TEST_FILE)
+  end
+
+  should "daemonize" do
+    begin
+      SweatShop.config['enable'] = true
+      SweatShop.logger = :silent
+  
+      HelloWorker.async_hello('Amos')
+  
+      worker = File.expand_path(File.dirname(__FILE__) + '/hello_worker')
+      sweatd = "#{File.dirname(__FILE__)}/../lib/sweat_shop/sweatd.rb"
+  
+      `ruby #{sweatd} --worker-file #{worker} start`
+      `ruby #{sweatd} stop`
+  
+      File.delete('sweatd.log') if File.exist?('sweatd.log')
+      assert_equal 'Hi, Amos', File.read(HelloWorker::TEST_FILE)
+  
+    rescue Exception => e
+      puts e.message
+      puts e.backtrace.join("\n")
+      fail "\n\n*** Functional test failed, is the rabbit server running on localhost? ***\n"
+    end
+  end
+
+  should "connect to fallback servers if the default one is down" do
+    begin
+      SweatShop.logger = :silent
+      SweatShop.config['enable'] = true
+      SweatShop.config['default']['cluster'] =
+        [
+         'localhost:5671', # invalid
+         'localhost:5672'  # valid
+        ]
+      HelloWorker.async_hello('Amos')
+  
+      assert_equal 'Amos', HelloWorker.dequeue[:args].first
+
+      HelloWorker.queue.client = nil
+
+      HelloWorker.stop
+      SweatShop.config['default']['cluster'] =
+        [
+         'localhost:5671',# valid
+         'localhost:5672' # invalid
+        ]
+  
+      HelloWorker.async_hello('Amos')
+      assert_equal 'Amos', HelloWorker.dequeue[:args].first
+  
+    rescue Exception => e
+      puts e.message
+      puts e.backtrace.join("\n")
+      fail "\n\n*** Functional test failed, is the rabbit server running on localhost? ***\n"
+    end
+  end
+
+end

data/test/test_helper.rb

@@ -0,0 +1,4 @@
+require 'test/unit'
+require 'shoulda'
+require 'pp'
+

data/test/test_sweatshop.rb

@@ -0,0 +1,78 @@
+require File.dirname(__FILE__) + '/test_helper'
+require File.dirname(__FILE__) + '/../lib/sweat_shop'
+
+class SweatShopTest < Test::Unit::TestCase
+  SweatShop.workers = []
+
+  class HelloWorker < SweatShop::Worker
+    def hello(name)
+      "Hi, #{name}"
+    end
+  end
+
+  class GroupedWorker < SweatShop::Worker
+    queue_group :foo
+  end
+
+  should "group workers" do
+    assert_equal [HelloWorker, GroupedWorker], SweatShop.workers_in_group(:all)
+    assert_equal [HelloWorker],   SweatShop.workers_in_group(:default)
+    assert_equal [GroupedWorker], SweatShop.workers_in_group(:foo)
+  end
+
+  should "synch call" do
+    worker = HelloWorker.new
+    assert_equal "Hi, Amos", worker.hello('Amos')
+  end
+
+  should "assign a uid" do
+    SweatShop.logger = :silent
+    SweatShop.config['enable'] = false
+    uid = HelloWorker.async_hello('Amos')
+    assert_not_nil uid
+  end
+
+  should "have before task" do
+    HelloWorker.before_task do
+      "hello"
+    end
+    assert_equal "hello", HelloWorker.before_task.call
+  end
+
+  should "have after task" do
+    HelloWorker.after_task do
+      "goodbye"
+    end
+    assert_equal "goodbye", HelloWorker.after_task.call
+  end
+
+  should "exception handler" do
+    SweatShop.logger = :silent
+
+    exception = nil
+    HelloWorker.on_exception do |e|
+      exception = e
+    end
+
+    HelloWorker.do_task(nil)
+    assert_equal NoMethodError, exception.class
+  end
+
+  should "chain before tasks" do
+    MESSAGES = []
+    class BaseWorker < SweatShop::Worker
+      before_task do |task|
+        MESSAGES << 'base'
+      end
+    end
+    class SubWorker < BaseWorker
+      before_task do |task|
+        MESSAGES << 'sub'
+      end
+    end
+    SubWorker.call_before_task('foo')
+    assert_equal ['base', 'sub'], MESSAGES
+    SweatShop.workers.delete(BaseWorker)
+    SweatShop.workers.delete(SubWorker)
+  end
+end

metadata

@@ -0,0 +1,91 @@
+--- !ruby/object:Gem::Specification 
+name: sweatshop
+version: !ruby/object:Gem::Version 
+  version: 1.4.0
+platform: ruby
+authors: 
+- Amos Elliston
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-10-19 00:00:00 -07:00
+default_executable: sweatd
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: famoseagle-carrot
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        version: 0.7.0
+    version: 
+description: See summary
+email: amos@geni.com
+executables: 
+- sweatd
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.markdown
+files: 
+- History.txt
+- LICENSE
+- README.markdown
+- Rakefile
+- VERSION.yml
+- config/defaults.yml
+- config/sweatshop.yml
+- install.rb
+- lib/message_queue/base.rb
+- lib/message_queue/kestrel.rb
+- lib/message_queue/rabbit.rb
+- lib/sweat_shop.rb
+- lib/sweat_shop/daemoned.rb
+- lib/sweat_shop/metaid.rb
+- lib/sweat_shop/sweatd.rb
+- lib/sweat_shop/worker.rb
+- script/initd.sh
+- script/kestrel
+- script/kestrel.sh
+- script/sweatshop
+- test/hello_worker.rb
+- test/test_functional_worker.rb
+- test/test_helper.rb
+- test/test_sweatshop.rb
+has_rdoc: true
+homepage: http://github.com/famoseagle/sweat-shop
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Sweatshop is a simple asynchronous worker queue build on top of rabbitmq/ampq
+test_files: 
+- test/hello_worker.rb
+- test/test_functional_worker.rb
+- test/test_helper.rb
+- test/test_sweatshop.rb