等待组
通过time.Sleep()这种方式去等待协程执行完毕是不可靠的,我们大多数情况下无法估计协程的运行时长,需要有一个能够准确获取到协程结束运行的方法。
在内置包sync
中有一个WaitGroup类型可以应对这种场景,其通过锁机制实现。其原理是在WaitGroup中有一个带锁的计数器,通过其Add()方法可以为计数器累加值;还有一个Done()方法,用于释放释放计数器值(减1);还有一个Wait()方法,该方法会一直阻塞直到计数器归零。
于是就可以这么做,在协程执行前为WaitGroup计数器加1,接着把WaitGroup做为参数传入并运行协程,在协程终止时执行WaitGroup的Done()方法释放计数器。在主方法体中执行WaitGroup的Wait()方法阻塞直到WaitGroup的计数器归零。这样只有当所有协程都执行结束时,计数器才会归零,主方法体也才能解除阻塞状态继续往下运行。
例子:使用WaitGroup等待协程退出
以上代码的执行结果:
最后更新于
这有帮助吗?