goroutine和channel是Go語言非常棒的特色,它們提供了一種非常輕便易用的并發(fā)能力。但是當您的應(yīng)用進程中有很多goroutine的時候,如何在主流程中等待所有的goroutine 退出呢?
創(chuàng)新互聯(lián)公司專注于崇信企業(yè)網(wǎng)站建設(shè),成都響應(yīng)式網(wǎng)站建設(shè)公司,商城網(wǎng)站建設(shè)。崇信網(wǎng)站建設(shè)公司,為崇信等地區(qū)提供建站服務(wù)。全流程按需開發(fā)網(wǎng)站,專業(yè)設(shè)計,全程項目跟蹤,創(chuàng)新互聯(lián)公司專業(yè)和態(tài)度為您提供的服務(wù)
1 通過Channel傳遞退出信號
Go的一大設(shè)計哲學就是:通過Channel共享數(shù)據(jù),而不是通過共享內(nèi)存共享數(shù)據(jù)。主流程可以通過channel向任何goroutine發(fā)送停止信號,就像下面這樣:
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") }
這種方式可以實現(xiàn)優(yōu)雅地停止goroutine,但是當goroutine特別多的時候,這種方式不管在代碼美觀上還是管理上都顯得笨拙不堪。
2 使用waitgroup
sync包中的Waitgroup結(jié)構(gòu),是Go語言為我們提供的多個goroutine之間同步的好刀。下面是官方文檔對它的描述:
通常情況下,我們像下面這樣使用waitgroup:
創(chuàng)建一個Waitgroup的實例,假設(shè)此處我們叫它wg
在每個goroutine啟動的時候,調(diào)用wg.Add(1),這個操作可以在goroutine啟動之前調(diào)用,也可以在goroutine里面調(diào)用。當然,也可以在創(chuàng)建n個goroutine前調(diào)用wg.Add(n)
當每個goroutine完成任務(wù)后,調(diào)用wg.Done()
在等待所有g(shù)oroutine的地方調(diào)用wg.Wait(),它在所有執(zhí)行了wg.Add(1)的goroutine都調(diào)用完wg.Done()前阻塞,當所有g(shù)oroutine都調(diào)用完wg.Done()之后它會返回。
以上就是golang 如何結(jié)束一個協(xié)程的詳細內(nèi)容,更多請關(guān)注創(chuàng)新互聯(lián)其它相關(guān)文章!