diff options
author | Keuin <[email protected]> | 2023-07-07 00:34:25 +0800 |
---|---|---|
committer | Keuin <[email protected]> | 2023-07-07 00:34:25 +0800 |
commit | 6eb5a7af48d04adc5625cbc8355e2556db4b992f (patch) | |
tree | ba9327214773497590e7c24995c4221b1c8a8b5a /common/retry.go | |
parent | c133daaba30d6c75eda2136a3c11682710f4562a (diff) |
Refactor: move retry into single package.
Diffstat (limited to 'common/retry.go')
-rw-r--r-- | common/retry.go | 47 |
1 files changed, 0 insertions, 47 deletions
diff --git a/common/retry.go b/common/retry.go deleted file mode 100644 index ff7b869..0000000 --- a/common/retry.go +++ /dev/null @@ -1,47 +0,0 @@ -package common - -import ( - "context" - "github.com/keuin/slbr/logging" - "time" -) - -// AutoRetry retries the supplier automatically, with given time limit and interval. -// If maximum retry time limit is reached and the supplier still fails, -// the last error will be returned. -// If logger is not nil, retry information will be printed to it. -func AutoRetry[T any]( - ctx context.Context, - supplier func() (T, error), - maxRetryTimes int, - retryInterval time.Duration, - logger *logging.Logger) (T, error) { - var err error - var ret T - for i := 0; i < maxRetryTimes+1; i++ { - ret, err = supplier() - if err != nil { - if logger != nil { - logger.Info("Try %v/%v (sleep %v): %v", - i, maxRetryTimes, retryInterval, err) - } - timer := time.NewTimer(retryInterval) - select { - case <-timer.C: - // time to have the next try - continue - case <-ctx.Done(): - // context is cancelled - var zero T - return zero, ctx.Err() - } - } - // success - return ret, nil - } - if logger != nil { - logger.Error("Max retry times reached, but it still fails. Last error: %v", err) - } - var zero T - return zero, err -} |