golang?select?機(jī)制和超時(shí)問題
golang 中的協(xié)程使用非常方便,但是協(xié)程什么時(shí)候結(jié)束是一個(gè)控制問題,可以用 select 配合使用。
首先聲明,golang 使用并不熟悉,本文僅僅是記錄使用過程中遇到的一些坑。
子協(xié)程和父協(xié)程的通信通常用 context 或者 chan。我遇到一個(gè)通常的使用場(chǎng)景,在子協(xié)程中嘗試多次處理,父協(xié)程等待一段時(shí)間超時(shí),我選擇用 chan 實(shí)現(xiàn)。我以為 select 和 C++ 中 switch 類似,所以最開始代碼類似如下:
for { select { case <-ctx.Done(): // process ctx done case <-time.After(time.Second * 3): // process after default: // process code } }
測(cè)試發(fā)現(xiàn)無法實(shí)現(xiàn) timeout,又仔細(xì)查看文檔,才發(fā)現(xiàn) golang 中 select 另有玄機(jī)。廢話少說,直接總結(jié)要點(diǎn):
- select 中的 case 必須是進(jìn)行 chan 的手法操作,也就是只能在 case 中操作 chan,并且是 非阻塞接收 。
- select 中的 case 是同時(shí)監(jiān)聽的,多個(gè) case 同時(shí)操作,并未 switch 中一個(gè)個(gè)順序判斷。如果多個(gè) case 滿足要求,隨機(jī)執(zhí)行一個(gè),如果一個(gè)沒有則阻塞當(dāng)前的協(xié)程(沒有 default 情況下)。 很類似 Linux 文件符操作的 select 語義 。
- 上面說的阻塞是沒有 default 的情況下,如果有 default,則執(zhí)行 default,然后退出 select,也就是不會(huì)阻塞當(dāng)前協(xié)程。
回到上述代碼,我這個(gè) select 會(huì)一直不斷的執(zhí)行 default, time.After
生成的 chan 并不會(huì)被阻塞判斷,所以根本無法完成我想要的效果。理解了之后重新修改代碼:
done := make(char int) go func(c chan int) { for { // process code if { c <- 1 return } } c <- 0 }(done) select { case <-ctx.Done(): // process ctx done case <-time.After(time.Second * 3): // process after case <-done: // process code }
開一個(gè)新的協(xié)程去不斷嘗試,在外的三個(gè) case 有一個(gè)滿足,則會(huì)執(zhí)行。但是這里有一個(gè)問題非常需要注意: 子協(xié)程什么時(shí)候退出? 。
因?yàn)?gorountine 不能被強(qiáng)制 kill,所以在上述超時(shí)的情況下,select 語句執(zhí)行 case time.After
之后退出, done
這個(gè) chan 已經(jīng)沒有接受方了,因此既沒有接受者,又沒有緩沖區(qū),結(jié)合 chan 的特性,則子協(xié)程會(huì)一直阻塞無法退出,所以本質(zhì)上這個(gè)實(shí)現(xiàn)會(huì)導(dǎo)致子協(xié)程累積下去,也就是 協(xié)程泄露 ,可能會(huì)使資源耗盡。
如何避免上述問題呢?一個(gè)很簡單的想法就是提供緩沖區(qū), done := make(char int, 1)
,這樣即使沒有接收方,子協(xié)程也能完成發(fā)送,不會(huì)被阻塞。
還要一種辦法,上面說了,select 操作 chan,并且可以指定 default,那是不是有思路了呢?
if { select { case done <- 1: default: return } }
我們嘗試往 chan 中發(fā)送,如果發(fā)不出去,則就退出,也實(shí)現(xiàn)了目的。
最后總結(jié)一下,goroutine 泄露的防范條例:
- 創(chuàng)建 goroutine 時(shí)就要想好該 goroutine 該如何結(jié)束。
- 使用 chan 時(shí),要考慮到 chan 阻塞時(shí)協(xié)程可能的行為。
- 實(shí)現(xiàn)循環(huán)語句時(shí)注意循環(huán)的退出條件,避免死循環(huán)。
到此這篇關(guān)于golang select 機(jī)制和超時(shí)的文章就介紹到這了,更多相關(guān)golang select 機(jī)制和超時(shí)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
詳解Golang中channel的實(shí)現(xiàn)
channel俗稱管道,用于數(shù)據(jù)傳遞或數(shù)據(jù)共享,其本質(zhì)是一個(gè)先進(jìn)先出的隊(duì)列,使用goroutine+channel進(jìn)行數(shù)據(jù)通訊簡單高效,同時(shí)也線程安全,本文就給大家講講Golang中channel的實(shí)現(xiàn),需要的朋友可以參考下2023-09-09go微服務(wù)PolarisMesh源碼解析服務(wù)端啟動(dòng)流程
這篇文章主要為大家介紹了go微服務(wù)PolarisMesh源碼解析服務(wù)端啟動(dòng)流程詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-01-01Go語言中init函數(shù)與匿名函數(shù)使用淺析
這篇文章主要介紹了Go語言中init函數(shù)與匿名函數(shù)使用淺析,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)吧2023-01-01Go語言使用net/http實(shí)現(xiàn)簡單登錄驗(yàn)證和文件上傳功能
這篇文章主要介紹了Go語言使用net/http實(shí)現(xiàn)簡單登錄驗(yàn)證和文件上傳功能,使用net/http模塊編寫了一個(gè)簡單的登錄驗(yàn)證和文件上傳的功能,在此做個(gè)簡單記錄,需要的朋友可以參考下2023-07-07Go如何實(shí)現(xiàn)json字符串與各類struct相互轉(zhuǎn)換
這篇文章主要介紹了Go如何實(shí)現(xiàn)json字符串與各類struct相互轉(zhuǎn)換,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2022-08-08