商城首页欢迎来到中国正版软件门户

您的位置:首页 >如何在Golang中使用WaitGroup等待任务完成

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

  发布于2025-11-18 阅读(0)

扫一扫,手机访问

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

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

在Go语言中,WaitGroupsync 包提供的一个同步原语,用于等待一组并发的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,就能正确等待所有任务结束。不复杂但容易忽略细节。

本文转载于:互联网 如有侵犯,请联系zhengruancom@outlook.com删除。
免责声明:正软商城发布此文仅为传递信息,不代表正软商城认同其观点或证实其描述。

热门关注