Golang WaitGroup协作控制实战-Golang-PHP中文网

Golang WaitGroup协作控制实战

P粉602998670

发布： 2025-10-08 08:38:01

原创

683人浏览过

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

golang waitgroup协作控制实战

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

WaitGroup 核心机制

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

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

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

基本使用示例

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

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

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;"><pre class="brush:php;toolbar:false;">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://www.php.cn/link/4d2fe2e8601f7a8018594d98f28706f2", "https://www.php.cn/link/5f69e19efaba426d62faeab93c308f5c", "https://www.php.cn/link/98a733901e53052474f2320d0a3a9473", }

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() 确保计数器正确减少。

卡拉OK视频制作

卡拉OK视频制作，在几分钟内制作出你的卡拉OK视频

178

查看详情

常见陷阱与最佳实践

使用 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;"><pre class="brush:php;toolbar:false;">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协作控制实战的详细内容，更多请关注php中文网其它相关文章！