解決go在函數(shù)退出后子協(xié)程的退出問題

更新時間：2021年04月30日 15:50:48 作者：longtails

這篇文章主要介紹了解決go在函數(shù)退出后子協(xié)程的退出問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

該問題來源于自己在讀fabric源碼時，看到的一個測試代碼，在一個函數(shù)中啟用協(xié)程，然后該函數(shù)退出了，由于平常沒有這樣處理過，以及受原有c++函數(shù)域的影響，認為函數(shù)退出，子協(xié)程應該也退出了呀。

這其實是自己對go協(xié)程的理解不到位引起的，go的協(xié)程作用域不是在某個函數(shù)中的，當然，如果那個函數(shù)是main函數(shù)，就符合要求了。

該代碼為solo算法的測試代碼：

func goWithWait(target func()) *waitableGo {
	wg := &waitableGo{
		done: make(chan struct{}),
	}
	go func() {
		target()//該協(xié)程會阻塞在這
		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算法的啟動函數(shù)，是個死循環(huán)，處理函數(shù)
	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:
	}
}

遇到該問題后，我寫了幾個測試：

單純的函數(shù)退出，是不會影響協(xié)程的

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()不會異常退出嗎？
 //協(xié)程并不是函數(shù)，不會因為這個函數(shù)的退出而退出
 //test()啟動一個deadloop子協(xié)程,這個會在主協(xié)程main結束后被強制退出
 return 0
}
func main() {
 c := test()
 ch <- 10
 fmt.Println("c", c)
}

我經(jīng)常在main里邊直接寫協(xié)程的測試demo,main退出會結束主協(xié)程，之后會強制結束子協(xié)程，一般不會遇到上述在普通函數(shù)退出的問題，也沒仔細思考，所以分析源碼時有點困惑。

子協(xié)程啟動子協(xié)程,父協(xié)程的退出，并沒有影響到子協(xié)程

liudeMacBook-Pro:~ liu$ cat tmp.go 
package main
import (
	"fmt"
	"time"
)
func test() {
	go func() { //父協(xié)程
		defer func() {
			fmt.Println("exit dad")
		}()
		go func() { //子協(xié)程
			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中父子協(xié)程生命周期問題，以及通過context優(yōu)雅關閉子協(xié)程

背景

上次基于mysql實現(xiàn)分布式鎖，今天經(jīng)過測試發(fā)現(xiàn)問題，主要是協(xié)程不斷獲取鎖的邏輯存在問題，因為獲取鎖的協(xié)程掛掉之后，但其新生成的用來不斷更新鎖的協(xié)程并不會退出，導致鎖一直不能被釋放，究其原因如下

原因

通過下面代碼即可說明

	fmt.Println("main 函數(shù) 開始...")
	go func() {
		fmt.Println("父 協(xié)程 開始...")
		go func() {
			for {
				fmt.Println("子 協(xié)程 執(zhí)行中...")
				timer := time.NewTimer(time.Second * 2)
				<-timer.C
			}
		}()
		time.Sleep(time.Second*5)
		fmt.Println("父 協(xié)程 退出...")
	}()
	time.Sleep(time.Second*10)
	fmt.Println("main 函數(shù) 退出")

main 函數(shù) 開始...

父協(xié)程開始...

子協(xié)程執(zhí)行中...

子協(xié)程執(zhí)行中...

子協(xié)程執(zhí)行中...

父協(xié)程退出...

子協(xié)程執(zhí)行中...

子協(xié)程執(zhí)行中...

main 函數(shù) 退出

由此可以看出：

main 函數(shù)退出，所有協(xié)程退出

協(xié)程無父子關系，即在父協(xié)程開啟新的協(xié)程，若父協(xié)程退出，不影響子協(xié)程

解決方式

通過context上下文來解決，當然也可以通過channel管道來解決，context解決方式如下：

fmt.Println("main 函數(shù) 開始...")
	go func() {
		ctx, cancel := context.WithCancel(context.Background())
		defer cancel()
		fmt.Println("父 協(xié)程 開始...")
		go func(ctx context.Context) {
			for {
				for {
					select {
					case <-ctx.Done():
						fmt.Println("子 協(xié)程 接受停止信號...")
						return
					default:
						fmt.Println("子 協(xié)程 執(zhí)行中...")
						timer := time.NewTimer(time.Second * 2)
						<-timer.C
					}
				}
			}
		}(ctx)
		time.Sleep(time.Second*5)
		fmt.Println("父 協(xié)程 退出...")
	}()
	time.Sleep(time.Second*10)
	fmt.Println("main 函數(shù) 退出")