sidekiq-recursive 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 1f8039623c6e6890c840050a797e2d2a4a336d0e
-  data.tar.gz: a34b1e3f4f7385f3286e0e6dffc997f54e6cd94e
+  metadata.gz: c314d036966c7977f1647dca8d10a139d2897a26
+  data.tar.gz: 87d913ed242b44275d845bf811521a71163854f6
 SHA512:
-  metadata.gz: 7d1b745ac1162cb03dc27c89d2e9e87674ff395b235d22ff75d3795d6da994470f4bf603cb3ff019fc684df1858b2a0d875bb45c3604be1c583e8ebd3f18bf18
-  data.tar.gz: 25f1d7fe4856aa606bb73935b3ae3580ff10e354bf6d12a1f096e2ba8bfb4af0bdccb85e923912528318a188737da6884913498fa78c3b47a70c42810a8e5386
+  metadata.gz: 3e0876def8a48e6cae0bf6201f616eb4a6d69a8a90ebbb47d8ef0188d43942ee9374a197c4e06c93e73ff98cdd4f56a8d98ee278949b6a0152f3aa1db327c485
+  data.tar.gz: 42305c2399da0bcd64de5766160a15a79967c2e5421218d28aa57f5b2369e8d4b15c69be51907d6b3e24126f954b46d8b57dc6cead6ae8de6b018c65c37c7334

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    sidekiq-recursive (0.1.0)
+    sidekiq-recursive (0.2.0)
       sidekiq (~> 5.1, >= 3.0)
 
 GEM

data/README.md

@@ -35,6 +35,30 @@ user_ids = User.where(is_old: true).ids
 CleanUpOldUsers.run(user_ids)
 ```
 
+#### before_all and after_all hooks
+```ruby
+class CleanUpOldUsers
+  include Sidekiq::Worker
+  include Sidekiq::Recursive::Worker
+
+  recursive_worker_count 5
+  before_all :bar
+  after_all :foo
+
+  def bar
+    # do something before starting all workers
+  end
+
+  def process(user_id)
+    User.destroy(user_id)
+  end
+
+  def foo
+    # do something after all workers are finished
+  end
+end
+```
+
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.

data/lib/sidekiq/recursive.rb

@@ -1,4 +1,7 @@
 require 'sidekiq/recursive/argument_queue'
+require 'sidekiq/recursive/hooks'
+require 'sidekiq/recursive/hooks/after_all'
+require 'sidekiq/recursive/hooks/before_all'
 require 'sidekiq/recursive/perform'
 require 'sidekiq/recursive/start'
 require 'sidekiq/recursive/version'

data/lib/sidekiq/recursive/hooks.rb

@@ -0,0 +1,2 @@
+module Sidekiq::Recursive::Hooks
+end

data/lib/sidekiq/recursive/hooks/after_all.rb

@@ -0,0 +1,9 @@
+class Sidekiq::Recursive::Hooks::AfterAll
+  def self.run(worker, worker_id)
+    action_name = worker.after_all
+    return unless action_name
+    return unless worker.recursive_worker_count == worker_id
+    worker.new.send(action_name)
+    true
+  end
+end

data/lib/sidekiq/recursive/hooks/before_all.rb

@@ -0,0 +1,8 @@
+class Sidekiq::Recursive::Hooks::BeforeAll
+  def self.run(worker)
+    action_name = worker.before_all
+    return unless action_name
+    worker.new.send(action_name)
+    true
+  end
+end

data/lib/sidekiq/recursive/perform.rb

@@ -4,7 +4,7 @@ class Sidekiq::Recursive::Perform
     return worker_instance.process(argument) if worker_id == :failed_worker
     safe_process(worker_instance, argument)
     next_argument = Sidekiq::Recursive::ArgumentQueue.pop(worker)
-    return unless next_argument
+    return Sidekiq::Recursive::Hooks::AfterAll.run(worker, worker_id) unless next_argument
     worker.perform_async(worker_id, next_argument)
 
     true

data/lib/sidekiq/recursive/start.rb

@@ -1,6 +1,7 @@
 class Sidekiq::Recursive::Start
   def self.run(worker, arguments)
     Sidekiq::Recursive::ArgumentQueue.push(worker, arguments)
+    Sidekiq::Recursive::Hooks::BeforeAll.run(worker)
 
     1.upto(worker.recursive_worker_count) do |worker_id|
       argument = Sidekiq::Recursive::ArgumentQueue.pop(worker)

data/lib/sidekiq/recursive/version.rb

@@ -1,5 +1,5 @@
 module Sidekiq
   module Recursive
-    VERSION = "0.1.0"
+    VERSION = '0.2.0'
   end
 end

data/lib/sidekiq/recursive/worker.rb

@@ -14,6 +14,14 @@ module Sidekiq::Recursive::Worker
       raise Sidekiq::Recursive::UndefinedWorkerCountError if count.nil? && @worker_count.nil?
       @worker_count ||= count
     end
+
+    def before_all(action_name = nil)
+      @before_all ||= action_name
+    end
+
+    def after_all(action_name = nil)
+      @after_all ||= action_name
+    end
   end
 
   module InstanceMethods

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sidekiq-recursive
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Justas Kazakauskas
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-03-31 00:00:00.000000000 Z
+date: 2018-04-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sidekiq
@@ -120,6 +120,9 @@ files:
 - bin/setup
 - lib/sidekiq/recursive.rb
 - lib/sidekiq/recursive/argument_queue.rb
+- lib/sidekiq/recursive/hooks.rb
+- lib/sidekiq/recursive/hooks/after_all.rb
+- lib/sidekiq/recursive/hooks/before_all.rb
 - lib/sidekiq/recursive/perform.rb
 - lib/sidekiq/recursive/start.rb
 - lib/sidekiq/recursive/version.rb