Go语言并发编程中的sync.WaitGroup: Add(1) 使用详解及误区分析
在Go语言的并发编程中，sync.WaitGroup 是一个强大的同步工具，用于等待一组 goroutine 完成。然而，其使用方法中存在一些容易混淆的点，特别是Add()方法的时机选择。本文将通过示例代码深入探讨sync.WaitGroup 的正确使用方法，并解释为什么在某些情况下必须预先设置 Add() 的值。
以下代码展示了一个使用sync.WaitGroup 的函数 wg() 及其辅助函数 wgcore()。在第一个版本中，wg.Add(1) 被放置在 wgcore 函数内部：
func wg() { wg := &sync.WaitGroup{} go wgcore(1, wg) go wgcore(2, wg) go wgcore(3, wg) wg.Wait()}func wgcore(i int64, wg *sync.WaitGroup) { wg.Add(1) defer wg.Done() fmt.Println(i)}
这段代码的预期输出是 1, 2, 3，然后 wg.Wait() 阻塞直到三个 goroutine 都执行完毕。然而，实际结果可能并非如此。这是因为 wg.Add(1) 在 wgcore 函数内部被调用，而 wg.Wait() 并不知道有多少 goroutine 会被启动。每个 wgcore 函数并发执行，wg.Add(1) 的调用时机不确定，wg.Wait() 可能在所有 wg.Add(1) 调用完成之前就执行了 Wait()，导致程序提前结束。
立即学习“go语言免费学习笔记（深入）”；
为了解决这个问题，我们修改代码，将 wg.Add(3) 放置在 wg() 函数的开始部分：
func Wg() { wg := &sync.WaitGroup{} wg.Add(3) go wgCore(1, wg) go wgCore(2, wg) go wgCore(3, wg) wg.Wait()}func wgCore(i int64, wg *sync.WaitGroup) { defer wg.Done() fmt.Println(i)}
在这个修改后的版本中，wg.Add(3) 明确地告诉 WaitGroup 需要等待三个 goroutine 完成。这样，wg.Wait() 就能正确阻塞，直到三个 goroutine 都执行了 wg.Done()，程序才会继续执行，最终打印出预期的结果。
关键在于，sync.WaitGroup 的计数器必须在启动 goroutine 之前 预先设置好，而不是在 goroutine 内部动态增加。如果不事先设置 Add() 的值，WaitGroup 就无法准确跟踪 goroutine 的数量，导致 Wait() 无法正常工作。 defer wg.Done() 确保每个 goroutine 完成后计数器都会正确递减。
以上就是Go语言sync.WaitGroup的Add(1)究竟该如何使用？的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1385206.html
微信扫一扫 
支付宝扫一扫 
