解決go在函數退出後子協程的退出問題
該問題來源於自己在讀fabric源碼時,看到的一個測試代碼,在一個函數中啟用協程,然後該函數退出瞭,由於平常沒有這樣處理過,以及受原有c++函數域的影響,認為函數退出,子協程應該也退出瞭呀。
這其實是自己對go協程的理解不到位引起的,go的協程作用域不是在某個函數中的,當然,如果那個函數是main函數,就符合要求瞭。
該代碼為solo算法的測試代碼:
func goWithWait(target func()) *waitableGo { wg := &waitableGo{ done: make(chan struct{}), } go func() { target()//該協程會阻塞在這 close(wg.done)//用來對外通知 }() //外邊結束,裡邊還不結束嗎? return wg } // This test checks that if consenter is halted before a timer fires, nothing is actually written. func TestHaltBeforeTimeout(t *testing.T) { batchTimeout, _ := time.ParseDuration("1ms") //support的構造還不清楚 support := &mockmultichannel.ConsenterSupport{ Blocks: make(chan *cb.Block), BlockCutterVal: mockblockcutter.NewReceiver(), SharedConfigVal: &mockconfig.Orderer{BatchTimeoutVal: batchTimeout}, } defer close(support.BlockCutterVal.Block) bs := newChain(support) //bs.main是solo算法的啟動函數,是個死循環,處理函數 wg := goWithWait(bs.main) defer bs.Halt()//中止 syncQueueMessage(testMessage, bs, support.BlockCutterVal) bs.Halt() select { case <-support.Blocks: t.Fatalf("Expected no invocations of Append") case <-wg.done: } }
遇到該問題後,我寫瞭幾個測試:
單純的函數退出,是不會影響協程的
package main import "fmt" var ch chan int func test() int { ch = make(chan int) go func() { for { fmt.Println(<-ch) fmt.Println("hello") } fmt.Println("aaaa") }() //不阻塞,那go func()不會異常退出嗎? //協程並不是函數,不會因為這個函數的退出而退出 //test()啟動一個deadloop子協程,這個會在主協程main結束後被強制退出 return 0 } func main() { c := test() ch <- 10 fmt.Println("c", c) }
我經常在main裡邊直接寫協程的測試demo,main退出會結束主協程,之後會強制結束子協程,一般不會遇到上述在普通函數退出的問題,也沒仔細思考,所以分析源碼時有點困惑。
子協程啟動子協程,父協程的退出,並沒有影響到子協程
liudeMacBook-Pro:~ liu$ cat tmp.go package main import ( "fmt" "time" ) func test() { go func() { //父協程 defer func() { fmt.Println("exit dad") }() go func() { //子協程 defer func() { fmt.Println("exit kid") }() }() }() } func main() { test() time.Sleep(time.Second) } liudeMacBook-Pro:~ liu$ go run tmp.go exit dad exit kid
補充:golang中父子協程生命周期問題,以及通過context優雅關閉子協程
背景
上次基於mysql實現分佈式鎖,今天經過測試發現問題,主要是協程不斷獲取鎖的邏輯存在問題,因為獲取鎖的協程掛掉之後,但其新生成的用來不斷更新鎖的協程並不會退出,導致鎖一直不能被釋放,究其原因如下
原因
通過下面代碼即可說明
fmt.Println("main 函數 開始...") go func() { fmt.Println("父 協程 開始...") go func() { for { fmt.Println("子 協程 執行中...") timer := time.NewTimer(time.Second * 2) <-timer.C } }() time.Sleep(time.Second*5) fmt.Println("父 協程 退出...") }() time.Sleep(time.Second*10) fmt.Println("main 函數 退出")
main 函數 開始…
父 協程 開始…
子 協程 執行中…
子 協程 執行中…
子 協程 執行中…
父 協程 退出…
子 協程 執行中…
子 協程 執行中…
main 函數 退出
由此可以看出:
main 函數退出,所有協程退出
協程無父子關系,即在父協程開啟新的協程,若父協程退出,不影響子協程
解決方式
通過context上下文來解決,當然也可以通過channel管道來解決,context解決方式如下:
fmt.Println("main 函數 開始...") go func() { ctx, cancel := context.WithCancel(context.Background()) defer cancel() fmt.Println("父 協程 開始...") go func(ctx context.Context) { for { for { select { case <-ctx.Done(): fmt.Println("子 協程 接受停止信號...") return default: fmt.Println("子 協程 執行中...") timer := time.NewTimer(time.Second * 2) <-timer.C } } } }(ctx) time.Sleep(time.Second*5) fmt.Println("父 協程 退出...") }() time.Sleep(time.Second*10) fmt.Println("main 函數 退出")
main 函數 開始…
父 協程 開始…
子 協程 執行中…
子 協程 執行中…
子 協程 執行中…
父 協程 退出…
子 協程 接受停止信號…
main 函數 退出
以上為個人經驗,希望能給大傢一個參考,也希望大傢多多支持WalkonNet。如有錯誤或未考慮完全的地方,望不吝賜教。