go 優雅的檢查channel關閉


原文作者:shitaibin
鏈接:https://www.jianshu.com/p/79d27f200bcf
來源:簡書

goroutine作為Golang並發的核心,我們不僅要關注它們的創建和管理,當然還要關注如何合理的退出這些協程,不(合理)退出不然可能會造成阻塞、panic、程序行為異常、數據結果不正確等問題。這篇文章介紹,如何合理的退出goroutine,減少軟件bug。
goroutine在退出方面,不像線程和進程,不能通過某種手段強制關閉它們,只能等待goroutine主動退出。但也無需為退出、關閉goroutine而煩惱,下面就介紹3種優雅退出goroutine的方法,只要采用這種最佳實踐去設計,基本上就可以確保goroutine退出上不會有問題,盡情享用。

第一種:使用for-range退出

for-range是使用頻率很高的結構,常用它來遍歷數據,range能夠感知channel的關閉,當channel被發送數據的協程關閉時,range就會結束,接着退出for循環。
它在並發中的使用場景是:當協程只從1個channel讀取數據,然后進行處理,處理后協程退出。下面這個示例程序,當in通道被關閉時,協程可自動退出。

go func(in <-chan int) {
    // Using for-range to exit goroutine
    // range has the ability to detect the close/end of a channel
    for x := range in {
        fmt.Printf("Process %d\n", x)
    }
}(inCh)

  

第二種:使用,ok退出

for-select也是使用頻率很高的結構,select提供了多路復用的能力,所以for-select可以讓函數具有持續多路處理多個channel的能力。但select沒有感知channel的關閉,這引出了2個問題:1)繼續在關閉的通道上讀,會讀到通道傳輸數據類型的零值,2)繼續在關閉的通道上寫,將會panic。問題2可使用的原則是,通道只由發送方關閉,接收方不可關閉,即某個寫通道只由使用該select的協程關閉,select中就不存在繼續在關閉的通道上寫數據的問題。

問題1可以使用,ok來檢測通道的關閉,使用情況有2種。
第一種:如果某個通道關閉后,需要退出協程,直接return即可。示例代碼中,該協程需要從in通道讀數據,還需要定時打印已經處理的數量,有2件事要做,所有不能使用for-range,需要使用for-select,當in關閉時,ok=false,我們直接返回。

go func() {
    // in for-select using ok to exit goroutine
    for {
        select {
        case x, ok := <-in:
            if !ok {
                return
            }
            fmt.Printf("Process %d\n", x)
            processedCnt++
        case <-t.C:
            fmt.Printf("Working, processedCnt = %d\n", processedCnt)
        }
    }
}()

第二種:如果某個通道關閉了,不再處理該通道,而是繼續處理其他case,退出是等待所有的可讀通道關閉。我們需要使用select的一個特征:select不會在nil的通道上進行等待。這種情況,把只讀通道設置為nil即可解決。

 
go func() {
    // in for-select using ok to exit goroutine
    for {
        select {
        case x, ok := <-in1:
            if !ok {
                in1 = nil
            }
            // Process
        case y, ok := <-in2:
            if !ok {
                in2 = nil
            }
            // Process
        case <-t.C:
            fmt.Printf("Working, processedCnt = %d\n", processedCnt)
        }

        // If both in channel are closed, goroutine exit
        if in1 == nil && in2 == nil {
            return
        }
    }
}()

  

第三種:使用退出通道退出

使用,ok來退出使用for-select協程,解決是當讀入數據的通道關閉時,沒數據讀時程序的正常結束。想想下面這2種場景,,ok還能適用嗎?

  1. 接收的協程要退出了,如果它直接退出,不告知發送協程,發送協程將阻塞。
  2. 啟動了一個工作協程處理數據,如何通知它退出?

使用一個專門的通道,發送退出的信號,可以解決這類問題。以第2個場景為例,協程入參包含一個停止通道stopCh,當stopCh被關閉,case <-stopCh會執行,直接返回即可。

當我啟動了100個worker時,只要main()執行關閉stopCh,每一個worker都會都到信號,進而關閉。如果main()向stopCh發送100個數據,這種就低效了。

func worker(stopCh <-chan struct{}) {
    go func() {
        defer fmt.Println("worker exit")
        // Using stop channel explicit exit
        for {
            select {
            case <-stopCh:
                fmt.Println("Recv stop signal")
                return
            case <-t.C:
                fmt.Println("Working .")
            }
        }
    }()
    return
}

最佳實踐回顧

  1. 發送協程主動關閉通道,接收協程不關閉通道。技巧:把接收方的通道入參聲明為只讀(<-chan),如果接收協程關閉只讀協程,編譯時就會報錯。
  2. 協程處理1個通道,並且是讀時,協程優先使用for-range,因為range可以關閉通道的關閉自動退出協程。
  3. ,ok可以處理多個讀通道關閉,需要關閉當前使用for-select的協程。
  4. 顯式關閉通道stopCh可以處理主動通知協程退出的場景。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM