sidekiq-queue-pause 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 8c930391c65c28ca0697a4b9dd5eeaad27ced197
+  data.tar.gz: eb5a3a8a3c11c452a58b1a7afea09c484ddf0ba9
+SHA512:
+  metadata.gz: 70551e734828c25b9bc7d57230dcdd8c1ab089279fca2c1250ffc4afec377d6a9f660496b84f51c383999d28fdb9379df1b772c53876a1891a2dc6ba1650e843
+  data.tar.gz: 7b57627868a2830c47eabbbd9d10f17e08b0493a1e0d5e70fd4823fc62e1c5f3621895bcf2f7db5fe924d71d7adc2b25e44e57b1619611907ef287946532f3e2

data/lib/sidekiq-queue-pause.rb

@@ -0,0 +1,84 @@
+require 'celluloid'
+require 'sidekiq'
+require 'sidekiq/fetch'
+
+module Sidekiq
+  module QueuePause
+    PREFIX = 'queue_pause'
+
+    class << self
+      attr_accessor :retry_after
+      attr_writer   :process_key
+
+      def process_key(&block)
+        if block_given?
+          @process_key = block
+        else
+          @process_key.is_a?(Proc) ? @process_key.call : @process_key
+        end
+      end
+
+      def pause(queue, pkey = nil)
+        Sidekiq.redis { |it| it.set rkey(queue, pkey), true }
+      end
+
+      def unpause(queue, pkey = nil)
+        Sidekiq.redis { |it| it.del rkey(queue, pkey) }
+      end
+
+      def paused?(queue, pkey = nil)
+        Sidekiq.redis { |it| it.exists rkey(queue, pkey) }
+      end
+
+      def unpause_all
+        Sidekiq.redis { |it| it.keys("#{PREFIX}:*").each { |k| it.del k } }
+      end
+
+      private
+
+      def rkey(queue, pkey)
+        pkey ? "#{PREFIX}:#{queue}:#{pkey}" : "#{PREFIX}:#{queue}"
+      end
+    end
+
+    class PausingFetch < Sidekiq::BasicFetch
+      def retrieve_work
+        qcmd = unpaused_queues_cmd
+
+        if qcmd.size > 1
+          retrieve_work_for_queues qcmd
+        else
+          sleep(Sidekiq::QueuePause.retry_after || Sidekiq::Fetcher::TIMEOUT)
+          nil
+        end
+      end
+
+      def retrieve_work_for_queues(qcmd)
+        work = Sidekiq.redis { |conn| conn.brpop(*qcmd) }
+        UnitOfWork.new(*work) if work
+      end
+
+      def unpaused_queues_cmd
+        queues = queues_cmd
+        queues.reject do |q|
+          q != Sidekiq::Fetcher::TIMEOUT &&
+            Sidekiq::QueuePause.paused?(q.gsub('queue:', ''), Sidekiq::QueuePause.process_key)
+        end
+      end
+    end
+  end
+
+  class Queue
+    def pause(pkey = nil)
+      Sidekiq::QueuePause.pause(name, pkey)
+    end
+
+    def unpause(pkey = nil)
+      Sidekiq::QueuePause.unpause(name, pkey)
+    end
+
+    def paused?(pkey = nil)
+      Sidekiq::QueuePause.paused?(name, pkey)
+    end
+  end
+end

metadata

@@ -0,0 +1,59 @@
+--- !ruby/object:Gem::Specification
+name: sidekiq-queue-pause
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+platform: ruby
+authors:
+- FlavourSys Technology GmbH
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-10-21 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sidekiq
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.1'
+description: Let's you pause/unpause individual sidekiq queues.
+email:
+- technology@flavoursys.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/sidekiq-queue-pause.rb
+homepage: http://github.com/FlavourSys/sidekiq-queue-pause
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Pause a Sidekiq queue
+test_files: []