golang 中的協(xié)程使用非常方便，但是協(xié)程什么時(shí)候結(jié)束是一個(gè)控制問(wèn)題，可以用 select 配合使用。

首先聲明，golang 使用并不熟悉，本文僅僅是記錄使用過(guò)程中遇到的一些坑。

子協(xié)程和父協(xié)程的通信通常用 context 或者 chan。我遇到一個(gè)通常的使用場(chǎng)景，在子協(xié)程中嘗試多次處理，父協(xié)程等待一段時(shí)間超時(shí)，我選擇用 chan 實(shí)現(xiàn)。我以為 select 和 C++ 中 switch 類(lèi)似，所以最開(kāi)始代碼類(lèi)似如下：

for {
    select {
        case <-ctx.Done():
            // process ctx done
        case <-time.After(time.Second * 3):
            // process after
        default:
            // process code
    }
}

測(cè)試發(fā)現(xiàn)無(wú)法實(shí)現(xiàn) timeout，又仔細(xì)查看文檔，才發(fā)現(xiàn) golang 中 select 另有玄機(jī)。廢話少說(shuō)，直接總結(jié)要點(diǎn)：

• select 中的 case 必須是進(jìn)行 chan 的手法操作，也就是只能在 case 中操作 chan，并且是 非阻塞接收 。
• select 中的 case 是同時(shí)監(jiān)聽(tīng)的，多個(gè) case 同時(shí)操作，并未 switch 中一個(gè)個(gè)順序判斷。如果多個(gè) case 滿足要求，隨機(jī)執(zhí)行一個(gè)，如果一個(gè)沒(méi)有則阻塞當(dāng)前的協(xié)程（沒(méi)有 default 情況下）。 很類(lèi)似 Linux 文件符操作的 select 語(yǔ)義 。
• 上面說(shuō)的阻塞是沒(méi)有 default 的情況下，如果有 default，則執(zhí)行 default，然后退出 select，也就是不會(huì)阻塞當(dāng)前協(xié)程。

回到上述代碼，我這個(gè) select 會(huì)一直不斷的執(zhí)行 default， time.After 生成的 chan 并不會(huì)被阻塞判斷，所以根本無(wú)法完成我想要的效果。理解了之后重新修改代碼：

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
}

開(kāi)一個(gè)新的協(xié)程去不斷嘗試，在外的三個(gè) case 有一個(gè)滿足，則會(huì)執(zhí)行。但是這里有一個(gè)問(wèn)題非常需要注意： 子協(xié)程什么時(shí)候退出？ 。

因?yàn)?gorountine 不能被強(qiáng)制 kill，所以在上述超時(shí)的情況下，select 語(yǔ)句執(zhí)行 case time.After 之后退出， done 這個(gè) chan 已經(jīng)沒(méi)有接受方了，因此既沒(méi)有接受者，又沒(méi)有緩沖區(qū)，結(jié)合 chan 的特性，則子協(xié)程會(huì)一直阻塞無(wú)法退出，所以本質(zhì)上這個(gè)實(shí)現(xiàn)會(huì)導(dǎo)致子協(xié)程累積下去，也就是 協(xié)程泄露 ，可能會(huì)使資源耗盡。

如何避免上述問(wèn)題呢？一個(gè)很簡(jiǎn)單的想法就是提供緩沖區(qū)， done := make(char int, 1) ，這樣即使沒(méi)有接收方，子協(xié)程也能完成發(fā)送，不會(huì)被阻塞。

還要一種辦法，上面說(shuō)了，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)語(yǔ)句時(shí)注意循環(huán)的退出條件，避免死循環(huán)。

到此這篇關(guān)于golang select 機(jī)制和超時(shí)的文章就介紹到這了,更多相關(guān)golang select 機(jī)制和超時(shí)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論