sidekiq-apriori 0.0.1

data/.gitignore

@@ -0,0 +1,7 @@
+.bundle/
+*.gem
+Gemfile.lock
+pkg/
+coverage/
+*.swp
+spec/support/test.db

data/.rspec

@@ -0,0 +1 @@
+--color

data/.simplecov

@@ -0,0 +1,12 @@
+unless ENV['COVERAGE'] == 'false'
+  require 'simplecov-rcov'
+
+  class SimpleCov::Formatter::MergedFormatter
+    def format(result)
+      SimpleCov::Formatter::HTMLFormatter.new.format(result)
+      SimpleCov::Formatter::RcovFormatter.new.format(result)
+    end
+  end
+
+  SimpleCov.formatter = SimpleCov::Formatter::MergedFormatter
+end

data/.travis.yml

@@ -0,0 +1,5 @@
+language: ruby
+rvm:
+  - 2.0.0
+  - 1.9.3
+  - 1.8.7

data/Gemfile

@@ -0,0 +1,2 @@
+source "http://rubygems.org"
+gemspec

data/LICENSE

@@ -0,0 +1,20 @@
+The MIT License (MIT)
+
+Copyright (c) 2013 Blake Thomas
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of
+this software and associated documentation files (the "Software"), to deal in
+the Software without restriction, including without limitation the rights to
+use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+the Software, and to permit persons to whom the Software is furnished to do so,
+subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,4 @@
+sidekiq-apriori
+===============
+
+Prioritization Middleware for Sidekiq

data/Rakefile

@@ -0,0 +1,11 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks
+
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new
+task :default => :spec
+
+desc 'Start IRB with preloaded environment'
+task :console do
+  exec 'irb', "-I#{File.join(File.dirname(__FILE__), 'lib')}", '-rsidekiq-apriori'
+end

data/lib/sidekiq-apriori/arb.rb

@@ -0,0 +1,23 @@
+module Sidekiq::Apriori
+  def self.included(base)
+    return unless defined?(ActiveRecord::Base) && base < ActiveRecord::Base
+
+    base.extend ClassMethods
+
+    ## Add validation for priority attribute
+    if base.attribute_names.include?('priority')
+      base.validates_inclusion_of :priority, :in => PRIORITIES
+    end
+  end
+
+  module ClassMethods
+    ## Declarative hook to prioritize instances
+    def prioritize(options = {}, &block)
+      method = ( block_given? && [-1, 0].include?(block.arity) ) ?
+        block : ( options[:using] || options[:with] )
+
+      before_validation method, :on => :create
+    end
+  end
+
+end

data/lib/sidekiq-apriori/middleware.rb

@@ -0,0 +1,14 @@
+require 'sidekiq'
+require 'sidekiq-apriori/prioritizer'
+
+Sidekiq.configure_client do |config|
+  config.client_middleware do |chain|
+    chain.add Sidekiq::Apriori::Prioritizer
+  end
+end
+
+Sidekiq.configure_server do |config|
+  config.client_middleware do |chain|
+    chain.add Sidekiq::Apriori::Prioritizer
+  end
+end

data/lib/sidekiq-apriori/prioritizer.rb

@@ -0,0 +1,20 @@
+module Sidekiq::Apriori
+  class Prioritizer
+    def call(worker, msg, queue)
+      priority = msg["args"].try(:last)
+      if priorities.include?(priority)
+        msg["queue"] = queue.to_s.sub(priority_regexp,"_#{priority}")
+      end
+
+      yield
+    end
+
+    def priorities
+      Sidekiq::Apriori::PRIORITIES.compact
+    end
+
+    def priority_regexp
+      /(_(#{priorities.join('|')}))*$/
+    end
+  end
+end

data/lib/sidekiq-apriori/version.rb

@@ -0,0 +1,5 @@
+module Sidekiq
+  module Apriori
+    VERSION = "0.0.1"
+  end
+end

data/lib/sidekiq-apriori.rb

@@ -0,0 +1,12 @@
+require 'sidekiq-apriori/version'
+require 'sidekiq-apriori/middleware'
+require 'sidekiq-apriori/arb.rb'
+
+module Sidekiq::Apriori
+  PRIORITIES = [
+    'immediate',
+    'high',
+    nil,
+    'low'
+  ].freeze unless defined?(PRIORITIES)
+end

data/sidekiq-apriori.gemspec

@@ -0,0 +1,29 @@
+# -*- encoding: utf-8 -*-
+
+$:.push File.expand_path("../lib", __FILE__)
+require "sidekiq-apriori/version"
+
+Gem::Specification.new do |s|
+  s.name        = "sidekiq-apriori"
+  s.license     = 'MIT'
+  s.version     = Sidekiq::Apriori::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Blake Thomas", "Enova"]
+  s.email       = ["bwthomas@gmail.com"]
+  s.homepage    = "https://github.com/bwthomas/sidekiq-apriori"
+  s.summary     = %q{Prioritization Middleware for Sidekiq}
+  s.description = %q{Prioritization middleware for Sidekiq}
+  s.rubyforge_project = "nowarning"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_development_dependency "pry"
+  s.add_development_dependency "rake"
+  s.add_development_dependency "rspec"
+  s.add_development_dependency "simplecov"
+
+  s.add_dependency "sidekiq"
+end

data/spec/sidekiq-apriori_spec.rb

@@ -0,0 +1,60 @@
+require 'spec_helper'
+
+describe Sidekiq::Apriori do
+  before(:all) do
+    ## See spec/support/arb.rb for Arb class definition
+    class Prioritized < Arb
+      prioritize using: :some_method
+    end
+  end
+
+  let(:arb)           { Arb.new  }
+  let(:prioritized)   { Prioritized.new }
+
+  it "should validate priorities" do
+    arb.priority = "none"
+    arb.should be_invalid
+
+    Sidekiq::Apriori::PRIORITIES.each do |priority|
+      arb.priority = priority
+      arb.should be_valid
+    end
+  end
+
+  it "should attempt prioritization on creation with named prioritizer" do
+    prioritized.should_receive(:some_method)
+    prioritized.save
+  end
+
+  describe Sidekiq::Apriori::Prioritizer do
+    subject(:middleware)  { Sidekiq::Apriori::Prioritizer.new }
+
+    let(:message)         { {'args' => [1, nil, 'high'], 'queue' => 'foo'} }
+    let(:queue)           { 'foo' }
+
+    describe '#call' do
+      it 'should respond' do
+        middleware.respond_to?(:call).should be_true
+      end
+
+      it 'should require three arguments' do
+        expect { middleware.call }.
+          to raise_error( ArgumentError,'wrong number of arguments (0 for 3)')
+      end
+
+      it 'should set priority queue' do
+        middleware.call(nil, message, queue) {}
+        message['queue'].should eql('foo_high')
+      end
+
+      it 'should allow only one priority suffix' do
+        message['queue'] = 'foo_low_high_immediate'
+        middleware.call(nil, message, queue) {}
+        message['queue'].should eql('foo_high')
+
+        middleware.call(nil, message, queue) {}
+        message['queue'].should eql('foo_high')
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,10 @@
+require 'simplecov'
+
+SimpleCov.start do
+  add_filter "vendor"
+  add_filter "spec"
+end
+
+require File.expand_path("../../lib/sidekiq-apriori", __FILE__)
+
+require 'support/arb'

data/spec/support/arb.rb

@@ -0,0 +1,19 @@
+require 'active_record'
+require 'sqlite3'
+
+ActiveRecord::Base.establish_connection(
+  :adapter => :sqlite3,
+  :database => 'spec/support/test.db'
+)
+
+ActiveRecord::Schema.define do
+  drop_table :arbs
+
+  create_table :arbs do |t|
+    t.column :priority, :string
+  end
+end
+
+class Arb < ActiveRecord::Base
+  include Sidekiq::Apriori
+end

metadata

@@ -0,0 +1,148 @@
+--- !ruby/object:Gem::Specification
+name: sidekiq-apriori
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Blake Thomas
+- Enova
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-11-28 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: sidekiq
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Prioritization middleware for Sidekiq
+email:
+- bwthomas@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rspec
+- .simplecov
+- .travis.yml
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- lib/sidekiq-apriori.rb
+- lib/sidekiq-apriori/arb.rb
+- lib/sidekiq-apriori/middleware.rb
+- lib/sidekiq-apriori/prioritizer.rb
+- lib/sidekiq-apriori/version.rb
+- sidekiq-apriori.gemspec
+- spec/sidekiq-apriori_spec.rb
+- spec/spec_helper.rb
+- spec/support/arb.rb
+homepage: https://github.com/bwthomas/sidekiq-apriori
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: nowarning
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Prioritization Middleware for Sidekiq
+test_files:
+- spec/sidekiq-apriori_spec.rb
+- spec/spec_helper.rb
+- spec/support/arb.rb
+has_rdoc: