feat: 新增重试函数及两个关于 func 执行的辅助函数

支持通过 super.Handle 和 super.HandleV 对函数进行执行,如果传入的函数不为 nil 则会执行
This commit is contained in:
kercylan98 2023-07-05 13:39:45 +08:00
parent cad87e746c
commit ee87612f60
2 changed files with 55 additions and 0 deletions

23
utils/super/f.go Normal file
View File

@ -0,0 +1,23 @@
package super
// Handle 执行 f 函数,如果 f 为 nil则不执行
func Handle(f func()) {
if f != nil {
f()
}
}
// HandleErr 执行 f 函数,如果 f 为 nil则不执行
func HandleErr(f func() error) error {
if f != nil {
return f()
}
return nil
}
// HandleV 执行 f 函数,如果 f 为 nil则不执行
func HandleV[V any](v V, f func(v V)) {
if f != nil {
f(v)
}
}

32
utils/super/retry.go Normal file
View File

@ -0,0 +1,32 @@
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)
}()
}