lotus/lib/retry/retry.go

32 lines
705 B
Go
Raw Permalink Normal View History

package retry
import (
2022-09-28 15:07:05 +00:00
"context"
"time"
logging "github.com/ipfs/go-log/v2"
"github.com/filecoin-project/lotus/api"
)
var log = logging.Logger("retry")
2022-09-28 15:07:05 +00:00
func Retry[T any](ctx context.Context, attempts int, initialBackoff time.Duration, errorTypes []error, f func() (T, error)) (result T, err error) {
for i := 0; i < attempts; i++ {
if i > 0 {
log.Info("Retrying after error:", err)
2022-08-22 21:32:43 +00:00
time.Sleep(initialBackoff)
initialBackoff *= 2
}
result, err = f()
if err == nil || !api.ErrorIsIn(err, errorTypes) {
2022-08-19 15:33:37 +00:00
return result, err
}
2022-09-28 15:07:05 +00:00
if ctx.Err() != nil {
return result, ctx.Err()
}
}
log.Errorf("Failed after %d attempts, last error: %s", attempts, err)
return result, err
}