sync.WaitGroup是被设计用来实现同步问题的一个函数。
WaitGroup维护一个计数器,初始为0。
它有三个方法来对计数器进行操作或判断:Add(), Done(), Wait()
1.Add(n) 把计数器设置为n
2.Done() 每次把计数器-1
3.wait() 会阻塞代码的运行,直到计数器地值减为0。
给出雨痕大佬的例子:
func main() {
wg := new(sync.WaitGroup)
wg.Add(2)
go func() {
defer wg.Done()
for i := 0; i < 6; i++ {
println(i)
if i == 3 {
runtime.Gosched()
}
}
}()
go func() {
defer wg.Done()
println("Hello, World!")
}()
wg.Wait()
}
这里的运行结果如下:
0
1
2
3
Hello, World!
4
5
其中runtime.Gosched()函数的作用是退出当前的go协程,将其暂时挂起等待下次调用。
函数将会先设置计数器为2,进入第一个匿名函数,循环4次后触发Goshced函数挂起当前函数,进入下一个匿名函数,打印Hello,world后触发defer wg.Done(),将计数器-1,此时调用Wait()函数,由于计数器不为0,进入挂起函数,继续打印45,之后触发defer wg.Done(),退出匿名函数后,满足Wait()条件,退出。