首先初始化WaitGroup，再通过Add增加计数，每个goroutine执行完调用Done，主线程调用Wait阻塞直至所有任务完成。
在Go语言中，sync.WaitGroup 是一种常用的协程同步机制，用于等待一组并发的goroutine执行完成。它特别适用于主线程需要等待多个子任务结束的场景，比如批量发起网络请求、并行处理数据等。使用 WaitGroup 可以避免程序提前退出，确保所有协程都执行完毕。
WaitGroup 核心方法
WaitGroup 提供了三个主要方法：
- Add(delta int)：增加计数器的值，通常用于添加要等待的goroutine数量。
- Done()：将计数器减1，一般在goroutine结束时调用，相当于 Add(-1)。
- Wait()：阻塞当前协程，直到计数器变为0。
基本使用示例
下面是一个简单的例子，展示如何使用 WaitGroup 等待多个goroutine完成：
package main
import (
"fmt"
"sync"
"time"
)
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done() // 任务完成，计数器减1
fmt.Printf("Worker %d starting\n", id)
time.Sleep(time.Second)
fmt.Printf("Worker %d done\n", id)
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 3; i++ {
wg.Add(1) // 每启动一个goroutine，计数器加1
go worker(i, &wg)
}
wg.Wait() // 等待所有goroutine完成
fmt.Println("All workers finished")
}
输出结果：
立即学习“go语言免费学习笔记（深入）”；
Worker 2 starting
Worker 3 starting
Worker 1 done
Worker 2 done
Worker 3 done
All workers finished
使用注意事项
虽然 WaitGroup 使用简单，但有几个关键点需要注意，避免常见错误：
- 避免Add负数或超出范围：在调用 Add 时传入负数会引发 panic，必须确保 Add 的值合理。
- 确保 Done 被调用：每个 Add 对应一次 Done，否则 Wait 会一直阻塞，导致死锁。
- 传递指针：WaitGroup 应以指针形式传递给goroutine，否则值拷贝会导致各个协程操作的是不同的实例。
- Add 应在 go 之前调用：如果在goroutine内部才 Add，可能主线程的 Wait 已经开始，导致遗漏。
实际应用场景
WaitGroup 常用于并行任务处理。例如，从多个API获取数据：
func fetchAllData() {
var wg sync.WaitGroup
urls := []string{"http://a.com", "http://b.com", "http://c.com"}
for _, url := range urls {
wg.Add(1)
go func(u string) {
defer wg.Done()
resp, err := http.Get(u)
if err != nil {
fmt.Printf("Error: %s\n", err)
return
}
defer resp.Body.Close()
fmt.Printf("Fetched %s\n", u)
}(url)
}
wg.Wait()
fmt.Println("All requests completed")
}
这段代码并发请求多个URL，并等待全部完成后再继续。
基本上就这些。WaitGroup 是Go中最基础也最实用的同步工具之一，掌握它的正确用法对编写可靠的并发程序至关重要。不复杂但容易忽略细节。
