1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
package utils
import (
"math"
"time"
)
// IncrementalDuration handles a time.Duration with max limits.
type IncrementalDuration struct {
Duration time.Duration
InitialDuration time.Duration
MaxDuration time.Duration
multiplier float64
}
// NewIncrementalDuration returns an IncrementalDuration object with initialized values.
func NewIncrementalDuration(duration, maxDuration time.Duration, multiplier float64) *IncrementalDuration {
return &IncrementalDuration{
Duration: duration,
InitialDuration: duration,
MaxDuration: maxDuration,
multiplier: multiplier,
}
}
// Reset sets current duration to initial duration.
func (id *IncrementalDuration) Reset() {
id.Duration = id.InitialDuration
}
func (id *IncrementalDuration) IncreaseDuration() {
id.Duration = time.Duration(math.Min(id.multiplier*float64(id.Duration), float64(id.MaxDuration)))
}
// NextTimeout returns a timeout channel based on current duration.
func (id *IncrementalDuration) NextTimeout() <-chan time.Time {
ch := time.After(id.Duration)
id.IncreaseDuration()
return ch
}
|
