golang 如何結束一個協程

goroutine和channel是Go語言非常棒的特色，它們提供了一個非常輕的並發能力。 但是當您的應用程式中有很多goroutine的時候，如何在主流程中等待所有的goroutine 退出呢？

1 透過頻道傳遞退出訊號                        （建議學習：go#)

Go的一大設計哲學時：透過Channel分享資料，而為：透過Channel分享資料，而不是透過共享記憶體共享資料。主流程可以透過channel向任何goroutine發送停止信號，就像下面這樣：

func run(done chan int) {
        for {
                select {
                case <-done:
                        fmt.Println("exiting...")
                        done <- 1
                        break
                default:
                }
 
                time.Sleep(time.Second * 1)
                fmt.Println("do something")
        }
}
 
func main() {
        c := make(chan int)
 
        go run(c)
 
        fmt.Println("wait")
        time.Sleep(time.Second * 5)
 
        c <- 1
        <-c
 
        fmt.Println("main exited")
}

這種方式可以實現優雅地停止goroutine，但是當goroutine特別多的時候，這種方式不管在程式碼美觀上還是管理上都顯得笨拙不堪。

2 使用waitgroup

sync套件中的Waitgroup結構，是Go語言為我們提供的多個goroutine之間同步的好刀。下面是官方文件對它的描述：

通常情況下，我們像下面這樣使用waitgroup:

建立一個Waitgroup的實例，假設這裡我們叫它wg

在每個goroutine啟動的時候，呼叫wg.Add(1)，這個操作可以在goroutine啟動之前調用，也可以在goroutine裡面調用。當然，也可以在創建n個goroutine前調用wg.Add(n)

當每個goroutine完成任務後，調用wg.Done()

#在等待所有goroutine的地方調用wg.Wait()，它在所有執行了wg.Add(1)的goroutine都調用完wg.Done()前阻塞，當所有goroutine都調用完wg.Done()之後它會返回。

以上是golang 如何結束一個協程的詳細內容。更多資訊請關注PHP中文網其他相關文章！