如何在Golang中使用WaitGroup等待任务完成-Golang-PHP中文网

如何在Golang中使用WaitGroup等待任务完成

P粉602998670

发布： 2025-10-16 10:07:02

原创

277人浏览过

WaitGroup用于等待多个goroutine完成，通过Add增加计数、Done减少计数、Wait阻塞直至计数为零，需确保Add与Done配对使用，避免并发错误。

如何在golang中使用waitgroup等待任务完成

在Go语言中，WaitGroup 是 sync 包提供的一个同步原语，用于等待一组并发的goroutine执行完成。它特别适用于主协程需要等待多个子任务结束后再继续的场景。

WaitGroup 基本用法

WaitGroup 有三个核心方法：

Add(delta int)：增加计数器，通常传入要启动的goroutine数量。
Done()：在每个goroutine结束时调用，表示该任务已完成（相当于 Add(-1)）。
Wait()：阻塞当前协程，直到计数器归零。

典型使用模式如下：

var wg sync.WaitGroup

for i := 0; i < 5; i++ {
    wg.Add(1)
    go func(id int) {
        defer wg.Done()
        // 模拟任务执行
        fmt.Printf("任务 %d 完成\n", id)
    }(i)
}

wg.Wait() // 等待所有任务完成
fmt.Println("所有任务已结束")

登录后复制

避免常见错误

使用 WaitGroup 时有几个关键点需要注意：

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

PPT.CN,PPTCN,PPT.CN是什么,PPT.CN官网,PPT.CN如何使用

一键操作，智能生成专业级PPT

查看详情

确保每次 Add 都有对应的 Done，否则程序可能永久阻塞或 panic。
不要在 goroutine 外部调用 Done，应放在 goroutine 内部并配合 defer 使用。
Add 调用可以在启动 goroutine 前完成，但不能晚于 Wait。

例如，下面这种写法是安全的：

wg.Add(3)
go task(&wg)
go task(&wg)
go task(&wg)
wg.Wait()

登录后复制

实际应用场景

WaitGroup 常用于批量请求、数据抓取、并行处理等场景。比如同时发起多个HTTP请求：

var wg sync.WaitGroup
urls := []string{"http://example.com", "http://httpbin.org", "http://httpstat.us"}

for _, url := range urls {
    wg.Add(1)
    go func(u string) {
        defer wg.Done()
        resp, err := http.Get(u)
        if err != nil {
            fmt.Printf("请求失败 %s: %v\n", u, err)
            return
        }
        defer resp.Body.Close()
        fmt.Printf("成功获取 %s, 状态码: %s\n", u, resp.Status)
    }(url)
}

wg.Wait()
fmt.Println("所有请求完成")

登录后复制

基本上就这些。只要记得配对使用 Add 和 Done，并在主线程调用 Wait，就能正确等待所有任务结束。不复杂但容易忽略细节。

以上就是如何在Golang中使用WaitGroup等待任务完成的详细内容，更多请关注php中文网其它相关文章！