解决go在函数退出后子协程的退出问题


Posted in Golang onApril 30, 2021

该问题来源于自己在读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 函数 退出

以上为个人经验,希望能给大家一个参考,也希望大家多多支持三水点靠木。如有错误或未考虑完全的地方,望不吝赐教。

Golang 相关文章推荐
Go语言带缓冲的通道实现
Apr 26 Golang
go语言中切片与内存复制 memcpy 的实现操作
Apr 27 Golang
golang json数组拼接的实例
Apr 28 Golang
Golang 使用Map实现去重与set的功能操作
Apr 29 Golang
Goland使用Go Modules创建/管理项目的操作
May 06 Golang
浅谈Golang 切片(slice)扩容机制的原理
Jun 09 Golang
深入理解go缓存库freecache的使用
Feb 15 Golang
golang为什么要统一错误处理
Apr 03 Golang
Golang 遍历二叉树
Apr 19 Golang
Golang 实现WebSockets
Apr 24 Golang
Go语言入门exec的基本使用
May 20 Golang
Golang实现可重入锁的示例代码
May 25 Golang
Go语言 go程释放操作(退出/销毁)
golang DNS服务器的简单实现操作
golang slice元素去重操作
Apr 30 #Golang
Golang中interface{}转为数组的操作
Apr 30 #Golang
解决Go gorm踩过的坑
Apr 30 #Golang
Golang 如何实现函数的任意类型传参
Apr 29 #Golang
解决Golang time.Parse和time.Format的时区问题
Apr 29 #Golang
You might like
聊天室php&amp;mysql(一)
2006/10/09 PHP
PHP二维数组实现去除重复项的方法【保留各个键值】
2017/12/21 PHP
PHP的PDO大对象(LOBs)
2019/01/27 PHP
PHP上传图片到数据库并显示的实例代码
2019/12/20 PHP
php中yii框架实例用法
2020/12/22 PHP
Aster vs Newbee BO5 第三场2.19
2021/03/10 DOTA
初学Javascript的一些总结
2008/11/03 Javascript
javascript 读取图片文件的大小
2009/06/25 Javascript
javascript 混合的构造函数和原型方式,动态原型方式
2009/12/07 Javascript
js 蒙版进度条(结合图片)
2010/03/10 Javascript
prettify 代码高亮着色器google出品
2010/12/28 Javascript
JavaScript初学者应注意的七个细节详细介绍
2012/12/27 Javascript
jquery浏览器滚动加载技术实现方案
2014/06/03 Javascript
VS2008中使用JavaScript调用WebServices
2014/12/18 Javascript
javascript省市区三级联动下拉框菜单实例演示
2015/11/29 Javascript
Bootstrap学习笔记之css样式设计(1)
2016/06/07 Javascript
JavaScript Promise 用法
2016/06/14 Javascript
Bootstrap前端开发案例二
2016/06/17 Javascript
全面了解JS中的匿名函数
2016/06/29 Javascript
js实现带简单弹性运动的导航条
2017/02/22 Javascript
基于vue,vue-router, vuex及addRoutes进行权限控制问题
2018/05/02 Javascript
[03:57]2016完美“圣”典风云人物:rOtk专访
2016/12/09 DOTA
[34:39]Secret vs VG 2018国际邀请赛淘汰赛BO3 第二场 8.23
2018/08/24 DOTA
微信跳一跳python自动代码解读1.0
2018/01/12 Python
对pyqt5中QTabWidget的相关操作详解
2019/06/21 Python
django 基于中间件实现限制ip频繁访问过程详解
2019/07/30 Python
Python Scrapy框架第一个入门程序示例
2020/02/05 Python
Django-silk性能测试工具安装及使用解析
2020/11/28 Python
html5实现输入框fixed定位在屏幕最底部兼容性
2020/07/03 HTML / CSS
消防安全责任书
2014/04/14 职场文书
2014年文学毕业生自我鉴定
2014/04/23 职场文书
公司担保书格式范文
2014/05/12 职场文书
2014教师研修学习体会
2014/07/08 职场文书
2015年后勤工作总结范文
2015/04/08 职场文书
2015年保洁工作总结范文
2015/04/28 职场文书
离职证明范本
2015/06/12 职场文书