sidekiq-congestion 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: aa39832645b56750973469661ef9d904fe684edf
-  data.tar.gz: e85779be4e776e654db4a44ddfcc8f6dcd114f7b
+  metadata.gz: d982878cf7a24a459b3f0fab33a4a323a080c56a
+  data.tar.gz: 74d14a82bf221c29a1b760a9929baac3a6bb05ce
 SHA512:
-  metadata.gz: 618d7512f650c99938b29232e827a4be9cc31ea81b18aa49364b701cf0a038f377b9b205cbc027ff9c5750aa388ac358aca9776ca08f63ef512e85825f04d1b9
-  data.tar.gz: b7dea32db308cc935806b76dd9a38b93ff6e960a14690a551464aafad4fe7943377e68e8d6b1ac2e04db08a515149905d7d17a55671e864ebc08ac24c763dad0
+  metadata.gz: 24596d57dc5467fcf45af8c1286979f2558d37a538e452e4cda666024f9134b4a0c3087f4c8c7c7dcc433c5a031e99f4fd7741441523dab42259a11aa17b646c
+  data.tar.gz: d1513ae196e8a684ed90203b6597294e106a0a0925cc27a79bc9e19b7eed00dd92683a9ac4e2058b7c92cffc1476821a4edb93ca3970780d95a7a1be89ce362f

data/.travis.yml

@@ -2,10 +2,9 @@ language: ruby
 sudo: false
 cache: bundler
 rvm:
-  - 1.9.3
   - 2.2.0
   - jruby-19mode
 services:
   - redis-server
-jruby_before_script: &jruby_before_script
+jruby_before_script:
   - bundle exec jbundle install

data/README.md

@@ -3,6 +3,7 @@
 [![Build Status](https://travis-ci.org/parrish/Sidekiq-Congestion.svg?branch=master)](https://travis-ci.org/parrish/Sidekiq-Congestion)
 [![Test Coverage](https://codeclimate.com/github/parrish/Sidekiq-Congestion/badges/coverage.svg)](https://codeclimate.com/github/parrish/Sidekiq-Congestion)
 [![Code Climate](https://codeclimate.com/github/parrish/Sidekiq-Congestion/badges/gpa.svg)](https://codeclimate.com/github/parrish/Sidekiq-Congestion)
+[![Gem Version](https://badge.fury.io/rb/sidekiq-congestion.svg)](http://badge.fury.io/rb/sidekiq-congestion)
 
 Sidekiq middleware for [Congestion](https://github.com/parrish/Congestion)
 
@@ -26,7 +27,41 @@ Or install it yourself as:
 
 ## Usage
 
-TODO: Write usage instructions here
+[Documentation of Congestion configuration can be found here](https://github.com/parrish/Congestion#user-content-configuration)
+
+However, `Sidekiq::Congestion` disables rejection tracking by default.
+
+Rejection tracking would cause attempted calls to your workers (even if they don't trigger a run) to count towards the worker limits -- which is probably undesirable.  If your worker is high throughput, you may want to enable it just so the request tracking is [atomic](http://en.wikipedia.org/wiki/Linearizability).
+
+In an initializer:
+
+```ruby
+# Set whatever default options you'd like
+# Congestion.default_options[:track_rejected] = false
+
+Sidekiq.configure_server do |config|
+  config.server_middleware do |chain|
+    chain.add Sidekiq::Congestion::Limiter
+  end
+end
+```
+
+In a worker:
+
+```ruby
+class YourWorker
+  include Sidekiq::Worker
+
+  # Allow 5 calls/hour, with at least 5 minutes between calls
+  # When the request is not allowed, it is rescheduled
+  sidekiq_options congestion: {
+    interval: 1.hour,
+    max_in_interval: 5,
+    min_delay: 5.minutes,
+    reject_with: :reschedule # (or :cancel)
+  }
+end
+```
 
 ## Development
 

data/lib/sidekiq/congestion.rb

@@ -1,6 +1,14 @@
+require 'congestion'
+require 'sidekiq'
 require 'sidekiq/congestion/version'
 
 module Sidekiq
   module Congestion
+    require 'sidekiq/congestion/request'
+    require 'sidekiq/congestion/limiter'
+    ::Congestion.default_options[:track_rejected] = true
+    ::Congestion.redis = ->{
+      ::Sidekiq.redis{ |redis| redis }
+    }
   end
 end

data/lib/sidekiq/congestion/limiter.rb

@@ -0,0 +1,20 @@
+module Sidekiq
+  module Congestion
+    class Limiter
+      def call(worker, job, queue, &block)
+        request = Sidekiq::Congestion::Request.new worker, job, queue
+        request.enabled? ? handle(request, block) : yield
+      end
+
+      def handle(request, block)
+        if request.allowed?
+          block.call
+        elsif request.reschedule?
+          request.reschedule!
+        else
+          # cancel request
+        end
+      end
+    end
+  end
+end

data/lib/sidekiq/congestion/request.rb

@@ -0,0 +1,54 @@
+require 'forwardable'
+
+module Sidekiq
+  module Congestion
+    class Request
+      extend Forwardable
+      attr_accessor :worker, :args, :options
+
+      def_delegators :congestion,
+        :allowed?, :backoff, :first_request, :last_request,
+        :rejected?, :too_frequent?, :too_many?, :total_requests
+
+      def initialize(worker, job, queue)
+        self.worker = worker
+        self.args = job['args']
+        opts = worker.sidekiq_options_hash || { }
+        self.options = opts['congestion']
+      end
+
+      def enabled?
+        !!options
+      end
+
+      def reschedule?
+        rejection_method == :reschedule
+      end
+
+      def reschedule!
+        worker.class.perform_in backoff, *args
+      end
+
+      def key
+        @key ||= case options[:key]
+        when Proc
+          options[:key].call *args
+        when String, Symbol
+          options[:key].to_s
+        else
+          worker.class.name
+        end
+      end
+
+      def congestion
+        @congestion ||= ::Congestion.request key, options
+      end
+
+      protected
+
+      def rejection_method
+        options.fetch(:reject_with, :reschedule).to_sym
+      end
+    end
+  end
+end

data/lib/sidekiq/congestion/version.rb

@@ -1,5 +1,5 @@
 module Sidekiq
   module Congestion
-    VERSION = '0.0.1'
+    VERSION = '0.0.2'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sidekiq-congestion
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Michael Parrish
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2015-04-14 00:00:00.000000000 Z
+date: 2015-04-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: congestion
@@ -170,6 +170,8 @@ files:
 - bin/console
 - bin/setup
 - lib/sidekiq/congestion.rb
+- lib/sidekiq/congestion/limiter.rb
+- lib/sidekiq/congestion/request.rb
 - lib/sidekiq/congestion/version.rb
 - sidekiq-congestion.gemspec
 homepage: https://github.com/parrish/sidekiq-congestion