fix some parent issues

breamware · Dec 17, 2022 · c9babda · c9babda
1 parent 4d90b32
commit c9babda
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/sidekiq/batch.rb b/lib/sidekiq/batch.rb
@@ -62,7 +62,7 @@ def jobs
 
       begin
         if !@existing && !@initialized
-          parent_bid, Thread.current[:parent_id] = Thread.current[:batch].bid if Thread.current[:batch]
+          parent_bid = Thread.current[:batch].bid if Thread.current[:batch]
 
           Sidekiq.redis do |r|
             r.multi do |pipeline|
@@ -84,9 +84,11 @@ def jobs
         begin
           parent = Thread.current[:batch]
           Thread.current[:batch] = self
+          Thread.current[:parent_bid] = parent_bid
           yield
         ensure
           Thread.current[:batch] = parent
+          Thread.current[:parent_bid] = nil
         end
 
         return [] if @queued_jids.size == 0
@@ -119,7 +121,7 @@ def increment_job_queue(jid)
 
     def conditional_redis_increment!(force=false)
       if should_increment? || force
-        parent_bid = Thread.current[:parent_id]
+        parent_bid = Thread.current[:parent_bid]
         Sidekiq.redis do |r|
           r.multi do |pipeline|
             if parent_bid