使用Go重試機制代碼更可靠

Posted on by

場景

服務依賴第三方服務或者其他服務，但有時第三方服務偶發性出現問題並且會快速的恢復，而我們的系統可能因為這些偶發性出現問題。

所以無法保證以來服務的可靠性，但是我們可以通過其他的方式來保證我們的服務是穩定的，可靠的。

重試機制可以協助我們構建健壯的服務。比如調用第三方服務或執行某操作時，出現執行失敗，我們可以讓其再重試幾次再拋出錯誤。

設計原理

重試機制：某方法或者函數執行失敗瞭重新執行，嘗試重新執行後，嘗試幾次失敗後結束。期間成功一次則表示成功。

  • 重試次數
  • 需要被重新執行方法
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 應該隻在不確定應該使用哪種上下文時使用；

以上就是使用Go重試機制代碼更可靠的詳細內容，更多關於Go重試機制的資料請關註WalkonNet其它相關文章！

推薦閱讀：