hedge: fix retry logic in SendToLeader

flowerinthenight · Jul 20, 2024 · 0cfc4de · 0cfc4de
1 parent a352a0c
commit 0cfc4de
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/hedge.go b/hedge.go
@@ -1417,22 +1417,22 @@ func SendToLeader(ctx context.Context, op *Op, m []byte, args ...*SendToLeaderAr
 
 		var i int
 		for {
-			if i >= retries {
+			if i >= retries && retries >= 0 {
 				break
 			}
 
 			if !op.IsRunning() {
 				time.Sleep(bo.Pause())
 			}
 
-			if retries > -1 {
+			if retries >= 0 {
 				i++
 			}
 		}
 
 		i = 0
 		for {
-			if i >= retries {
+			if i >= retries && retries >= 0 {
 				err = fmt.Errorf("hedge: retries exhausted")
 				return
 			}
@@ -1446,7 +1446,7 @@ func SendToLeader(ctx context.Context, op *Op, m []byte, args ...*SendToLeaderAr
 				return
 			}
 
-			if retries > -1 {
+			if retries >= 0 {
 				i++
 			}
 		}