2021-02-01 12:29:04 -08:00
|
|
|
package backoff
|
2016-09-05 07:34:37 -07:00
|
|
|
|
|
|
|
import (
|
|
|
|
"math/rand"
|
|
|
|
"time"
|
|
|
|
)
|
|
|
|
|
|
|
|
// This one was ripped from https://github.com/jpillora/backoff/blob/master/backoff.go
|
|
|
|
|
|
|
|
// Backoff is a time.Duration counter. It starts at Min. After every
|
|
|
|
// call to Duration() it is multiplied by Factor. It is capped at
|
|
|
|
// Max. It returns to Min on every call to Reset(). Used in
|
|
|
|
// conjunction with the time package.
|
2021-02-01 12:29:04 -08:00
|
|
|
type Backoff struct {
|
2016-09-05 07:34:37 -07:00
|
|
|
attempts int
|
2019-09-07 13:46:58 -07:00
|
|
|
// Initial value to scale out
|
|
|
|
Initial time.Duration
|
|
|
|
// Jitter value randomizes an additional delay between 0 and Jitter
|
|
|
|
Jitter time.Duration
|
|
|
|
// Max maximum values of the backoff
|
|
|
|
Max time.Duration
|
2016-09-05 07:34:37 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// Returns the current value of the counter and then multiplies it
|
|
|
|
// Factor
|
2021-02-01 12:29:04 -08:00
|
|
|
func (b *Backoff) Duration() (dur time.Duration) {
|
2019-09-07 13:46:58 -07:00
|
|
|
// Zero-values are nonsensical, so we use
|
|
|
|
// them to apply defaults
|
2016-09-05 07:34:37 -07:00
|
|
|
if b.Max == 0 {
|
|
|
|
b.Max = 10 * time.Second
|
|
|
|
}
|
2019-09-07 13:46:58 -07:00
|
|
|
|
|
|
|
if b.Initial == 0 {
|
|
|
|
b.Initial = 100 * time.Millisecond
|
2016-09-05 07:34:37 -07:00
|
|
|
}
|
2019-09-07 13:46:58 -07:00
|
|
|
|
|
|
|
// calculate this duration
|
|
|
|
if dur = time.Duration(1 << uint(b.attempts)); dur > 0 {
|
|
|
|
dur = dur * b.Initial
|
|
|
|
} else {
|
|
|
|
dur = b.Max
|
2016-09-05 07:34:37 -07:00
|
|
|
}
|
2019-09-07 13:46:58 -07:00
|
|
|
|
|
|
|
if b.Jitter > 0 {
|
|
|
|
dur = dur + time.Duration(rand.Intn(int(b.Jitter)))
|
2016-09-05 07:34:37 -07:00
|
|
|
}
|
2019-09-07 13:46:58 -07:00
|
|
|
|
|
|
|
// bump attempts count
|
2016-09-05 07:34:37 -07:00
|
|
|
b.attempts++
|
2019-09-07 13:46:58 -07:00
|
|
|
|
|
|
|
return dur
|
2016-09-05 07:34:37 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
//Resets the current value of the counter back to Min
|
2021-02-01 12:29:04 -08:00
|
|
|
func (b *Backoff) Reset() {
|
2016-09-05 07:34:37 -07:00
|
|
|
b.attempts = 0
|
|
|
|
}
|
2021-02-01 12:29:04 -08:00
|
|
|
|
|
|
|
// Attempts returns the number of attempts that we had done so far
|
|
|
|
func (b *Backoff) Attempts() int {
|
|
|
|
return b.attempts
|
|
|
|
}
|