fix: witness loop

0xPolygon · Feb 4, 2025 · 990cbb3 · 990cbb3
1 parent 822782c
commit 990cbb3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/aggregator/aggregator.go b/aggregator/aggregator.go
@@ -1215,7 +1215,8 @@ func (a *Aggregator) getAndLockBatchToProve(
 
 	// Request the witness from the server, if it is busy just keep looping until it is available
 	start := time.Now()
-	witness, err := a.rpcClient.GetWitness(batchNumberToVerify, a.cfg.UseFullWitness)
+	var witness []byte
+	witness, err = a.rpcClient.GetWitness(batchNumberToVerify, a.cfg.UseFullWitness)
 	for err != nil {
 		if errors.Is(err, rpc.ErrBusy) {
 			a.logger.Debugf(
@@ -1226,6 +1227,7 @@ func (a *Aggregator) getAndLockBatchToProve(
 			a.logger.Errorf("Failed to get witness for batch %d, err: %v", batchNumberToVerify, err)
 		}
 		time.Sleep(a.cfg.RetryTime.Duration)
+		witness, err = a.rpcClient.GetWitness(batchNumberToVerify, a.cfg.UseFullWitness)
 	}
 	end := time.Now()
 	a.logger.Debugf("Time to get witness for batch %d: %v", batchNumberToVerify, end.Sub(start))