Golang實(shí)現(xiàn)超時(shí)退出的三種方式

前段時(shí)間發(fā)現(xiàn)線上有個(gè)服務(wù)接口，總是間歇性告警，有時(shí)候一天兩三次，有時(shí)候一天都沒有。

告警的邏輯是在一個(gè)接口中異步調(diào)用了另一個(gè)HTTP接口，這個(gè)HTTP接口調(diào)用出現(xiàn)超時(shí)。但是我去問了負(fù)責(zé)這個(gè)HTTP接口的同學(xué)，人家說他們的接口相應(yīng)都是毫秒級(jí)別，還截圖監(jiān)控了，有圖有真相，我還能說啥。

但是，超時(shí)是確實(shí)存在的，只是請(qǐng)求還可能沒有到人家服務(wù)那邊。

這種偶發(fā)性問題不好復(fù)現(xiàn)，偶爾來個(gè)告警也挺煩的，第一反應(yīng)還是先解決問題，思路也簡單，失敗后重試。

解決方法

且不談重試策略，先說說什么時(shí)候觸發(fā)重試。

我們可以在接口請(qǐng)求出錯(cuò)拋出err的時(shí)候重試，但是這種不好控制，如果一個(gè)請(qǐng)求出去，十來秒都沒有響應(yīng)，則這個(gè)協(xié)程就要傻傻的等他報(bào)錯(cuò)才能重試，浪費(fèi)生命啊~

所以結(jié)合上面同學(xué)給出的毫秒級(jí)響應(yīng)指標(biāo)，可以設(shè)定一個(gè)超時(shí)時(shí)間，如果在指定超時(shí)時(shí)間后沒有返回結(jié)果，則重試（這篇重試不是重點(diǎn)）。

func AsyncCall() {
 ctx, cancel := context.WithTimeout(context.Background(), time.Duration(time.Millisecond*800))
 defer cancel()
 go func(ctx context.Context) {
 // 發(fā)送HTTP請(qǐng)求
 }()

 select {
 case -ctx.Done():
 fmt.Println("call successfully!!!")
 return
 case -time.After(time.Duration(time.Millisecond * 900)):
 fmt.Println("timeout!!!")
 return
 }
}

說明

1、通過context的WithTimeout設(shè)置一個(gè)有效時(shí)間為800毫秒的context。

2、該context會(huì)在耗盡800毫秒后或者方法執(zhí)行完成后結(jié)束，結(jié)束的時(shí)候會(huì)向通道ctx.Done發(fā)送信號(hào)。

3、有人可能要問，你這里已經(jīng)設(shè)置了context的有效時(shí)間，為什么還要加上這個(gè)time.After呢？

這是因?yàn)樵摲椒▋?nèi)的context是自己申明的，可以手動(dòng)設(shè)置對(duì)應(yīng)的超時(shí)時(shí)間，但是在大多數(shù)場景，這里的ctx是從上游一直傳遞過來的，對(duì)于上游傳遞過來的context還剩多少時(shí)間，我們是不知道的，所以這時(shí)候通過time.After設(shè)置一個(gè)自己預(yù)期的超時(shí)時(shí)間就很有必要了。

4、注意，這里要記得調(diào)用cancel()，不然即使提前執(zhí)行完了，還要傻傻等到800毫秒后context才會(huì)被釋放。

總結(jié)

上面的超時(shí)控制是搭配使用了ctx.Done和time.After。

Done通道負(fù)責(zé)監(jiān)聽context啥時(shí)候完事，如果在time.After設(shè)置的超時(shí)時(shí)間到了，你還沒完事，那我就不等了，執(zhí)行超時(shí)后的邏輯代碼。

舉一反三

那么，除了上面這種超時(shí)控制策略，還有其他的套路嗎？

有，但是大同小異。

第一種：使用time.NewTimer

func AsyncCall() {
 ctx, cancel := context.WithTimeout(context.Background(), time.Duration(time.Millisecond * 800))
 defer cancel()
 timer := time.NewTimer(time.Duration(time.Millisecond * 900))

 go func(ctx context.Context) {
 // 發(fā)送HTTP請(qǐng)求
 }()

 select {
 case -ctx.Done():
 timer.Stop()
 timer.Reset(time.Second)
 fmt.Println("call successfully!!!")
 return
 case -timer.C:
 fmt.Println("timeout!!!")
 return
 }
}

這里的主要區(qū)別是將time.After換成了time.NewTimer，也是同樣的思路如果接口調(diào)用提前完成，則監(jiān)聽到Done信號(hào)，然后關(guān)閉定時(shí)器。

否則的話，會(huì)在指定的timer即900毫秒后執(zhí)行超時(shí)后的業(yè)務(wù)邏輯。

第二種：使用通道

func AsyncCall() {
 ctx := context.Background()
 done := make(chan struct{}, 1)

 go func(ctx context.Context) {
 // 發(fā)送HTTP請(qǐng)求
 done - struct{}{}
 }()

 select {
 case -done:
 fmt.Println("call successfully!!!")
 return
 case -time.After(time.Duration(800 * time.Millisecond)):
 fmt.Println("timeout!!!")
 return
 }
}

1、這里主要利用通道可以在協(xié)程之間通信的特點(diǎn)，當(dāng)調(diào)用成功后，向done通道發(fā)送信號(hào)。

2、監(jiān)聽Done信號(hào)，如果在time.After超時(shí)時(shí)間之前接收到，則正常返回，否則走向time.After的超時(shí)邏輯，執(zhí)行超時(shí)邏輯代碼。

3、這里使用的是通道和time.After組合，也可以使用通道和time.NewTimer組合。

總結(jié)

本篇主要介紹如何實(shí)現(xiàn)超時(shí)控制，主要有三種

1、context.WithTimeout/context.WithDeadline + time.After

2、context.WithTimeout/context.WithDeadline + time.NewTimer

3、channel + time.After/time.NewTimer

到此這篇關(guān)于Golang三種方式實(shí)現(xiàn)超時(shí)退出的文章就介紹到這了,更多相關(guān)Golang超時(shí)退出內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！