chenyh
/
vRp.CD2g_test
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

feat: super.RetryByExponentialBackoff 和 super.ConditionalRetryByExponentialBackoff 支持设置忽略的错误，当返回忽略的错误时将不再进行重试

This commit is contained in:
kercylan98 2023-12-12 10:52:51 +08:00
parent 12619b5fa4
commit 5714a437cc
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
utils/super/retry.go
View File

@ -1,6 +1,7 @@
package super package super
import ( import (
"errors"
"fmt" "fmt"
"math" "math"
"math/rand" "math/rand"
@ -46,8 +47,9 @@ func RetryByRule(f func() error, rule func(count int) time.Duration) error {
// - maxDelay：最大延迟 // - maxDelay：最大延迟
// - multiplier：延迟时间的乘数，通常为 2 // - multiplier：延迟时间的乘数，通常为 2
// - randomization：延迟时间的随机化因子，通常为 0.5 // - randomization：延迟时间的随机化因子，通常为 0.5
func RetryByExponentialBackoff(f func() error, maxRetries int, baseDelay, maxDelay time.Duration, multiplier, randomization float64) error { // - ignoreErrors：忽略的错误，当 f 返回的错误在 ignoreErrors 中时，将不会进行重试
return ConditionalRetryByExponentialBackoff(f, nil, maxRetries, baseDelay, maxDelay, multiplier, randomization) func RetryByExponentialBackoff(f func() error, maxRetries int, baseDelay, maxDelay time.Duration, multiplier, randomization float64, ignoreErrors ...error) error {
return ConditionalRetryByExponentialBackoff(f, nil, maxRetries, baseDelay, maxDelay, multiplier, randomization, ignoreErrors...)
} }
// ConditionalRetryByExponentialBackoff 该函数与 RetryByExponentialBackoff 类似，但是可以被中断 // ConditionalRetryByExponentialBackoff 该函数与 RetryByExponentialBackoff 类似，但是可以被中断
@ -55,7 +57,7 @@ func RetryByExponentialBackoff(f func() error, maxRetries int, baseDelay, maxDel
// //
// 该函数通常用于在重试过程中，需要中断重试的场景，例如： // 该函数通常用于在重试过程中，需要中断重试的场景，例如：
// - 用户请求开始游戏，由于网络等情况，进入重试状态。此时用户再次发送开始游戏请求，此时需要中断之前的重试，避免重复进入游戏 // - 用户请求开始游戏，由于网络等情况，进入重试状态。此时用户再次发送开始游戏请求，此时需要中断之前的重试，避免重复进入游戏
func ConditionalRetryByExponentialBackoff(f func() error, cond func() bool, maxRetries int, baseDelay, maxDelay time.Duration, multiplier, randomization float64) error { func ConditionalRetryByExponentialBackoff(f func() error, cond func() bool, maxRetries int, baseDelay, maxDelay time.Duration, multiplier, randomization float64, ignoreErrors ...error) error {
retry := 0 retry := 0
for { for {
if cond != nil && !cond() { if cond != nil && !cond() {
@ -65,9 +67,14 @@ func ConditionalRetryByExponentialBackoff(f func() error, cond func() bool, maxR
if err == nil { if err == nil {
return nil return nil
} }
for _, ignore := range ignoreErrors {
if errors.Is(err, ignore) {
return err
}
}
if retry >= maxRetries { if retry >= maxRetries {
return fmt.Errorf("max retries reached: %v", err) return fmt.Errorf("max retries reached: %w", err)
} }
delay := float64(baseDelay) * math.Pow(multiplier, float64(retry)) delay := float64(baseDelay) * math.Pow(multiplier, float64(retry))