grafana/pkg/util/retryer/retryer.go

package retryer

import (
	"time"
)

type RetrySignal = int

const (
	FuncSuccess RetrySignal = iota
	FuncFailure
	FuncComplete
	FuncError
)

// Retry retries the provided function using exponential backoff, starting with `minDelay` between attempts, and increasing to
// `maxDelay` after each failure. Stops when the provided function returns `FuncComplete`, or `maxRetries` is reached.
func Retry(body func() (RetrySignal, error), maxRetries int, minDelay time.Duration, maxDelay time.Duration) error {
	currentDelay := minDelay
	ticker := time.NewTicker(currentDelay)
	defer ticker.Stop()

	retries := 0
	for range ticker.C {
		response, err := body()
		if err != nil {
			return err
		}

		switch response {
		case FuncSuccess:
			currentDelay = minDelay
			ticker.Reset(currentDelay)
			retries = 0
		case FuncFailure:
			currentDelay = minDuration(currentDelay*2, maxDelay)
			ticker.Reset(currentDelay)
			retries++
		case FuncComplete:
			return nil
		}

		if retries >= maxRetries {
			return nil
		}
	}

	return nil
}

func minDuration(a time.Duration, b time.Duration) time.Duration {
	if a < b {
		return a
	}
	return b
}