WaitGroup跟java的CountdownLatch以及python里的join方法
差不多,也是阻塞等待所有任务完成之后再继续执行。
WaitGroup的用途:它能够一直等到所有的goroutine执行完成,并且阻塞主线程的执行,直到所有的goroutine执行完成。
WaitGroup总共有三个方法:Add(delta int),Done(),Wait()。简单的说一下这三个方法的作用。
Add:添加或者减少等待goroutine的数量
Done:相当于Add(-1) //wg.Done()最好用defer注册一下,避免函数内部出错执行不到
Wait:执行阻塞,直到所有的WaitGroup数量变成0
package main
import (
"fmt"
"sync"
)
/*
WaitGroup的用途:它能够一直等到所有
的goroutine执行完成,并且阻塞主线程的执行,
直到所有的goroutine执行完成。
*/
var wg sync.WaitGroup
//用于并发的目的函数
func say(v int) {
fmt.Println("talk to me...", v)
wg.Done() //并发的一个任务完成,计数器减一,相当于Add(-1)
//wg.Done()最好用defer注册一下,避免函数内部出错执行不到
}
func main() {
wg.Add(10) //说明要开10个并发,内部计数器,
//也可以再循环内部,每次Add(1)
for i := 0; i < 10; i++ {
go say(i)
}
fmt.Println("I'm main func...")
wg.Wait() //执行阻塞,直到所有的WaitGroup数量变成0
fmt.Println("All concurrent applications are completed...")
}
输出结果与分析: