use time.Now for Delay/RetryAfter methods

theplant · Aug 19, 2024 · 094ebe1 · 094ebe1
1 parent 9162b04
commit 094ebe1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ratelimiter.go b/ratelimiter.go
@@ -40,7 +40,7 @@ func (r *Reservation) DelayFrom(t time.Time) time.Duration {
 }
 
 func (r *Reservation) Delay() time.Duration {
-	return r.DelayFrom(r.Now)
+	return r.DelayFrom(time.Now())
 }
 
 func (r *Reservation) RetryAfterFrom(t time.Time) time.Duration {
@@ -56,7 +56,7 @@ func (r *Reservation) RetryAfterFrom(t time.Time) time.Duration {
 }
 
 func (r *Reservation) RetryAfter() time.Duration {
-	return r.RetryAfterFrom(r.Now)
+	return r.RetryAfterFrom(time.Now())
 }
 
 type Driver interface {