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

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

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

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

測試發(fā)現(xiàn)無法實現(xiàn) timeout，又仔細(xì)查看文檔，才發(fā)現(xiàn) golang 中 select 另有玄機。廢話少說，直接總結(jié)要點：

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

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

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
}

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

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

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

還要一種辦法，上面說了，select 操作 chan，并且可以指定 default，那是不是有思路了呢？

if {
    select {
        case done <- 1:
        default:
            return
    }
}

我們嘗試往 chan 中發(fā)送，如果發(fā)不出去，則就退出，也實現(xiàn)了目的。

最后總結(jié)一下，goroutine 泄露的防范條例：

• 創(chuàng)建 goroutine 時就要想好該 goroutine 該如何結(jié)束。
• 使用 chan 時，要考慮到 chan 阻塞時協(xié)程可能的行為。
• 實現(xiàn)循環(huán)語句時注意循環(huán)的退出條件，避免死循環(huán)。

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

最新評論