Go 语言WaitGroup基础知识
在实际开发中我们并不能保证每个协程执行的时间,如果需要等待多个协程,全部任务结束之后,再执行某个 业务逻辑,就需要用到WaitGroup;
WaitGroup常用的几个方法:
- Add:初始化为0,这里直接传入子协程的数量,传入的值会往计数器上加
- Done:当某个子协程完成之后,可调用者个方法,计数器会减一,就是子协程的数量会减一,通常使用defer来调用
- Wait:阻塞当前协程,直到实例里的计数器归零
使用信道
信道可以实现多个协程间的通信,于是我们可以定义一个信道,在执行完成之后,往信道中写入true,然后在主协程中获取到true,就可以认为子协程已经执行完成
func main() {
isDone := make(chan bool)
go func() {
for i := 0; i < 5; i++ {
fmt.Println(i)
}
isDone <- true
}()
<-isDone
fmt.Println("main group finished")
}
使用WaitGroup
使用sync包中的WaitGroup类型,WaitGroup用于等待一批Go协程执行结束,程序控制会一直阻塞,直到这些子协程全部执行完成,当然WaitGroup也可以用来实现线程池
func DoTask(taskNum int, wg *sync.WaitGroup) {
defer wg.Done()
for i := 0; i < 3; i++ {
fmt.Printf("task: %d : %d\n", taskNum, i)
}
}
func main() {
//初始化sync.WaitGroup
var wg sync.WaitGroup
//传入子协程数量
wg.Add(3)
//开启子协程1
go DoTask(1, &wg)
//开启子协程2
go DoTask(2, &wg)
//开启子协程3
go DoTask(3, &wg)
//阻塞当前协程,等待所有子协程执行完成
wg.Wait()
}