internal.Retry now slightly faster

Danlock · Oct 14, 2023 · 26c1de3 · 26c1de3
1 parent 9171254
commit 26c1de3
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/internal/util.go b/internal/util.go
@@ -63,17 +63,17 @@ func Retry(ctx context.Context, delayForAttempt func(int) time.Duration, do func
 	var delay time.Duration
 	var attempt = 0
 	for {
-		select {
-		case <-ctx.Done():
-			return
-		case <-time.After(delay):
-		}
-
 		delay = delayForAttempt(attempt)
 		attempt++
 		lifetime, ok := do(delay)
 		if ok && lifetime >= healthyLifetime {
 			delay, attempt = 0, 0
 		}
+
+		select {
+		case <-ctx.Done():
+			return
+		case <-time.After(delay):
+		}
 	}
 }