upstream/icingadb
1
0
Fork 0
mirror of https://github.com/Icinga/icingadb.git synced 2026-06-06 15:22:08 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Introduce Settings#OnError

Browse source 
refs #351
This commit is contained in:
Alexander A. Klimov 2021-09-22 15:34:39 +02:00
parent 8d57ec107a
commit 653df82a1e
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
pkg/retry/retry.go
View file

@ -17,6 +17,8 @@ type IsRetryable func(error) bool
type Settings struct {
// Timeout lets WithBackoff give up once elapsed (if >0).
Timeout time.Duration
// OnError is called if an error occurs.
OnError func(elapsed time.Duration, attempt uint64, err, lastErr error)
}
// WithBackoff retries the passed function if it fails and the error allows it to retry.
@ -30,13 +32,18 @@ func WithBackoff(
defer cancelCtx()
}
for attempt := 0; ; /* true */ attempt++ {
start := time.Now()
for attempt := uint64(0); ; /* true */ attempt++ {
prevErr := err
if err = retryableFunc(ctx); err == nil {
return
}
if settings.OnError != nil {
settings.OnError(time.Since(start), attempt, err, prevErr)
}
isRetryable := retryable(err)
if prevErr != nil && (errors.Is(err, context.DeadlineExceeded) || errors.Is(err, context.Canceled)) {
@ -49,7 +56,7 @@ func WithBackoff(
return
}
sleep := b(uint64(attempt))
sleep := b(attempt)
select {
case <-ctx.Done():
if err == nil {