33 lines
929 B
Go
33 lines
929 B
Go
package super
|
||
|
||
import "time"
|
||
|
||
// Retry 根据提供的 count 次数尝试执行 f 函数，如果 f 函数返回错误，则在 interval 后重试，直到成功或者达到 count 次数
|
||
func Retry(count int, interval time.Duration, f func() error) error {
|
||
var err error
|
||
for i := 0; i < count; i++ {
|
||
if err = f(); err == nil {
|
||
return nil
|
||
}
|
||
time.Sleep(interval)
|
||
}
|
||
return err
|
||
}
|
||
|
||
// RetryAsync 与 Retry 类似，但是是异步执行
|
||
// - 传入的 callback 函数会在执行完毕后被调用，如果执行成功，则 err 为 nil，否则为错误信息
|
||
// - 如果 callback 为 nil，则不会在执行完毕后被调用
|
||
func RetryAsync(count int, interval time.Duration, f func() error, callback func(err error)) {
|
||
go func() {
|
||
var err error
|
||
for i := 0; i < count; i++ {
|
||
if err = f(); err == nil {
|
||
HandleV(nil, callback)
|
||
return
|
||
}
|
||
time.Sleep(interval)
|
||
}
|
||
HandleV(err, callback)
|
||
}()
|
||
}
|