Go重试机制

时间:2022-09-29 六号积极分子人气:0

场景

服务依赖第三方服务或者其他服务，但有时第三方服务偶发性出现问题并且会快速的恢复，而我们的系统可能因为这些偶发性出现问题。

所以无法保证以来服务的可靠性，但是我们可以通过其他的方式来保证我们的服务是稳定的，可靠的。

重试机制可以协助我们构建健壮的服务。比如调用第三方服务或执行某操作时，出现执行失败，我们可以让其再重试几次再抛出错误。

设计原理

重试机制：某方法或者函数执行失败了重新执行，尝试重新执行后，尝试几次失败后结束。期间成功一次则表示成功。

重试次数
需要被重新执行方法

func add() (string, error) {
    var i = 1
    var j = 2
    result := i + j
    return strconv.Itoa(result), nil
}
func RetryFunc(count int, cback func() (string, error)) {
    for i := 0; i < count; i++ {
        result, err := cback()
        if err == nil {
            return result
        }
    }
}
// 应用
package main
func main(){
    Retry(3, add)
}

重试函数接收2个参数，一个是重试次数，一个是需要重新执行的函数。但函数都是固定的。可以进一步优化。

type Effector func(context.Context) (string, error)
func Retry(count int ,delay time.Duration, effector){
    return func(ctx. context.Context)(string, error){
        for r:=0; ;r++{
            response, err := effector()
            if err == nil || r>= count {
                return  response, err
             }
            fmt.Printf("Function call failed, retrying in %v", delay)
            select {
                case <- time.After(delay)
                case <- ctx.Done():
                    return "", ctx.Err()
            }
        }
    }
}

增加延时参数 delay 控制重试延时，使用匿名函数定义一个函数接收器，返回值为匿名函数的。所以重试函数接受三个参数：一个效应器，一个整数，描述函数重试传递的效应器的次数以及重试之间的延迟。

函数接收器的参数是 context ，context 主要用来在 goroutine 之间传递上下文信息。这里使用context主要是用于超时取消，然后快速返回。 context包主要提供了两种方式创建context:

context.Background 是上下文的默认值，所有其他的上下文都应该从它衍生（Derived）出来。

context.TODO 应该只在不确定应该使用哪种上下文时使用；

加载全部内容