在go中假如我起了一个goroutine,在主协程中为了让子协程运行完,可以使用 time.Sleep
的方式,主动的等到子协程运行完之后再退出。
但是假如我不知道这个子协程运行的具体时间,这个时候为了让主协程可以等待协程运行完之后退出可以使用 sync.WaitGroup
。
sync.WaitGroup
提供了三个方法:
Add(), Done(), Wait()
第一个Add,可以添加需要等待的协程数量,第二个Done可以让等待的协程数量减一,最后一个Wait可以放在main函数中,效果是当等待的协程数量清零之后才会运行后面的内容。例如:
func main() {
var wg sync.WaitGroup
for i := 0; i < 10; i ++ {
wg.Add(1)
go func(i int) {
defer wg.Done()
fmt.Println(i)
}(i)
}
wg.Wait()
fmt.Print("全部协程都运行完了呢!")
}
运行之后的结果应该类似于:
0
9
3
2
4
7
6
8
5
1
全部协程都运行完了呢!