Go并发(五):WaitGroup

本篇其实是Go并发编程的补充篇章,只是用来介绍WaitGroup的使用介绍。

对于Go的并发编程来说,主协程需要知道其他的协程何时结束。通常我们的做法是采用channel的方式来控制,这种做法很好,可以参见我的另外一个帖子(Go并发之channel篇:https://mp.weixin.qq.com/s/PIb-gGBootc6581pHhi5ew)。但是对于一些简单的协程控制来说,channel显得有些大材小用,而WaitGroup就可以派上用场了。

一、WaitGroup定义

WaitGroup是Go pkg中sync 包里的内容:https://golang.org/pkg/sync/,一个 WaitGroup 会等待一系列 goroutine 直到它们全部运行完毕为止。主 goroutine 通过调用 Add 方法来设置需要等待的 goroutine 数量, 而每个运行的 goroutine 则在它们运行完毕时调用 Done 方法。与此同时, 调用 Wait 方法可以阻塞直到所有 goroutine 都运行完毕为止。

1.主协程对其他协程什么都不处理的例子:

640?wx_fmt=png

通过上面的输出可以看出,主协程并没有等待其他协程的结束,就自己退出了,所以看不到关于其他协程的任何打印。

2.主协程通过超时机制来等待其他协程结束的例子:

640?wx_fmt=png

在主协程中新加了超时1s来等待其他协程的结束,通过打印可以看出来,其他协程已经运行并结束。

3.使用wg的例子:

640?wx_fmt=png

通过打印信息可以看到,主协程通过wg.Wait()来判断计数器有没有变为0,来判断其他的协程有没有结束,在主协程检测到wg的计数器变为0之后,便知道其他协程都已经结束了。

二、WaitGroup中的API介绍

1. (*WaitGroup) Add 方法

func(wg*WaitGroup)Add(delta int) 

delta可以为正数,也可以为负数,在wg对应的计数器没有变为0之前delta可以为负数,但是切记不要让wg计数器变为负数,因为一旦变为负数,便会panic。

通常Add在协程启动之前使用。

2. (*WaitGroup) Done 方法

func(wg*WaitGroup)Done() 

对wg计数器进行减1操作。

通常Done在协程执行结束的时候使用。

3.(*WaitGroup) Wait 方法

func(wg*WaitGroup)Wait()

阻塞直至 WaitGroup 计数器的值为 0 。

通常在主协程中使用,用来等待其他的协程结束。


参考文献:

sync 的pkg包:https://golang.org/pkg/sync/


欢迎关注公众号:灰子学技术

640?wx_fmt=jpeg

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值