sidekiq-group 0.1.7 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fb1e0ba3a53c5a009c265ceabc99fc5153be564c12ed6c30f30e13040fc304f6
-  data.tar.gz: 1d99d5a585a6b37e0fae401cf06a6198bc47b017a1ecae620d7659d642356573
+  metadata.gz: c01687c2cafb914630079a57d1924310a4627b01f3175e4eb8f926531d0d710c
+  data.tar.gz: ce7e0d3ff94646c2758f4b44e8a8e009bc72711b85cb1e1f06380b63a6b8b39a
 SHA512:
-  metadata.gz: 949c2e7b90cee062a5bbd4d00ad9236a77c8432fd96fc7f14006067af0c9b8862d05bb4d126d672029250fe96021c8f3769ab8a3e24d9438f7ea4600da917f81
-  data.tar.gz: e6b7a7cc4c5d90b1bca83f510eaa47db8d63dcdf3eb2e30b8e3a7fdd5ec3c04fba907446dc01f9ddeb170d078e95791ff21c40fcfa17e1643a54e6001bfe2165
+  metadata.gz: 28a17e1b54c0386a8a7edd0c4633cfdb92076d1bcd979cf7ec9fbd027257fb88d32950eadbacba0932d4b000748fb151cc1362040b1606ef7df8572271e90735
+  data.tar.gz: 30d4868239cbbab6878f7af54a045bc02d03922bf0f0490430360f0ad4eab741125753717a81c0346697541e077ee14b89e9f0218ee1fe308b1a49fe608ccd59

data/lib/sidekiq/group/collection.rb

@@ -8,6 +8,7 @@ module Sidekiq
       LOCK_TTL = 3600
 
       attr_reader :cid, :callback_class, :callback_options
+      alias group_id cid
 
       def initialize(cid = nil)
         @cid = cid || SecureRandom.urlsafe_base64(16)
@@ -23,13 +24,18 @@ module Sidekiq
         persist('callback_options', value.to_json)
       end
 
+      def initialize_total_value
+        persist('total', 0)
+      end
+
       def add(jid)
-        Sidekiq::Logging.logger.info "Scheduling child job #{jid} for parent #{@cid}" if Sidekiq::Group.debug
+        Sidekiq.logger.info "Scheduling child job #{jid} for parent #{@cid}" if Sidekiq::Group.debug
 
         Sidekiq.redis do |r|
           r.multi do |pipeline|
             pipeline.sadd("#{@cid}-jids", jid)
             pipeline.expire("#{@cid}-jids", CID_EXPIRE_TTL)
+            pipeline.hincrby(@cid, 'total', 1)
           end
         end
       end
@@ -47,20 +53,32 @@ module Sidekiq
         callback_class, callback_options = callback_data
         options = JSON(callback_options)
 
-        Sidekiq::Logging.logger.info "Scheduling callback job #{callback_class} with #{options}" if Sidekiq::Group.debug
+        Sidekiq.logger.info "Scheduling callback job #{callback_class} with #{options}" if Sidekiq::Group.debug
         Sidekiq::Group::Worker.perform_async(callback_class, options)
 
         cleanup_redis
       end
 
+      def total
+        return unless spawned_all_jobs?
+
+        Sidekiq.redis { |r| r.hget(@cid, 'total').to_i }
+      end
+
+      def processed
+        return unless spawned_all_jobs?
+
+        total - pending
+      end
+
       private
 
       def remove_processed(jid)
-        Sidekiq::Logging.logger.info "Child job #{jid} completed" if Sidekiq::Group.debug
+        Sidekiq.logger.info "Child job #{jid} completed" if Sidekiq::Group.debug
 
         return if Sidekiq.redis { |r| r.srem("#{@cid}-jids", jid) }
 
-        Sidekiq::Logging.logger.info "Could not remove child job #{jid} from Redis" if Sidekiq::Group.debug
+        Sidekiq.logger.info "Could not remove child job #{jid} from Redis" if Sidekiq::Group.debug
         sleep 1
         Sidekiq.redis { |r| r.srem("#{@cid}-jids", jid) }
       end
@@ -70,7 +88,7 @@ module Sidekiq
       end
 
       def processed_all_jobs?
-        Sidekiq::Logging.logger.info "Pending jobs: #{pending}" if Sidekiq::Group.debug
+        Sidekiq.logger.info "Pending jobs: #{pending}" if Sidekiq::Group.debug
 
         spawned_all_jobs? && pending.zero?
       end

data/lib/sidekiq/group/version.rb

@@ -1,5 +1,5 @@
 module Sidekiq
   module Group
-    VERSION = '0.1.7'.freeze
+    VERSION = '0.1.8'.freeze
   end
 end

data/lib/sidekiq/group.rb

@@ -8,6 +8,12 @@ module Sidekiq
 
     class << self
       attr_accessor :debug
+
+      def progress(group_id)
+        group = Sidekiq::Group::Collection.new(group_id)
+
+        { total: group.total, processed: group.processed }
+      end
     end
 
     def sidekiq_group(options = {})
@@ -16,6 +22,7 @@ module Sidekiq
       group = Sidekiq::Group::Collection.new
       group.callback_class = self.class.name
       group.callback_options = options
+      group.initialize_total_value
 
       Thread.current[:group_collection] = group
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sidekiq-group
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.8
 platform: ruby
 authors:
 - Matic developers
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-05-11 00:00:00.000000000 Z
+date: 2022-08-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport