mirror of
https://github.com/nais/wonderwall.git
synced 2026-08-21 20:16:29 +00:00
Trong Huu Nguyen
aa74b67a33
Both were reported by deadcode: retry.WithBase was never used, and PrometheusMiddleware.Initialize is a leftover from the chi-prometheus code this middleware was adapted from.
48 lines
938 B
Go
48 lines
938 B
Go
package retry
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/sethvargo/go-retry"
|
|
)
|
|
|
|
var RetryableError = retry.RetryableError
|
|
|
|
type fibonacci struct {
|
|
base time.Duration
|
|
max time.Duration
|
|
}
|
|
|
|
type Option func(*fibonacci)
|
|
|
|
func WithMax(d time.Duration) Option {
|
|
return func(f *fibonacci) {
|
|
f.max = d
|
|
}
|
|
}
|
|
|
|
func Do(ctx context.Context, f retry.RetryFunc, opts ...Option) error {
|
|
return retry.Do(ctx, fibonacciBackoff(opts...), f)
|
|
}
|
|
|
|
func DoValue[T any](ctx context.Context, f retry.RetryFuncValue[T], opts ...Option) (T, error) {
|
|
return retry.DoValue(ctx, fibonacciBackoff(opts...), f)
|
|
}
|
|
|
|
func fibonacciBackoff(opts ...Option) retry.Backoff {
|
|
f := &fibonacci{
|
|
base: 50 * time.Millisecond,
|
|
max: 5 * time.Second,
|
|
}
|
|
|
|
for _, opt := range opts {
|
|
opt(f)
|
|
}
|
|
|
|
b := retry.NewFibonacci(f.base)
|
|
// beware: this starts a timer when invoked, on which the max duration is evaluated against
|
|
b = retry.WithMaxDuration(f.max, b)
|
|
return b
|
|
}
|