forked from mirror/backoff
Make ForAttempt concurrent-safe in all cases, golint.
This commit is contained in:
parent
a6f64285fd
commit
4dd47ae5ac
70
backoff.go
70
backoff.go
|
@ -1,3 +1,4 @@
|
||||||
|
// Package backoff provides an exponential-backoff implementation.
|
||||||
package backoff
|
package backoff
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
@ -6,14 +7,12 @@ import (
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
//Backoff is a time.Duration counter. It starts at Min.
|
// Backoff is a time.Duration counter, starting at Min. After every call to
|
||||||
//After every call to Duration() it is multiplied by Factor.
|
// the Duration method the current timing is multiplied by Factor, but it
|
||||||
//It is capped at Max. It returns to Min on every call to Reset().
|
// never exceeds Max.
|
||||||
//Used in conjunction with the time package.
|
|
||||||
//
|
//
|
||||||
// Backoff is not threadsafe, but the ForAttempt method can be
|
// Backoff is not generally concurrent-safe, but the ForAttempt method can
|
||||||
// used concurrently if non-zero values for Factor, Max, and Min
|
// be used concurrently.
|
||||||
// are set on the Backoff shared among threads.
|
|
||||||
type Backoff struct {
|
type Backoff struct {
|
||||||
//Factor is the multiplying factor for each increment step
|
//Factor is the multiplying factor for each increment step
|
||||||
attempt, Factor float64
|
attempt, Factor float64
|
||||||
|
@ -23,8 +22,8 @@ type Backoff struct {
|
||||||
Min, Max time.Duration
|
Min, Max time.Duration
|
||||||
}
|
}
|
||||||
|
|
||||||
//Returns the current value of the counter and then
|
// Duration returns the duration for the current attempt before incrementing
|
||||||
//multiplies it Factor
|
// the attempt counter. See ForAttempt.
|
||||||
func (b *Backoff) Duration() time.Duration {
|
func (b *Backoff) Duration() time.Duration {
|
||||||
d := b.ForAttempt(b.attempt)
|
d := b.ForAttempt(b.attempt)
|
||||||
b.attempt++
|
b.attempt++
|
||||||
|
@ -36,44 +35,47 @@ func (b *Backoff) Duration() time.Duration {
|
||||||
// unnecessary memory storing the Backoff parameters per Backoff. The first
|
// unnecessary memory storing the Backoff parameters per Backoff. The first
|
||||||
// attempt should be 0.
|
// attempt should be 0.
|
||||||
//
|
//
|
||||||
// ForAttempt is threadsafe iff non-zero values for Factor, Max, and Min
|
// ForAttempt is concurrent-safe.
|
||||||
// are set before any calls to ForAttempt are made.
|
|
||||||
func (b *Backoff) ForAttempt(attempt float64) time.Duration {
|
func (b *Backoff) ForAttempt(attempt float64) time.Duration {
|
||||||
if float64(b.Min) > float64(b.Max) {
|
// Zero-values are nonsensical, so we use
|
||||||
return b.Max
|
// them to apply defaults
|
||||||
|
min := b.Min
|
||||||
|
if min <= 0 {
|
||||||
|
min = 100 * time.Millisecond
|
||||||
|
}
|
||||||
|
max := b.Max
|
||||||
|
if max <= 0 {
|
||||||
|
max = 10 * time.Second
|
||||||
|
}
|
||||||
|
if min >= max {
|
||||||
|
// short-circuit
|
||||||
|
return max
|
||||||
}
|
}
|
||||||
|
|
||||||
//Zero-values are nonsensical, so we use
|
factor := b.Factor
|
||||||
//them to apply defaults
|
if factor <= 0 {
|
||||||
if b.Min == 0 {
|
factor = 2
|
||||||
b.Min = 100 * time.Millisecond
|
|
||||||
}
|
}
|
||||||
if b.Max == 0 {
|
|
||||||
b.Max = 10 * time.Second
|
|
||||||
}
|
|
||||||
if b.Factor == 0 {
|
|
||||||
b.Factor = 2
|
|
||||||
}
|
|
||||||
|
|
||||||
//calculate this duration
|
//calculate this duration
|
||||||
dur := float64(b.Min) * math.Pow(b.Factor, attempt)
|
minf := float64(min)
|
||||||
if b.Jitter == true {
|
durf := minf * math.Pow(factor, attempt)
|
||||||
dur = rand.Float64()*(dur-float64(b.Min)) + float64(b.Min)
|
if b.Jitter {
|
||||||
|
durf = rand.Float64()*(durf-minf) + minf
|
||||||
}
|
}
|
||||||
//cap!
|
dur := time.Duration(durf)
|
||||||
if dur > float64(b.Max) {
|
if dur > max {
|
||||||
return b.Max
|
//cap!
|
||||||
|
return max
|
||||||
}
|
}
|
||||||
//return as a time.Duration
|
return dur
|
||||||
return time.Duration(dur)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
//Resets the current value of the counter back to Min
|
// Reset restarts the current attempt counter at zero.
|
||||||
func (b *Backoff) Reset() {
|
func (b *Backoff) Reset() {
|
||||||
b.attempt = 0
|
b.attempt = 0
|
||||||
}
|
}
|
||||||
|
|
||||||
//Get the current backoff attempt
|
// Attempt returns the current attempt counter value.
|
||||||
func (b *Backoff) Attempt() float64 {
|
func (b *Backoff) Attempt() float64 {
|
||||||
return b.attempt
|
return b.attempt
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue