Go语言中等待组的使用方法 在并发编程中,我们经常需要在某些操作完成后再执行其他操作。这时候我们可以使用等待组(WaitGroup)来实现协程的同步。Go语言中提供了很方便的等待组,可以方便地控制协程的并发执行。 等待组的实现原理 等待组实际上就是一个计数器,它初始化为0,每个协程执行前都会调用等待组的Add方法将计数器加上1;协程执行完成后,调用等待组的Done方法将计数器减1。当计数器为0时,等待组会通过阻塞主协程的方式实现等待所有协程完成后再继续执行。 等待组的使用方法 在Go语言中,我们可以使用sync包下的WaitGroup类型来实现等待组。常用的方法有Add、Done和Wait。 Add方法用于给等待组添加协程数量。其参数就是需要添加的协程的数量。在实际使用中,我们可以在主协程中先定义一个等待组,并在需要等待的协程中添加等待组的计数器。 Done方法用于在协程执行完成后将计数器减1。每个协程执行完毕后都需要调用Done方法来将计数器减1。 Wait方法则是等待所有协程执行完毕。当计数器为0时,Wait方法解除阻塞,主协程继续执行后续代码。如果在Wait方法之前已经有协程执行完成并调用了Done方法,那么主协程会立即解除阻塞。 等待组的示例代码 下面是一个简单的示例,使用等待组控制多个协程的并发执行。 ``` package main import ( "fmt" "sync" ) func main() { var wg sync.WaitGroup for i := 1; i <= 3; i++ { wg.Add(1) go func(num int) { fmt.Printf("协程%d开始执行\n", num) wg.Done() fmt.Printf("协程%d执行完毕\n", num) }(i) } wg.Wait() fmt.Println("所有协程执行完毕") } ``` 在上面的代码中,我们首先定义了一个等待组wg,然后依次启动了三个协程,每个协程的执行体中都先输出了开始执行的信息,然后调用Done方法将计数器减1,最后再输出执行完毕的信息。 执行结果为: ``` 协程1开始执行 协程2开始执行 协程3开始执行 协程2执行完毕 协程3执行完毕 协程1执行完毕 所有协程执行完毕 ``` 可以看到,三个协程按照启动的顺序依次执行,并在执行完成后输出了执行完毕的信息。 总结 等待组在Go语言中是非常常用的协程同步工具。它可以方便地控制多个协程的并发执行,等待所有协程执行完成后再执行后续的代码。在实际应用中,我们可以通过定义一个等待组并在每个协程中添加计数器的方式来实现等待组的功能,这样可以大大提高程序的并发能力。