golang中的waitergroup

golang中的waitergroup

在golang中，WaitGroup可以帮助我们等待一组goroutine完成任务。当我们需要等待一组goroutine都执行完毕后，再进行其他操作时，WaitGroup是非常有用的。

WaitGroup是指一个等待任务的group，它是一个特殊的计数器实现。WaitGroup有三个主要的方法：Add、Done和Wait。

首先，我们使用WaitGroup之前，需要导入sync包：

import "sync"

接下来，我们可以创建一个新的WaitGroup对象：

var wg sync.WaitGroup

在开始并发操作之前，我们调用Add方法来设置WaitGroup的计数器。例如，如果我们有10个goroutine需要等待完成，那么我们可以调用Add(10)方法：

wg.Add(10)

然后，我们分别启动10个goroutine，并在每个goroutine中完成任务。当每个goroutine任务完成时，我们需要调用Done方法来表示任务已经完成：

go func() {     // 执行任务     wg.Done() }()

最后，我们在主goroutine中使用Wait方法来阻塞，直到所有goroutine都完成任务：

wg.Wait()

使用WaitGroup的示例

让我们通过一个简单的示例来演示如何使用WaitGroup来等待一组goroutine完成任务。

// 导入所需的包 import (     "fmt"     "sync" ) func main() {     var wg sync.WaitGroup     numJobs := 5     wg.Add(numJobs)     for i := 0; i < numJobs; i++ {         go func(jobIndex int) {             // 模拟任务执行             fmt.Printf("Job %d start\n", jobIndex)             // 模拟耗时操作             time.Sleep(time.Second)             fmt.Printf("Job %d done\n", jobIndex)             wg.Done()         }(i)     }     wg.Wait()     // 所有任务都已完成     fmt.Println("All jobs done") }

在这个示例中，我们创建了一个长度为5的WaitGroup对象。然后，我们使用一个for循环启动了5个goroutine，每个goroutine模拟执行一个任务。每个任务执行完成后，调用Done方法来减少计数器。

最后，我们在主goroutine中调用Wait方法来等待所有goroutine完成任务。当所有任务都完成后，会打印"All jobs done"。

总之，WaitGroup是golang中一个非常有用的工具，可以帮助我们等待一组goroutine完成任务。通过调用Add、Done和Wait方法，我们可以控制并发操作的流程，确保所有任务完成后再进行其他操作。希望本文对你理解和使用WaitGroup有所帮助。