Golang优雅保持main函数不退出的办法
作者:MiKogy_ 发布时间:2023-07-12 21:15:54
高能预警
本文包含演示部分,请读者自行copy代码编译体验。
参考资料:sync.WaitGroup / signal.Notify / context.CancelFunc
我们有时会希望我们的程序保持执行,但是有一种情况是:我们的代码全部塞入go routine时,主函数会立刻退出,本文将和大家分享如何让main函数优雅地保持执行。
问题演示:
func main() {
go func() {
for i := 0; i<10000;i ++ {
fmt.Println(i)
}
}()
}
此时我们可以看到,控制台几乎不会输出任何内容。究其原因,是主函数在go routine执行前就已经结束,也就是说go routine不会阻塞主函数。
可能有些读者会想到,我直接加个死循环在下面,让主函数不退出不就行啦?博主表示十分赞同,因为博主就是采用这个方法,导致服务器跑满CPU从而不停的告警。
那么解决办法是:让死循环慢一点执行,即添加以下内容:
for {
time.Sleep(time.Second)
}
但是在博主的完美主义光环加持下,还是希望我们的代码能更加优雅,下面将介绍另外三种比较优雅的保持main函数的办法。
解决办法演示
操作系统信号阻塞
先上代码:
func main() {
c := make(chan os.Signal)
signal.Notify(c)
go func() {
fmt.Println("Go routine running")
time.Sleep(3*time.Second)
fmt.Println("Go routine done")
}()
<-c
fmt.Println("bye")
}
官网机翻:signal.Notify()方法使信号将传入c。如果没有提供信号,所有传入的信号将被中继到c。
这里我们创建了一个os.Signal类型的管道。当管道为空的时候,读管道操作“<-”会阻塞住,直到我们向进程发送一个信号(例如 Ctrl+C),才会继续执行该操作后面的代码。
上下文操作阻塞
再上代码:
func main() {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
go func() {
fmt.Println("Go routine running")
time.Sleep(3 * time.Second)
fmt.Println("Go routine done")
cancel()
}()
<-ctx.Done()
fmt.Println("bye")
}
官网机翻:CancelFunc() 通知操作放弃其(当前的)工作。CancelFunc() 不会等待工作停止。
这也是一个十分优雅的办法,我们创建一个可以终止的上下文——context.WithCancel(),并在go routine执行完毕时调用其返回的CancelFunc() 方法,即表示该上下文已经结束了。而在这之前,我们会使用<-ctx.Done()来一直等待上下文的结束,也就是说main函数被成功阻塞,并等待go routine执行完毕并执行了cancel()方法后优雅退出。
WaitGroup阻塞
然后上代码:
func main() {
wg := &sync.WaitGroup{}
wg.Add(2)
go func() {
time.Sleep(3*time.Second)
fmt.Println("3 second passed")
wg.Done()
}()
go func() {
time.Sleep(5*time.Second)
fmt.Println("5 second passed")
wg.Done()
}()
wg.Wait()
fmt.Println("bye")
}
官网机翻:WaitGroup 等待一组 go routine 完成。主 go routine 调用 Add() 来设置要等待的 go routine 的数量。
我们首先创建一个WaitGroup{}对象,然后调用Add()方法,在里面传入我们接下来会创建的go routine的数量,每当我们执行完一个go routine时,调用一次Done()方法,使得正执行的go routine的数量减一,当减到0时,Wait()方法将不再等待(阻塞),使main函数继续向下执行
来源:https://blog.csdn.net/Liing0/article/details/122481451
猜你喜欢
- rss的优点 1.您可以有选择地浏览您感兴趣的以及与您的工作相关的新闻。 2.您可以把需要的信息从不需要的信息(兜售信息,垃圾邮件等)中分离
- IE 浏览器中 CSS Expression 特性的最大的问题:会反复执行,每秒钟可能执行了成百上千次,有严重的性能问题。如何对 CSS E
- 代码如下:SELECT * FROM (SELECT TRUNC(SYSDATE, 'mm') + ROWNUM
- <%Function BytesToBstr(body,Cset)dim objstreamset&n
- 看着这个题目可能大家会有点陌生,不知道是什么,说的直白一点,RDFa就是让你的页面更好的被机器所理解,而不简简单单是展示你的页面,下面就关于
- 对数据库的备份是网站管理人员的必修课,那么常用的数据库备份方式有哪些呢?应如何选择?数据库备份有四种类型,分别应用于不同的场合,下面简要介绍
- 在Web上使用菜单可以极大地节约页面的空间,同时也比较的符合用户从Windows上继承下来的UI操作体验。在以往的Web页菜单设计中,我们普
- 我设了个排卖的起始时间然后设了一个结束时间现在要精确到小时比如提示 “离到期时间:1天XX小时”大家有什么想法吗?我就是在小时总是不能实现,
- 现在基于WEB页的HTML的编辑器在新闻系统,文章系统中用得越来越广,一个网页一粘就可以保持原来的样式,同时图片也可以在这个页中保持。但是在
- PDO::preparePDO::prepare — 准备要执行的SQL语句并返回一个 PDOStatement 对象(PHP 5 >
- 今天做自定义的404页面,需要垂直水平居中,垂直居中不像水平居中那么简单,呵呵。这种效果会在企业站首页,或一些提示性页面经常用到,今天做了个
- 就我个人经验来讲:除了oracle,没有比HP 客户支持中心更糟糕的体验了。刚买不到一个月的HP MINI笔记本,莫名其妙键盘上的引号键松动
- 根据google最新的算法规则:用户行为模式的重视程度越来越高,这也就要求网页设计的时候应注意“合理的网页结构”,SEO业界也有个共识“网页
- 1、新建DLL打开VB6-->文件-->新建工程-->选择ActiveX DLL-->确定2、将默认工程、类重命名工
- 1。注意用SQL分析器可以看select出来的东西select right(convert(varchar(30),getdate(),12
- 如何正确显示数据库里同时存在的GB码和BIG5码? Public Function CheckBIG(strS
- 最近发现数据库服务器压力很大,CPU经常达到100%。查看进程,发现有大量的sp_cursorclose;1进程信息。网上查了下,出现sp_
- 本文将演示如何使用SQL 7.0服务器的Enterprise Manager创建、配置一个示例数据库。我们还会演示如何增加具有合适权限的用户
- 如何同时处理数据库和页面错误? If Err.Number = 0 And ob
- 日前,Mozilla 的 Arun Ranganathan 向 W3C 提交了一个草案,旨在推出一个 JavaScript API,让 Ja