Go并发编程利器：一文彻底搞定sync.WaitGroup的正确用法

2026/05/16 | 默认 | 玄火 | 1629 阅读 | 0 评论

Go并发编程利器：一文彻底搞定sync.WaitGroup的正确用法

在Go语言的并发编程世界里，Goroutine（协程）无疑是最耀眼的明星。它以极低的成本让开发者能够轻松创建成千上万个并发任务。然而，正如拥有了无数匹烈马却不懂得如何驾驭一样，如果无法有效地管理这些并发任务的生命周期，程序就会陷入混乱。

很多时候，我们启动了多个Goroutine来并行处理任务，但主程序却因为不知道它们何时结束而提前退出，导致任务半途而废。为了解决这一核心痛点，Go语言在标准库sync包中为我们提供了一个强有力的工具——sync.WaitGroup。今天，我们就来深入剖析它的用法与最佳实践。

什么是WaitGroup？

简单来说，sync.WaitGroup是一个并发原语，用于等待一组Goroutine的结束。形象地理解，它就像是一个计数器或者是“班级点名册”。

假设你是一名老师（主Goroutine），安排了10名学生（子Goroutine）去做作业。你肯定不能在学生刚拿起笔时就离开教室，你需要等待所有人都交卷后才能离开。WaitGroup就是那个帮你记录还有多少人没交卷的工具。当计数器归零时，意味着所有任务完成，主程序就可以继续往下执行或安全退出。

核心方法详解

WaitGroup的使用非常简单，它只有三个核心方法：

随机图片

  1. Add(delta int)：给计数器加减delta值。通常我们在启动一个Goroutine之前，调用Add(1)，表示增加一个待等待的任务。
  2. Done()：表示一个任务已完成。它实际上是对Add(-1)的封装。通常在Goroutine内部，任务执行完毕后调用。
  3. Wait()：阻塞当前的Goroutine，直到计数器归零。如果计数器为0，该方法会立即返回。

实战演练：从串行到并发

让我们通过一段代码来看看WaitGroup是如何工作的。假设我们需要执行三个耗时的任务。

package main

import (
    "fmt"
    "sync"
    "time"
)

func worker(id int, wg *sync.WaitGroup) {
    // 函数退出时调用Done，确保计数器最终减少
    defer wg.Done()

    fmt.Printf("Worker %d: 开始工作...\n", id)
    time.Sleep(time.Second) // 模拟耗时操作
    fmt.Printf("Worker %d: 工作完成！\n", id)
}

func main() {
    // 声明一个WaitGroup
    var wg sync.WaitGroup

    fmt.Println("主程序：开始派遣任务")

    // 预先设定等待3个任务
    wg.Add(3)

    for i := 1; i <= 3; i++ {
        go worker(i, &wg)
    }

    fmt.Println("主程序：等待所有任务完成...")

    // 阻塞等待
    wg.Wait()

    fmt.Println("主程序：所有任务已完成，程序退出")
}

在这个例子中，主程序在wg.Wait()处阻塞，直到三个worker都调用了wg.Done()，计数器变为0，阻塞解除。如果不使用WaitGroup，主程序可能在Goroutine还没运行起来就已经退出了。

避坑指南：使用中的常见错误

虽然WaitGroup原理简单，但在实际开发中，很多新手容易踩进以下几个坑：

1. 计数器为负数 WaitGroup的计数器不能为负数。如果你调用了Done()的次数超过了Add()的次数，程序会触发Panic（恐慌）。这通常是因为Add和Done的数量不匹配导致的，务必保证逻辑上的成对出现。

2. Add的位置放错了 这是最经典的错误。请看下面这段有问题的代码：

for i := 0; i < 3; i++ {
    go func() {
        wg.Add(1) // 错误！在Goroutine内部Add
        // ... do work
        wg.Done()
    }()
}
wg.Wait()

为什么这样写有问题？因为go关键字启动一个Goroutine是需要调度时间的。主程序很可能在任何一个Goroutine真正运行到wg.Add(1)之前，就已经执行到了wg.Wait()。此时计数器还是0，主程序会直接退出，根本等不到子任务。 正确的做法是：在启动Goroutine之前，先调用Add。

3. 值拷贝导致的失效 sync.WaitGroup内部包含一个不允许复制的状态字段。如果你将WaitGroup作为值参数传递给函数，函数内部操作的就只是一个副本，原本的计数器不会发生变化，Wait()将永远阻塞。 *正确的做法是：必须通过指针传递（`sync.WaitGroup`）。**

进阶技巧：WaitGroup与Channel的配合

虽然WaitGroup擅长等待一组任务结束，但它本身不具备返回值传递或错误处理的能力。在复杂的业务场景中，我们通常会结合Channel使用。

例如，我们需要并发抓取多个网页，并等待所有结果汇总。可以使用WaitGroup控制并发节奏，配合一个结果Channel来收集数据。甚至在某些架构设计中，使用errGroup（golang.org/x/sync/errgroup）库，它是基于WaitGroup的封装，能够优雅地处理子任务中的错误传播，一旦某个任务出错，可以取消其他正在运行的任务。

总结

sync.WaitGroup是Go语言并发编程中不可或缺的基石。它以一种极其优雅的方式解决了“等待一组任务完成”这一高频需求。掌握它的关键在于理解计数器的流转逻辑，并严格遵守“先Add、后Done、指针传参”的原则。

当你能够熟练地运用WaitGroup来编排成百上千个Goroutine时，你才算真正踏入了Go语言并发编程的大门。它让你的程序既能享受并发带来的速度红利，又能保持逻辑的清晰与可控。希望这篇文章能让你对WaitGroup有一个全面而深刻的认识！

发表评论