WaitGroup用于等待协程完成，通过Add、Done、Wait方法实现同步。示例中启动3个协程，主协程等待其全部完成。
在go语言中，协程（goroutine）是实现并发的核心机制。但多个协程并行执行时，主程序可能在协程完成前就退出，导致任务丢失。为解决这个问题，sync.WaitGroup 是最常用的协程同步工具。
WaitGroup 是什么？
WaitGroup 用于等待一组协程完成任务。它内部维护一个计数器，通过三个方法控制流程：
- Add(n)：增加计数器，表示要等待 n 个协程
- Done()：计数器减1，通常在协程末尾调用
- Wait()：阻塞主协程，直到计数器归零
基本使用示例
下面是一个简单的例子，启动3个协程并发执行任务，主协程等待它们全部完成：
package main import ( "fmt" "sync" "time" ) func worker(id int, wg *sync.WaitGroup) { defer wg.Done() // 任务完成，计数器减1 fmt.Printf("Worker %d startingn", id) time.Sleep(time.Second) fmt.Printf("Worker %d donen", id) } func main() { var wg sync.WaitGroup for i := 1; i <= 3; i++ { wg.Add(1) // 每启动一个协程，计数+1 go worker(i, &wg) } wg.Wait() // 阻塞，直到所有协程调用 Done() fmt.Println("All workers finished") }
输出结果会是三个 worker 依次打印开始和完成，最后输出 “All workers finished”。
立即学习“go语言免费学习笔记（深入）”；
常见使用技巧和注意事项
使用 WaitGroup 时，有几个关键点要注意，避免死锁或 panic：
- 确保每个 Add 对应一个 Done，否则 Wait 会一直阻塞
- 避免在协程中调用 Add，可能导致主协程还没 Wait 就结束
- 通常将 WaitGroup 指针传入协程函数，避免值拷贝
- 使用 defer wg.Done() 可确保即使发生 panic 也能正确计数
错误示例（可能导致死锁）：
go func() { wg.Add(1) // ❌ 不应在协程内 Add defer wg.Done() // ... }()
正确做法是在 go 之前调用 Add：
wg.Add(1) go func() { defer wg.Done() // 任务逻辑 }()
实际应用场景
WaitGroup 常用于批量处理任务，比如并发请求多个API、并行处理文件、爬虫抓取等。
例如，同时请求多个外部服务：
urls := []string{"http://example1.com", "http://example2.com"} var wg sync.WaitGroup 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: %sn", err) return } fmt.Printf("Success: %s, Status: %sn", u, resp.Status) resp.Body.Close() }(url) } wg.Wait() fmt.Println("All requests completed")
基本上就这些。WaitGroup 简单高效，是Go并发编程中最基础也最重要的同步手段之一。用好它，就能安全地管理多个协程的生命周期。
评论（已关闭）
评论已关闭