GolangWaitGroup使用与控制技巧详解

**Golang WaitGroup实战应用与控制技巧：轻松玩转并发同步** 在Go语言开发中，`sync.WaitGroup`是实现goroutine同步的利器。本文深入解析`WaitGroup`的核心机制，包括`Add`、`Done`和`Wait`方法的使用，助你掌握并发任务的管理与控制。通过实际案例，例如并行HTTP请求，展示`WaitGroup`在并发请求和批量处理等场景中的应用。同时，本文还剖析了常见陷阱，提供最佳实践，并介绍如何与其他并发结构（如channel）结合实现超时控制等高级功能。掌握`WaitGroup`，编写高效稳定的Go并发程序不再是难题！

WaitGroup通过Add、Done、Wait方法实现goroutine同步，主协程调用Add设置任务数，子协程完成时调用Done，主协程Wait阻塞直至所有任务结束，常用于并发请求、批量处理等场景。

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

WaitGroup 核心机制

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

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

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

基本使用示例

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

package main
<p>import (
"fmt"
"net/http"
"sync"
)</p><p>func fetchURL(url string, wg *sync.WaitGroup) {
defer wg.Done()</p><pre class="brush:php;toolbar:false"><code>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)</code>

}

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)
<pre class="brush:php;toolbar:false"><code>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 // 超时
}</code>

}

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

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

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~