Allocate a new taskBatch for each batch of tasks in the WorkerManager

elodina · Jul 17, 2015 · 83504cf · 83504cf
1 parent 34c98ac
commit 83504cf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/workers.go b/workers.go
@@ -147,6 +147,7 @@ func (wm *WorkerManager) Stop() chan bool {
 
 func (wm *WorkerManager) startBatch(batch []*Message) {
 	inLock(&wm.stopLock, func() {
+		wm.currentBatch = newTaskBatch()
 		wm.batchOrder = make([]TaskId, 0)
 		for _, message := range batch {
 			topicPartition := TopicAndPartition{message.Topic, message.Partition}