sidekiq 0.5.0

data/lib/sidekiq/middleware.rb

@@ -0,0 +1,89 @@
+module Sidekiq
+  # Middleware is code configured to run before/after
+  # a message is processed.  It is patterned after Rack
+  # middleware.  The default middleware chain:
+  #
+  # Sidekiq::Middleware::Chain.register do
+  #   use Sidekiq::Airbrake
+  #   use Sidekiq::ActiveRecord
+  # end
+  #
+  # This is an example of a minimal middleware:
+  #
+  # class MyHook
+  #   def initialize(options=nil)
+  #   end
+  #   def call(worker, msg)
+  #     puts "Before work"
+  #     yield
+  #     puts "After work"
+  #   end
+  # end
+  #
+  module Middleware
+    class Chain
+      def self.register(&block)
+        instance_exec(&block)
+      end
+
+      def self.default
+        @default ||= [Entry.new(Airbrake), Entry.new(ActiveRecord)]
+      end
+
+      def self.use(klass, *args)
+        chain << Entry.new(klass, args)
+      end
+
+      def self.chain
+        @chain ||= default
+      end
+
+      def self.retrieve
+        Thread.current[:sidekiq_chain] ||= chain.map { |entry| entry.make_new }
+      end
+    end
+
+    class Entry
+      attr_reader :klass
+      def initialize(klass, args = [])
+        @klass = klass
+        @args  = args
+      end
+
+      def make_new
+        @klass.new(*@args)
+      end
+    end
+  end
+
+  class Airbrake
+    def initialize(options=nil)
+    end
+
+    def call(worker, msg)
+      yield
+    rescue => ex
+      send_to_airbrake(msg, ex) if defined?(::Airbrake)
+      raise
+    end
+
+    private
+
+    def send_to_airbrake(msg, ex)
+      ::Airbrake.notify(:error_class   => ex.class.name,
+                        :error_message => "#{ex.class.name}: #{ex.message}",
+                        :parameters    => msg)
+    end
+  end
+
+  class ActiveRecord
+    def initialize(options=nil)
+    end
+
+    def call(*)
+      yield
+    ensure
+      ActiveRecord::Base.clear_active_connections! if defined?(::ActiveRecord)
+    end
+  end
+end

data/lib/sidekiq/processor.rb

@@ -0,0 +1,37 @@
+require 'sidekiq/util'
+require 'sidekiq/middleware'
+require 'celluloid'
+
+module Sidekiq
+  class Processor
+    include Util
+    include Celluloid
+
+    def initialize(boss)
+      @boss = boss
+    end
+
+    def process(msg)
+      klass = constantize(msg['class'])
+      invoke_chain(klass.new, msg)
+      @boss.processor_done!(current_actor)
+    end
+
+    def invoke_chain(worker, msg)
+      chain = Sidekiq::Middleware::Chain.retrieve.dup
+      traverse_chain = lambda do
+        if chain.empty?
+          worker.perform(*msg['args'])
+        else
+          chain.shift.call(worker, msg, &traverse_chain)
+        end
+      end
+      traverse_chain.call
+    end
+
+    # See http://github.com/tarcieri/celluloid/issues/22
+    def inspect
+      "Sidekiq::Processor<#{object_id}>"
+    end
+  end
+end

data/lib/sidekiq/rails.rb

@@ -0,0 +1,5 @@
+module Sidekiq
+  class Rails < ::Rails::Engine
+    config.autoload_paths << File.expand_path("#{config.root}/app/workers") if File.exist?("#{config.root}/app/workers")
+  end
+end

data/lib/sidekiq/util.rb

@@ -0,0 +1,36 @@
+module Sidekiq
+  module Util
+
+    def constantize(camel_cased_word)
+      names = camel_cased_word.split('::')
+      names.shift if names.empty? || names.first.empty?
+
+      constant = Object
+      names.each do |name|
+        constant = constant.const_defined?(name) ? constant.const_get(name) : constant.const_missing(name)
+      end
+      constant
+    end
+
+    def watchdog(last_words)
+      yield
+    rescue => ex
+      STDERR.puts last_words
+      STDERR.puts ex
+      STDERR.puts ex.backtrace.join("\n")
+    end
+
+    def err(msg)
+      STDERR.puts(msg)
+    end
+
+    def log(msg)
+      STDOUT.puts(msg) unless $TESTING
+    end
+
+    def verbose(msg)
+      STDOUT.puts(msg) if $DEBUG
+    end
+
+  end
+end

data/lib/sidekiq/version.rb

@@ -0,0 +1,3 @@
+module Sidekiq
+  VERSION = "0.5.0"
+end

data/lib/sidekiq/worker.rb

@@ -0,0 +1,42 @@
+require 'sidekiq/client'
+
+module Sidekiq
+
+  ##
+  # Include this module in your worker class and you can easily create
+  # asynchronous jobs:
+  #
+  # class HardWorker
+  #   include Sidekiq::Worker
+  #
+  #   def perform(*args)
+  #     # do some work
+  #   end
+  # end
+  #
+  # Then in your Rails app, you can do this:
+  #
+  #   HardWorker.perform_async(1, 2, 3)
+  #
+  # Note that perform_async is a class method, perform is an instance method.
+  module Worker
+    def self.included(base)
+      base.extend(ClassMethods)
+    end
+
+    def info(msg)
+      print "#{msg}\n"
+    end
+    alias_method :log, :info
+
+    def debug(msg)
+      print "#{msg}\n" if $DEBUG
+    end
+
+    module ClassMethods
+      def perform_async(*args)
+        Sidekiq::Client.push('class' => self.name, 'args' => args)
+      end
+    end
+  end
+end

data/myapp/.gitignore

@@ -0,0 +1,15 @@
+# See http://help.github.com/ignore-files/ for more about ignoring files.
+#
+# If you find yourself ignoring temporary files generated by your text editor
+# or operating system, you probably want to add a global ignore instead:
+#   git config --global core.excludesfile ~/.gitignore_global
+
+# Ignore bundler config
+/.bundle
+
+# Ignore the default SQLite database.
+/db/*.sqlite3
+
+# Ignore all logfiles and tempfiles.
+/log/*.log
+/tmp

data/myapp/Gemfile

@@ -0,0 +1,5 @@
+source 'https://rubygems.org'
+
+gem 'rails', '3.2.0'
+gem 'sqlite3'
+gem 'sidekiq', :path => '..'

data/myapp/Gemfile.lock

@@ -0,0 +1,100 @@
+PATH
+  remote: ..
+  specs:
+    sidekiq (0.1.0)
+      celluloid
+      connection_pool
+      multi_json
+      redis
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    actionmailer (3.2.0)
+      actionpack (= 3.2.0)
+      mail (~> 2.4.0)
+    actionpack (3.2.0)
+      activemodel (= 3.2.0)
+      activesupport (= 3.2.0)
+      builder (~> 3.0.0)
+      erubis (~> 2.7.0)
+      journey (~> 1.0.0)
+      rack (~> 1.4.0)
+      rack-cache (~> 1.1)
+      rack-test (~> 0.6.1)
+      sprockets (~> 2.1.2)
+    activemodel (3.2.0)
+      activesupport (= 3.2.0)
+      builder (~> 3.0.0)
+    activerecord (3.2.0)
+      activemodel (= 3.2.0)
+      activesupport (= 3.2.0)
+      arel (~> 3.0.0)
+      tzinfo (~> 0.3.29)
+    activeresource (3.2.0)
+      activemodel (= 3.2.0)
+      activesupport (= 3.2.0)
+    activesupport (3.2.0)
+      i18n (~> 0.6)
+      multi_json (~> 1.0)
+    arel (3.0.0)
+    builder (3.0.0)
+    celluloid (0.7.2)
+    connection_pool (0.1.0)
+    erubis (2.7.0)
+    hike (1.2.1)
+    i18n (0.6.0)
+    journey (1.0.0)
+    json (1.6.5)
+    mail (2.4.1)
+      i18n (>= 0.4.0)
+      mime-types (~> 1.16)
+      treetop (~> 1.4.8)
+    mime-types (1.17.2)
+    multi_json (1.0.4)
+    polyglot (0.3.3)
+    rack (1.4.1)
+    rack-cache (1.1)
+      rack (>= 0.4)
+    rack-ssl (1.3.2)
+      rack
+    rack-test (0.6.1)
+      rack (>= 1.0)
+    rails (3.2.0)
+      actionmailer (= 3.2.0)
+      actionpack (= 3.2.0)
+      activerecord (= 3.2.0)
+      activeresource (= 3.2.0)
+      activesupport (= 3.2.0)
+      bundler (~> 1.0)
+      railties (= 3.2.0)
+    railties (3.2.0)
+      actionpack (= 3.2.0)
+      activesupport (= 3.2.0)
+      rack-ssl (~> 1.3.2)
+      rake (>= 0.8.7)
+      rdoc (~> 3.4)
+      thor (~> 0.14.6)
+    rake (0.9.2.2)
+    rdoc (3.12)
+      json (~> 1.4)
+    redis (2.2.2)
+    sprockets (2.1.2)
+      hike (~> 1.2)
+      rack (~> 1.0)
+      tilt (~> 1.1, != 1.3.0)
+    sqlite3 (1.3.5)
+    thor (0.14.6)
+    tilt (1.3.3)
+    treetop (1.4.10)
+      polyglot
+      polyglot (>= 0.3.1)
+    tzinfo (0.3.31)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  rails (= 3.2.0)
+  sidekiq!
+  sqlite3

data/myapp/Rakefile

@@ -0,0 +1,7 @@
+#!/usr/bin/env rake
+# Add your own tasks in files placed in lib/tasks ending in .rake,
+# for example lib/tasks/capistrano.rake, and they will automatically be available to Rake.
+
+require File.expand_path('../config/application', __FILE__)
+
+Myapp::Application.load_tasks

data/myapp/app/controllers/application_controller.rb

@@ -0,0 +1,3 @@
+class ApplicationController < ActionController::Base
+  protect_from_forgery
+end

data/myapp/app/helpers/application_helper.rb

@@ -0,0 +1,2 @@
+module ApplicationHelper
+end

data/myapp/app/models/post.rb

@@ -0,0 +1,2 @@
+class Post < ActiveRecord::Base
+end

data/myapp/app/views/layouts/application.html.erb

@@ -0,0 +1,14 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>Myapp</title>
+  <%= stylesheet_link_tag    "application", :media => "all" %>
+  <%= javascript_include_tag "application" %>
+  <%= csrf_meta_tags %>
+</head>
+<body>
+
+<%= yield %>
+
+</body>
+</html>

data/myapp/app/workers/hard_worker.rb

@@ -0,0 +1,6 @@
+class HardWorker
+  def perform(name, count)
+    sleep 0.01
+    puts 'done'
+  end
+end

data/myapp/config.ru

@@ -0,0 +1,4 @@
+# This file is used by Rack-based servers to start the application.
+
+require ::File.expand_path('../config/environment',  __FILE__)
+run Myapp::Application

data/myapp/config/application.rb

@@ -0,0 +1,59 @@
+require File.expand_path('../boot', __FILE__)
+
+require 'rails/all'
+
+if defined?(Bundler)
+  # If you precompile assets before deploying to production, use this line
+  Bundler.require(*Rails.groups(:assets => %w(development test)))
+  # If you want your assets lazily compiled in production, use this line
+  # Bundler.require(:default, :assets, Rails.env)
+end
+
+module Myapp
+  class Application < Rails::Application
+    # Settings in config/environments/* take precedence over those specified here.
+    # Application configuration should go into files in config/initializers
+    # -- all .rb files in that directory are automatically loaded.
+
+    # Custom directories with classes and modules you want to be autoloadable.
+    # config.autoload_paths += %W(#{config.root}/extras)
+
+    # Only load the plugins named here, in the order given (default is alphabetical).
+    # :all can be used as a placeholder for all plugins not explicitly named.
+    # config.plugins = [ :exception_notification, :ssl_requirement, :all ]
+
+    # Activate observers that should always be running.
+    # config.active_record.observers = :cacher, :garbage_collector, :forum_observer
+
+    # Set Time.zone default to the specified zone and make Active Record auto-convert to this zone.
+    # Run "rake -D time" for a list of tasks for finding time zone names. Default is UTC.
+    # config.time_zone = 'Central Time (US & Canada)'
+
+    # The default locale is :en and all translations from config/locales/*.rb,yml are auto loaded.
+    # config.i18n.load_path += Dir[Rails.root.join('my', 'locales', '*.{rb,yml}').to_s]
+    # config.i18n.default_locale = :de
+
+    # Configure the default encoding used in templates for Ruby 1.9.
+    config.encoding = "utf-8"
+
+    # Configure sensitive parameters which will be filtered from the log file.
+    config.filter_parameters += [:password]
+
+    # Use SQL instead of Active Record's schema dumper when creating the database.
+    # This is necessary if your schema can't be completely dumped by the schema dumper,
+    # like if you have constraints or database-specific column types
+    # config.active_record.schema_format = :sql
+
+    # Enforce whitelist mode for mass assignment.
+    # This will create an empty whitelist of attributes available for mass-assignment for all models
+    # in your app. As such, your models will need to explicitly whitelist or blacklist accessible
+    # parameters by using an attr_accessible or attr_protected declaration.
+    # config.active_record.whitelist_attributes = true
+
+    # Enable the asset pipeline
+    config.assets.enabled = true
+
+    # Version of your assets, change this if you want to expire all your assets
+    config.assets.version = '1.0'
+  end
+end