37 lines
744 B
Go
37 lines
744 B
Go
package retry
|
|
|
|
import (
|
|
"errors"
|
|
"time"
|
|
|
|
logging "github.com/ipfs/go-log/v2"
|
|
)
|
|
|
|
var log = logging.Logger("retry")
|
|
|
|
func errorIsIn(err error, errorTypes []error) bool {
|
|
for _, etype := range errorTypes {
|
|
tmp := etype
|
|
if errors.As(err, &tmp) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func Retry[T any](attempts int, sleep int, 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)
|
|
time.Sleep(time.Duration(sleep) * time.Second)
|
|
sleep *= 2
|
|
}
|
|
result, err = f()
|
|
if err == nil || !errorIsIn(err, errorTypes) {
|
|
return result, nil
|
|
}
|
|
}
|
|
log.Errorf("Failed after %d attempts, last error: %s", attempts, err)
|
|
return result, err
|
|
}
|