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
42
43
44
45
46
47
48
49
50
51
52
|
package backoff_test
import (
"context"
"errors"
"testing"
cenkalti "github.com/cenkalti/backoff"
lestrrat "github.com/lestrrat-go/backoff/v2"
)
func Benchmark(b *testing.B) {
// This is a dummy function
fn := func() int {
var v int
for i := 1; i <= 10; i++ {
v += i
}
return v
}
b.Run("cenkalti", func(b *testing.B) {
for i := 0; i < b.N; i++ {
b.StopTimer()
backoff := cenkalti.NewExponentialBackOff()
b.StartTimer()
cenkalti.Retry(func() error {
_ = fn()
return errors.New(`dummy`)
}, cenkalti.WithMaxRetries(backoff, 5))
}
})
b.Run("lestrrat", func(b *testing.B) {
b.StopTimer()
policy := lestrrat.Exponential(lestrrat.WithMaxRetries(5), lestrrat.WithJitterFactor(1.2))
for i := 0; i < b.N; i++ {
b.StartTimer()
backoff := policy.Start(context.Background())
MAIN:
for {
fn()
select {
case <-backoff.Done():
break MAIN
case <-backoff.Next():
_ = fn()
}
}
b.StopTimer()
}
})
}
|
