Go并发(五):WaitGroup

2023-10-30 15:33:03 浏览数 (1)

本篇其实是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.主协程对其他协程什么都不处理的例子:

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

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

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

3.使用wg的例子:

通过打印信息可以看到,主协程通过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/

0 人点赞