Increase child batch count as new batches are enqueued, even in existing batches

Author: jpcamara

lib/solid_queue/batch.rb

@@ -68,8 +68,7 @@ def enqueue_batch(buffer)
         if batch_record.new_record?
           enqueue_new_batch(buffer)
         else
-          jobs = buffer.jobs.values
-          enqueue_existing_batch(jobs)
+          enqueue_existing_batch(buffer)
         end
       end
 
@@ -101,8 +100,9 @@ def enqueue_new_batch(buffer)
         end
       end
 
-      def enqueue_existing_batch(active_jobs)
-        jobs = Array.wrap(active_jobs)
+      def enqueue_existing_batch(buffer)
+        jobs = buffer.jobs.values
+        new_child_batches = buffer.child_batches.size
 
         SolidQueue::BatchRecord.transaction do
           enqueued_count = SolidQueue::Job.enqueue_all(jobs)
@@ -111,8 +111,8 @@ def enqueue_existing_batch(active_jobs)
           SolidQueue::BatchExecution.track_job_creation(persisted_jobs, batch_id)
 
           BatchRecord.where(batch_id: batch_id).update_all([
-            "total_jobs = total_jobs + ?, pending_jobs = pending_jobs + ?",
-            enqueued_count, enqueued_count
+            "total_jobs = total_jobs + ?, pending_jobs = pending_jobs + ?, total_child_batches = total_child_batches + ?",
+            enqueued_count, enqueued_count, new_child_batches
           ])
         end