Make ForAttempt concurrent-safe in all cases, golint.

This commit is contained in:
Kevin Gillette 2016-12-22 13:36:47 -07:00
parent a6f64285fd
commit 4dd47ae5ac
1 changed files with 36 additions and 34 deletions

View File

@ -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
} }