在gobyexample中遇到了sync.WaitGroup的例子,很有意思,做个小总结。
作用
sync.WaitGroup如其名,用于主线程等待goroutine都运行完毕。
为什么一定要用sync.WaitGroup,其他方法可以吗?
可以,但是不够好用。
1)time.Sleep,具体的等待时间无法匹配程序运行的随机时间。
2)channel,当然可以,但是大材小用了。而sync.WaitGroup刚刚好。
如何使用
sync.WaitGroup对象内部有一个计数器,最终从0开始,然后通过如下3个方法对计数器数量进行控制。
x.Add(n),把计数器设置为n
x.Done(),表明完成了,将计数器减1
x.Wait(),会堵塞代码的运行,直至计数器为0
避坑
1)go func(&wg) 通过函数传值的时候需要使用地址。
2)计数器不能为负值,会报错