You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
669 B
Go
44 lines
669 B
Go
package fx
|
|
|
|
import "zero/core/errorx"
|
|
|
|
const defaultRetryTimes = 3
|
|
|
|
type (
|
|
RetryOption func(*retryOptions)
|
|
|
|
retryOptions struct {
|
|
times int
|
|
}
|
|
)
|
|
|
|
func DoWithRetries(fn func() error, opts ...RetryOption) error {
|
|
var options = newRetryOptions()
|
|
for _, opt := range opts {
|
|
opt(options)
|
|
}
|
|
|
|
var berr errorx.BatchError
|
|
for i := 0; i < options.times; i++ {
|
|
if err := fn(); err != nil {
|
|
berr.Add(err)
|
|
} else {
|
|
return nil
|
|
}
|
|
}
|
|
|
|
return berr.Err()
|
|
}
|
|
|
|
func WithRetries(times int) RetryOption {
|
|
return func(options *retryOptions) {
|
|
options.times = times
|
|
}
|
|
}
|
|
|
|
func newRetryOptions() *retryOptions {
|
|
return &retryOptions{
|
|
times: defaultRetryTimes,
|
|
}
|
|
}
|