Skip to content

Commit 48818d8

Browse files
committed
Fix backoff constructor and tests
1 parent a49656f commit 48818d8

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

pulsar/backoff/backoff.go

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,10 +43,13 @@ type DefaultBackoff struct {
4343
}
4444

4545
func NewDefaultBackoff() Policy {
46-
return &DefaultBackoff{rnd: rand.New(rand.NewSource(time.Now().UnixNano()))}
46+
return NewDefaultBackoffWithInitialBackOff(0)
4747
}
4848
func NewDefaultBackoffWithInitialBackOff(backoff time.Duration) Policy {
49-
return &DefaultBackoff{backoff: backoff / 2}
49+
return &DefaultBackoff{
50+
rnd: rand.New(rand.NewSource(time.Now().UnixNano())),
51+
backoff: backoff / 2,
52+
}
5053
}
5154

5255
const maxBackoff = 60 * time.Second

pulsar/backoff/backoff_test.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,14 +25,14 @@ import (
2525
)
2626

2727
func TestBackoff_NextMinValue(t *testing.T) {
28-
backoff := &DefaultBackoff{}
28+
backoff := NewDefaultBackoff()
2929
delay := backoff.Next()
3030
assert.GreaterOrEqual(t, int64(delay), int64(100*time.Millisecond))
3131
assert.LessOrEqual(t, int64(delay), int64(120*time.Millisecond))
3232
}
3333

3434
func TestBackoff_NextExponentialBackoff(t *testing.T) {
35-
backoff := &DefaultBackoff{}
35+
backoff := NewDefaultBackoff()
3636
previousDelay := backoff.Next()
3737
// the last value before capping to the max value is 51.2 s (.1, .2, .4, .8, 1.6, 3.2, 6.4, 12.8, 25.6, 51.2)
3838
for previousDelay < 51*time.Second {
@@ -47,7 +47,7 @@ func TestBackoff_NextExponentialBackoff(t *testing.T) {
4747
}
4848

4949
func TestBackoff_NextMaxValue(t *testing.T) {
50-
backoff := &DefaultBackoff{}
50+
backoff := NewDefaultBackoff()
5151
var delay time.Duration
5252
for delay < maxBackoff {
5353
delay = backoff.Next()

0 commit comments

Comments
 (0)