Golang中WaitGroup等待任务完成方法

亲爱的编程学习爱好者，如果你点开了这篇文章，说明你对《Golang中使用WaitGroup等待任务完成方法》很感兴趣。本篇文章就来给大家详细解析一下，主要介绍一下，希望所有认真读完的童鞋们，都有实质性的提高。

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

在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 时有几个关键点需要注意：

• 确保每次 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等待任务完成方法》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！