Golang WaitGroup实战应用与控制技巧

本文深入解析了 Go 语言中 sync.WaitGroup 的实战应用与关键控制技巧，从其基于计数器的同步机制（Add、Done、Wait 三方法协同）出发，结合并行 HTTP 请求等真实示例，清晰展示了如何安全高效地等待多个 goroutine 完成；同时直击开发中常见陷阱——如在 goroutine 内 Add 导致竞争、值拷贝引发状态丢失、漏调 Done 引起永久阻塞，并给出 defer wg.Done()、前置 Add、传指针等最佳实践；最后延伸至超时控制等进阶组合用法，帮助读者夯实并发编程基础，写出更健壮、可维护的 Go 代码。

在Go语言开发中，WaitGroup 是 sync 包提供的一个非常实用的同步原语，用于等待一组并发的 goroutine 执行完成。它特别适用于主协程需要等待多个子任务结束后再继续执行的场景。掌握 WaitGroup 的正确用法，是编写高效、稳定并发程序的基础。

WaitGroup 核心机制

WaitGroup 内部维护一个计数器，通过三个方法控制：

• Add(n)：增加计数器值，通常用于添加要等待的 goroutine 数量
• Done()：计数器减1，一般在 goroutine 结束时调用
• Wait()：阻塞当前协程，直到计数器归零

典型使用模式是：主协程调用 Add 设置任务数量，每个子协程执行完后调用 Done，主协程通过 Wait 阻塞等待所有任务完成。

基本使用示例

以下是一个简单的并行HTTP请求示例：

package main
import (
"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://httpbin.org/get",
"https://httpbin.org/delay/1",
"https://httpbin.org/status/200",
}
var wg sync.WaitGroup

for _, url := range urls {
    wg.Add(1)
    go fetchURL(url, &wg)
}

wg.Wait()
fmt.Println("All requests completed.")
}

在这个例子中，每次启动一个 goroutine 前调用 Add(1)，并在函数末尾通过 defer 调用 Done() 确保计数器正确减少。

常见陷阱与最佳实践

使用 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 true
case <-time.After(3 * time.Second):
    return false // 超时
}
}

这种模式将 WaitGroup 的等待逻辑放到单独的协程中，使主流程能支持超时退出。

基本上就这些。WaitGroup 虽然简单，但用好并不容易。关键是理解其计数机制，避免竞态，并结合实际场景灵活运用。在真实项目中，它常用于批量任务处理、服务启动关闭、数据预加载等需要协同等待的场合。

今天关于《Golang WaitGroup实战应用与控制技巧》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！