WaitGroup通过Add、Done、Wait方法实现goroutine同步，主协程调用Add设置任务数，子协程完成时调用Done，主协程Wait阻塞直至所有任务结束，常用于并发请求、批量处理等场景。
在Go语言开发中，WaitGroup 是 sync 包提供的一个非常实用的同步原语，用于等待一组并发的 goroutine 执行完成。它特别适用于主协程需要等待多个子任务结束后再继续执行的场景。掌握 WaitGroup 的正确用法，是编写高效、稳定并发程序的基础。
WaitGroup 核心机制
WaitGroup 内部维护一个计数器，通过三个方法控制：
Add(n)：增加计数器值，通常用于添加要等待的 goroutine 数量Done()：计数器减1，一般在 goroutine 结束时调用Wait()：阻塞当前协程，直到计数器归零
典型使用模式是：主协程调用 Add 设置任务数量，每个子协程执行完后调用 Done，主协程通过 Wait 阻塞等待所有任务完成。
基本使用示例
以下是一个简单的并行HTTP请求示例：
立即学习“go语言免费学习笔记（深入）”；
package mainimport ("fmt""net/http""sync")
func fetchURL(url string, wg *sync.WaitGroup) {defer wg.Done()
resp, err := http.Get(url)if err != nil { fmt.Printf("Error fetching %s: %v\n", url, err) return}defer resp.Body.Close()fmt.Printf("Success: %s -> Status %d\n", url, resp.StatusCode)}
func main() {urls := []string{"https://www.php.cn/link/4d2fe2e8601f7a8018594d98f28706f2","https://www.php.cn/link/5f69e19efaba426d62faeab93c308f5c","https://www.php.cn/link/98a733901e53052474f2320d0a3a9473",}
var wg sync.WaitGroupfor _, url := range urls { wg.Add(1) go fetchURL(url, &wg)}wg.Wait()fmt.Println("All requests completed.")}
在这个例子中，每次启动一个 goroutine 前调用 Add(1)，并在函数末尾通过 defer 调用 Done() 确保计数器正确减少。
多功能AI图像生成和编辑平台
常见陷阱与最佳实践
使用 WaitGroup 时有几个关键点需要注意：
避免在 goroutine 内部调用 Add，否则可能造成竞争或漏加确保每个 Add 都有对应的 Done，否则会永久阻塞传递 WaitGroup 指针而非值，防止拷贝导致状态不一致合理使用 defer Done()，确保即使发生 panic 也能正确计数
错误示例：在 goroutine 中 Add 会导致不可预测行为
// ❌ 错误做法go func() { wg.Add(1) // 危险！可能在 Wait 后才执行 defer wg.Done() // ...}()正确方式是在启动 goroutine 前完成 Add 操作。
进阶应用场景
WaitGroup 可与其他并发结构组合使用。例如配合 channel 实现超时控制：
func doWithTimeout() bool { var wg sync.WaitGroup done := make(chan bool, 1)for i := 0; i < 5; i++ { wg.Add(1) go func(id int) { defer wg.Done() // 模拟耗时任务 // time.Sleep(time.Second) fmt.Printf("Worker %d done\n", id) }(i)}go func() { wg.Wait() done <- true}()select {case <-done: return truecase <-time.After(3 * time.Second): return false // 超时}}
这种模式将 WaitGroup 的等待逻辑放到单独的协程中，使主流程能支持超时退出。
基本上就这些。WaitGroup 虽然简单，但用好并不容易。关键是理解其计数机制，避免竞态，并结合实际场景灵活运用。在真实项目中，它常用于批量任务处理、服务启动关闭、数据预加载等需要协同等待的场合。
以上就是Golang WaitGroup协作控制实战的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1137064.html
微信扫一扫 
支付宝扫一扫 
