Go语言sync.WaitGroup如何正确使用才能避免并发竞争？-Golang-PHP中文网

Go语言sync.WaitGroup如何正确使用才能避免并发竞争？

聖光之護

发布： 2025-03-05 08:56:10

原创

524人浏览过

go语言并发控制之sync.waitgroup详解

在go语言并发编程中，sync.waitgroup是用于等待一组goroutine完成的常用工具。本文将深入探讨sync.waitgroup的使用，并解释一个常见的误用案例。

问题代码片段如下：

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)
}

登录后复制

这段代码意图是启动三个goroutine，每个goroutine打印一个数字，然后等待所有goroutine完成。然而，运行结果可能并非预期的那样，goroutine可能在wg.wait()之前就结束了。这是因为wg.add(1)放置在了wgcore函数内部，而这三个goroutine几乎同时启动，wg.add(1)的调用时机与wg.wait()的调用时机存在竞争关系。 wg.wait()并不知道应该等待多少个goroutine完成，因为add操作发生在goroutine内部，而wait发生在主goroutine中，主goroutine无法及时感知到add操作的完成。

修改后的代码如下：

Gnomic智能体平台

国内首家无需魔法免费无限制使用的ChatGPT4.0，网站内设置了大量智能体供大家免费使用，还有五款语言大模型供大家免费使用~

查看详情

立即学习“go语言免费学习笔记（深入）”；

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执行完毕，从而得到预期的结果。

关键在于，sync.waitgroup的计数器需要在启动goroutine之前预先设置好，这保证了wait方法能够准确地跟踪goroutine的完成情况。如果在每个goroutine内部调用add，由于并发执行的特性，主goroutine无法保证在wait之前已经正确计数。

以上就是Go语言sync.WaitGroup如何正确使用才能避免并发竞争？的详细内容，更多请关注php中文网其它相关文章！