快速部署：Go WaitGroup在Golang项目中的应用指南
引言：
在并发编程中，等待所有goroutine完成任务是一项常见的任务。Go WaitGroup是Golang的一个强大的工具，用于实现这个目的。本文将介绍Go WaitGroup的基本概念，并提供一些具体的代码示例，帮助读者快速部署和应用WaitGroup在Golang项目中。
一、什么是WaitGroup？
WaitGroup是Golang中的一个并发原语，它用于等待一组goroutine完成任务。它主要提供了三个方法：
- Add：向WaitGroup中添加一个待等待的goroutine。
- Done：标记一个待等待的goroutine任务已完成。
- Wait：等待所有待等待的goroutine完成任务。
通过这三个方法的配合使用，我们可以方便地等待所有goroutine任务完成，进而进行后续的操作。
立即学习“go语言免费学习笔记（深入）”；
二、代码示例：
为了更好地理解WaitGroup的使用方式和应用场景，下面将给出两个具体的代码示例。
示例一：
在这个例子中，我们使用WaitGroup来等待一组goroutine执行完毕。
package main
import (
"fmt"
"sync"
)
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done()
fmt.Printf("Worker %d starting
", id)
// 模拟一些耗时的操作
for i := 0; i < 5; i++ {
fmt.Printf("Worker %d working...
", id)
}
fmt.Printf("Worker %d finished
", id)
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 5; i++ {
wg.Add(1)
go worker(i, &wg)
}
wg.Wait()
fmt.Println("All workers finished")
}在上述代码中，我们定义了一个worker函数，模拟了一些耗时的操作。在主函数中，我们使用循环创建了5个goroutine，并通过调用wg.Add(1)来向WaitGroup中添加待等待的goroutine。然后，每个goroutine执行完任务后，通过调用wg.Done()标记任务完成。最后，在主函数中调用wg.Wait()等待所有goroutine完成，然后输出"All workers finished"。
示例二：
在这个例子中，我们使用WaitGroup和channel实现主goroutine等待所有子goroutine完成任务并汇总结果。
package main
import (
"fmt"
"sync"
)
func worker(id int, wg *sync.WaitGroup, result chan<- int) {
defer wg.Done()
fmt.Printf("Worker %d starting
", id)
// 模拟一些耗时的操作
sum := 0
for i := 0; i < 5; i++ {
sum += i
}
fmt.Printf("Worker %d finished, result: %d
", id, sum)
result <- sum
}
func main() {
var wg sync.WaitGroup
result := make(chan int, 5)
for i := 1; i <= 5; i++ {
wg.Add(1)
go worker(i, &wg, result)
}
wg.Wait()
close(result)
total := 0
for r := range result {
total += r
}
fmt.Printf("All workers finished, total result: %d
", total)
}在上述代码中，我们在worker函数中将每个goroutine计算结果通过channel传递回主goroutine，最后在主goroutine中汇总所有结果。通过使用WaitGroup和channel的组合，我们能够更灵活地实现goroutine的任务等待和结果汇总。
三、总结：
Go WaitGroup是Golang中实现并发任务等待的重要工具。本文通过给出两个具体的代码示例，帮助读者快速了解和部署WaitGroup在Golang项目中的应用。希望读者能够充分利用这个强大的工具，提升项目的并发性能和可靠性。
