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有所帮助。
评论