Reset fail timer on success

giongto35 · Dec 1, 2024 · 9caf45a · 9caf45a
1 parent 56e3ce3
commit 9caf45a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 11 deletions.
diff --git a/pkg/network/retry.go b/pkg/network/retry.go
@@ -13,14 +13,7 @@ func NewRetry() Retry {
 	return Retry{t: retry}
 }
 
-func (r *Retry) Fail() *Retry   { r.fail = true; time.Sleep(r.t); return r }
-func (r *Retry) Failed() bool   { return r.fail }
-func (r *Retry) Multiply(x int) { r.t *= time.Duration(x) }
-func (r *Retry) SuccessCheck() {
-	if r.fail {
-		return
-	}
-	r.t = retry
-	r.fail = false
-}
+func (r *Retry) Fail() *Retry        { r.fail = true; time.Sleep(r.t); return r }
+func (r *Retry) Multiply(x int)      { r.t *= time.Duration(x) }
+func (r *Retry) Success()            { r.t = retry; r.fail = false }
 func (r *Retry) Time() time.Duration { return r.t }
diff --git a/pkg/worker/worker.go b/pkg/worker/worker.go
@@ -126,7 +126,7 @@ func (w *Worker) Start(done chan struct{}) {
 				w.cord.SendLibrary(w)
 				w.cord.SendPrevSessions(w)
 				<-wait
-				retry.SuccessCheck()
+				retry.Success()
 			}
 		}
 	}()