解决go在函数退出后子协程的退出问题
- 作者: 达?矢抾哆拉?
- 来源: 51数据库
- 2021-09-04
该问题来源于自己在读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 函数 退出
以上为个人经验,希望能给大家一个参考,也希望大家多多支持。如有错误或未考虑完全的地方,望不吝赐教。
