backoff: Limit the number of attempts using cenkalti/backoff

limited.go

import "github.com/cenkalti/backoff"

type limitedAttempts struct {
	limit    int
	attempts int
	strategy backoff.BackOff
}

func NewLimitedAttempts(limit int, strategy backoff.BackOff) *limitedAttempts {
	return &limitedAttempts{limit: limit, strategy: strategy}
}

func (b *limitedAttempts) Reset() { b.attempts = 0 }

func (b *limitedAttempts) NextBackOff() time.Duration {
	if b.attempts >= b.limit {
		return backoff.Stop
	}
	b.attempts++
	return b.strategy.NextBackOff()
}