retry1.go

package retry

import "time"

type retrier struct {
	timeout time.Duration
	retries int
}

func New(timeout time.Duration, retries int) *retrier {
	return &retrier{
		timeout: timeout,
		retries: retries,
	}
}

func (br *retrier) nextTimeout() time.Duration {
	br.retries--
	return br.timeout
}

func (br *retrier) keepTrying() bool {
	return br.retries > 0
}

func (br *retrier) TotalTimeout() (total time.Duration) {
	r := *br // Make a copy to preserve unchanged original

	for r.keepTrying() {
		total += r.nextTimeout()
	}
	return total
}