fix loop

0xPolygonHermez · Aug 13, 2024 · 57ace4c · 57ace4c
1 parent 9c639f1
commit 57ace4c
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/aggregator/aggregator.go b/aggregator/aggregator.go
@@ -182,6 +182,8 @@ func New(
 		witnessRetrievalChan:    make(chan *state.DBBatch, cfg.MaxWitnessRetrievalWorkers),
 	}
 
+	log.Infof("MaxWitnessRetrievalWorkers set to %d", cfg.MaxWitnessRetrievalWorkers)
+
 	// Set function to handle the batches from the data stream
 	a.streamClient.SetProcessEntryFunc(a.handleReceivedDataStream)
 	a.l1Syncr.SetCallbackOnReorgDone(a.handleReorg)
@@ -190,11 +192,13 @@ func New(
 }
 
 func (a *Aggregator) retrieveWitnesses() {
-	select {
-	case <-a.ctx.Done():
-		return
-	case dbBatch := <-a.witnessRetrievalChan:
-		go a.retrieveWitness(dbBatch)
+	for {
+		select {
+		case <-a.ctx.Done():
+			return
+		case dbBatch := <-a.witnessRetrievalChan:
+			go a.retrieveWitness(dbBatch)
+		}
 	}
 }