快捷導(dǎo)航

Golang技巧之重試機(jī)制詳解

更新時間：2023年05月24日 14:09:32 作者：tracy小貓

重試機(jī)制是一種在程序執(zhí)行過程中出現(xiàn)錯誤后重新嘗試執(zhí)行程序的一種機(jī)制，可以減少程序運(yùn)行過程中出現(xiàn)的錯誤，從而提高程序的可靠性，本文就來講講Golang中是如何實(shí)現(xiàn)重試機(jī)制的吧

重試機(jī)制是一種在程序執(zhí)行過程中出現(xiàn)錯誤后重新嘗試執(zhí)行程序的一種機(jī)制。通過重啟軟件、重新連接網(wǎng)絡(luò)、重發(fā)請求等方式，來減少程序運(yùn)行過程中出現(xiàn)的錯誤，從而提高程序的可靠性。

重試機(jī)制通常應(yīng)用在網(wǎng)絡(luò)通訊等領(lǐng)域，比如在調(diào)用遠(yuǎn)程接口時，由于網(wǎng)絡(luò)波動等原因?qū)е抡埱笫?，就需要進(jìn)行重試，這樣可以最大限度的提高程序的正常運(yùn)行率。

封裝go重試機(jī)制的思路

封裝go重試機(jī)制需要考慮以下幾個點(diǎn)：

定義重試最大次數(shù)

包括最大嘗試次數(shù)、重試時間間隔等參數(shù)，這些參數(shù)可以根據(jù)實(shí)際情況進(jìn)行配置和修改。例如，在請求遠(yuǎn)程接口時，如果經(jīng)過3次重試后仍然失敗，則應(yīng)該結(jié)束本次請求。

處理重試時的錯誤類型

重試時可能因?yàn)椴煌蛞l(fā)不同的錯誤，需要考慮在重試機(jī)制中處理這些錯誤，以便進(jìn)行后續(xù)的嘗試。

定義重試機(jī)制的超時時間

超時時間通常在請求遠(yuǎn)程服務(wù)時用到，稱為超時設(shè)置，是指等待服務(wù)端返回數(shù)據(jù)的時間間隔，超出這個時間間隔將引發(fā)錯誤。在重試機(jī)制中，如果進(jìn)行多次重試后，服務(wù)端仍未返回數(shù)據(jù)，則需要在一定時間內(nèi)強(qiáng)制結(jié)束重試。

對重試過程進(jìn)行日志記錄

在重試機(jī)制中，需要記錄每次重試的結(jié)果和原因，包括成功和失敗的情況，這樣便于排查問題和進(jìn)行后續(xù)優(yōu)化。

go重試機(jī)制實(shí)現(xiàn)

重試機(jī)制的實(shí)現(xiàn)方案可以分為兩種：函數(shù)封裝和結(jié)構(gòu)體封裝

函數(shù)封裝是一個簡單的方式，它將所有重試機(jī)制相關(guān)的參數(shù)都封裝在參數(shù)里面，以便于用戶調(diào)用程序，但它缺少高級功能，如日志記錄和函數(shù)實(shí)現(xiàn)。

結(jié)構(gòu)體封裝是另外一種封裝方式，它將重試機(jī)制相關(guān)的參數(shù)封裝在結(jié)構(gòu)體里面，允許用戶定義結(jié)構(gòu)體以及修改結(jié)構(gòu)體中的參數(shù)，同時可以跟蹤重試機(jī)制的狀態(tài)。

在本文中，我們將介紹一種使用結(jié)構(gòu)體進(jìn)行封裝的重試機(jī)制，代碼實(shí)現(xiàn)如下：

package retry
import (
    "time"
)
type Retry struct {
    retryCount        int           // 最大重試次數(shù)
    retryDelay        time.Duration // 重試之間的延遲
    retryTimeout      time.Duration // 超時重試時間
    retryIntervalFunc func(int) time.Duration
}
type RetriableError struct {
    error error
}
func (e RetriableError) Error() string {
    return e.error.Error()
}
// 重試方法
func (r *Retry) Retry(fn func() error) error {
    retryStart := time.Now()
    var err error
    retryCount := 0
    for {
        duration := r.retryIntervalFunc(retryCount)
        time.Sleep(duration)
        err = fn()
        if err == nil {
            return nil
        }
        if _, ok := err.(RetriableError); ok {
            if retryCount < r.retryCount-1 {
                retryCount++
                continue
            }
            break
        }
        if retryStart.Add(r.retryTimeout).Before(time.Now()) {
            break
        }
    }
    return err
}
func New(
    retryCount int,
    retryDelay time.Duration,
    retryTimeout time.Duration,
    retryIntervalFunc func(attempt int) time.Duration,
) *Retry {
    return &Retry{
        retryCount:        retryCount,
        retryDelay:        retryDelay,
        retryTimeout:      retryTimeout,
        retryIntervalFunc: retryIntervalFunc,
    }
}
func ExampleRetry() {
    r := New(3, 1*time.Second, 5*time.Minute, func(attempt int) time.Duration {
        return r.retryDelay
    })
    err := r.Retry(func() error {
        // Do some work here
        return RetriableError{error: errors.New("foo")}
    })
    if err != nil {
        panic(err)
    }
}

在上面的代碼中，我們將Retry結(jié)構(gòu)體進(jìn)行了封裝。結(jié)構(gòu)體中包含了重試最大次數(shù)（RetryCount）、重試間隔（RetryDelay）、超時時間（RetryTimeout）和重試時間間隔函數(shù)（RetryIntervalFunc）四個參數(shù)，通過 New（）方法進(jìn)行初始化。

其中RetryIntervalFunc接受一個嘗試計(jì)數(shù)器為參數(shù)，并據(jù)此返回下一次重試之前的時間間隔。fn 函數(shù)是要重試的函數(shù)，返回一個錯誤作為重試方法的返回值。

Retry方法包括一個無限循環(huán)，其中會嘗試調(diào)用 fn 函數(shù)。如果 fn 函數(shù)返回 nil，method返回 nil，如果錯誤是RetriableError，則重試，否則返回錯誤。

重試方法會在每次重試之間等待指定的延遲時間（RetryDelay），延遲的時間間隔將在RetryIntervalFunc函數(shù)中進(jìn)行計(jì)算。

使用go重試機(jī)制的例子

在本章中，我們將以調(diào)用遠(yuǎn)程API為例子來演示如何使用go重試機(jī)制。

我們將使用重試機(jī)制來調(diào)用遠(yuǎn)程服務(wù)器并處理原始響應(yīng)。假設(shè)我們正在使用Go語言從一個遠(yuǎn)程API獲取數(shù)據(jù)，我們需要處理這些重試請求。如果在前五次嘗試后，我們?nèi)晕传@得數(shù)據(jù)，則返回“重試失敗”錯誤。在這里，我們要求嘗試前等待1秒鐘，并在重試6次之前最多等待5分鐘（300秒）。

func ExampleRemoteAPICall() {
    r := New(5, 1*time.Second, 5*time.Minute, func(attempt int) time.Duration {
        return r.retryDelay
    })
    resp, err := r.Retry(func() error {
        r, err := http.NewRequest("GET", "<http://api.example.com>", nil)
        if err != nil {
            return RetriableError{error: err}
        }
        client := &http.Client{}
        resp, err := client.Do(r)
        if err != nil {
            return RetriableError{error: err}
        }
        if resp.StatusCode != http.StatusOK {
            return RetriableError{error: fmt.Errorf("Unexpected status code (%v)", resp.Status)}
        }
        // 遠(yuǎn)程API調(diào)用成功，我們將返回數(shù)據(jù)。
        return nil
    })

到此這篇關(guān)于Golang技巧之重試機(jī)制詳解的文章就介紹到這了,更多相關(guān)Golang重試機(jī)制內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: