go-sample-webpage/vendor/github.com/lestrrat-go/backoff/v2/constant.go

67 lines
1.3 KiB
Go
Raw Normal View History

2021-11-04 01:14:51 +00:00
package backoff
import (
"context"
"time"
)
type ConstantInterval struct {
interval time.Duration
jitter jitter
}
func NewConstantInterval(options ...ConstantOption) *ConstantInterval {
jitterFactor := 0.0
interval := time.Minute
var rng Random
for _, option := range options {
switch option.Ident() {
case identInterval{}:
interval = option.Value().(time.Duration)
case identJitterFactor{}:
jitterFactor = option.Value().(float64)
case identRNG{}:
rng = option.Value().(Random)
}
}
return &ConstantInterval{
interval: interval,
jitter: newJitter(jitterFactor, rng),
}
}
func (g *ConstantInterval) Next() time.Duration {
return time.Duration(g.jitter.apply(float64(g.interval)))
}
type ConstantPolicy struct {
cOptions []ControllerOption
igOptions []ConstantOption
}
func NewConstantPolicy(options ...Option) *ConstantPolicy {
var cOptions []ControllerOption
var igOptions []ConstantOption
for _, option := range options {
switch opt := option.(type) {
case ControllerOption:
cOptions = append(cOptions, opt)
default:
igOptions = append(igOptions, opt.(ConstantOption))
}
}
return &ConstantPolicy{
cOptions: cOptions,
igOptions: igOptions,
}
}
func (p *ConstantPolicy) Start(ctx context.Context) Controller {
ig := NewConstantInterval(p.igOptions...)
return newController(ctx, ig, p.cOptions...)
}