登录
首页 >  Golang >  Go教程

Golang协程同步技巧:WaitGroup使用详解

时间:2025-12-17 08:22:30 345浏览 收藏

推广推荐
免费电影APP ➜
支持 PC / 移动端,安全直达

大家好,我们又见面了啊~本文《Golang协程同步技巧:如何使用WaitGroup》的内容中将会涉及到等等。如果你正在学习Golang相关知识,欢迎关注我,以后会给大家带来更多Golang相关文章,希望我们能一起进步!下面就开始本文的正式内容~

sync.WaitGroup 是 Go 中用于等待一组 goroutine 完成的轻量同步工具,核心为 Add、Done、Wait 三步,需注意计数匹配、避免复制和误调用。

如何使用Golang通过WaitGroup同步协程_Golang协程同步机制讲解

Go 语言中,sync.WaitGroup 是最常用、最轻量的协程(goroutine)同步工具之一,用于等待一组协程执行完毕。它不关心协程内部逻辑,只负责“计数”和“阻塞等待”,适合主协程等待多个子协程完成后再继续执行的场景。

WaitGroup 的核心三步用法

WaitGroup 的使用遵循固定模式:声明 → 增加计数 → 启动协程 → 等待完成。漏掉任意一步都可能导致 panic 或死锁。

  • 声明 WaitGroup 变量:通常在主 goroutine 中定义,如 var wg sync.WaitGroup
  • 调用 wg.Add(n):在启动协程前,把要等待的协程数量加进去(n > 0)。注意不能在协程里调用 Add,否则可能因竞态导致计数错误
  • 协程内调用 wg.Done():每个协程结束前必须调用一次,等价于 wg.Add(-1)。推荐用 defer wg.Done() 确保执行
  • 主协程调用 wg.Wait():会阻塞直到内部计数归零。该调用应放在所有 go 语句之后,且只能被调用一次

常见误用与避坑指南

WaitGroup 表面简单,但几个典型错误会让程序崩溃或卡死:

  • Add 在 goroutine 内调用:因为 Add 和 Wait 之间没有同步保障,可能 Wait 先执行,计数仍为 0,直接返回,后续 Done 导致 panic(计数负值)
  • Done 调用次数不匹配:比如某个分支没执行 defer Done,或重复调用 Done,都会破坏计数平衡
  • Wait 被多次调用:Wait 不是重入安全的,第二次调用会 panic
  • WaitGroup 被复制传递:WaitGroup 不可拷贝(含结构体字段赋值、函数传值、切片 append 等),应始终传指针:func worker(wg *sync.WaitGroup, job int)

一个实用示例:并发请求并汇总结果

模拟向多个 URL 发起 HTTP 请求,并等待全部完成:

func fetchAll(urls []string) {
    var wg sync.WaitGroup
    results := make(chan string, len(urls))
<pre class="brush:php;toolbar:false"><code>for _, url := range urls {
    wg.Add(1)
    go func(u string) {
        defer wg.Done()
        resp, err := http.Get(u)
        if err != nil {
            results &lt;- "error: " + err.Error()
            return
        }
        results &lt;- fmt.Sprintf("success: %s", resp.Status)
    }(url) // 注意:需传参避免闭包引用同一变量
}

go func() {
    wg.Wait()
    close(results)
}()

for res := range results {
    fmt.Println(res)
}</code>

}

这个例子体现了 WaitGroup 和 channel 的配合:WaitGroup 确保所有请求发起并结束,channel 收集结果,避免额外锁或共享变量。

WaitGroup vs 其他同步机制

不是所有并发等待都要用 WaitGroup:

  • 需要精确控制顺序或条件等待:用 sync.Mutex + sync.Condsync.Once
  • 等待带超时或取消:用 context.Context 配合 channel select,WaitGroup 本身不支持超时
  • 只需通知一次事件sync.Once 更合适;若需多次通知,考虑 chan struct{}sync.Cond
  • 等待多个不同信号且有数据传递:优先选 channel,语义更清晰、组合性更强

WaitGroup 的定位很明确:**仅做“数量同步”,不做“通信”或“协调”。用对了,简洁高效;用错了,调试困难。

基本上就这些。WaitGroup 不复杂但容易忽略细节,掌握 Add/Wait/Done 的时机和生命周期,就能稳稳控住并发节奏。

终于介绍完啦!小伙伴们,这篇关于《Golang协程同步技巧:WaitGroup使用详解》的介绍应该让你收获多多了吧!欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识,快来关注吧!

相关阅读
更多>
最新阅读
更多>
课程推荐
更多>