Go 语言WaitGroup基础知识

Go 语言WaitGroup基础知识

在实际开发中我们并不能保证每个协程执行的时间，如果需要等待多个协程，全部任务结束之后，再执行某个 业务逻辑，就需要用到WaitGroup；

WaitGroup常用的几个方法：

1. Add：初始化为0，这里直接传入子协程的数量，传入的值会往计数器上加
2. Done：当某个子协程完成之后，可调用者个方法，计数器会减一，就是子协程的数量会减一，通常使用defer来调用
3. 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()
}