Fix reference count bug in partition batcher

aditya-systems-hub · aditya-systems-hub · commit 8026535dd9ed · 2026-01-18T04:28:56.000+05:30
diff --git a/exporter/exporterhelper/internal/queuebatch/partition_batcher.go b/exporter/exporterhelper/internal/queuebatch/partition_batcher.go
@@ -136,7 +136,7 @@ func (qb *partitionBatcher) Consume(ctx context.Context, req request.Request, do
 		numRefs++
 	}
 	if numRefs > 1 {
-		done = newRefCountDone(done, int64(len(reqList)))
+		done = newRefCountDone(done, int64(numRefs))
 		if mergeSplitErr != nil {
 			done.OnDone(mergeSplitErr)
 		}

Original file line number	Diff line number	Diff line change
`@@ -136,7 +136,7 @@ func (qb *partitionBatcher) Consume(ctx context.Context, req request.Request, do`
`136`	`136`	`numRefs++`
`137`	`137`	`}`
`138`	`138`	`if numRefs > 1 {`
`139`		`- done = newRefCountDone(done, int64(len(reqList)))`
	`139`	`+ done = newRefCountDone(done, int64(numRefs))`
`140`	`140`	`if mergeSplitErr != nil {`
`141`	`141`	`done.OnDone(mergeSplitErr)`
`142`	`142`	`}`