如何在Golang中使用sync.WaitGroup等待并发完成-Golang-PHP中文网

如何在Golang中使用sync.WaitGroup等待并发完成

P粉602998670

发布： 2025-10-18 13:54:02

原创

494人浏览过

答案：sync.WaitGroup用于等待一组goroutine完成，通过Add增加计数，Done减少计数，Wait阻塞直至计数归零。示例中启动5个worker，主协程等待全部完成。

如何在golang中使用sync.waitgroup等待并发完成

在Golang中，sync.WaitGroup 是一种常用的同步机制，用于等待一组并发的goroutine执行完成。它特别适合在主函数或主线程中启动多个子任务，并确保所有任务都结束后再继续执行后续逻辑。

基本原理

WaitGroup通过计数器来跟踪正在运行的goroutine数量。调用 Add(n) 增加计数器，每个goroutine执行完成后调用 Done() 减少计数器，而 Wait() 会阻塞直到计数器归零。

使用步骤和示例

以下是使用 WaitGroup 的典型流程：

创建 WaitGroup 变量：通常是一个指针或值类型变量。
在启动goroutine前调用 Add：每次启动一个goroutine就调用 Add(1)。
在每个goroutine末尾调用 Done：确保无论正常结束还是出错都能触发 Done。
在需要等待的地方调用 Wait：主协程等待所有任务完成。

示例代码：

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

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

查看详情

package main

import (
    "fmt"
    "sync"
    "time"
)

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done() // 任务完成时减一
    fmt.Printf("Worker %d starting\n", id)
    time.Sleep(time.Second) // 模拟工作
    fmt.Printf("Worker %d done\n", id)
}

func main() {
    var wg sync.WaitGroup

    for i := 1; i <= 3; i++ {
        wg.Add(1)           // 增加计数
        go worker(i, &wg)   // 启动goroutine
    }

    wg.Wait() // 阻塞，直到所有worker调用Done
    fmt.Println("All workers finished")
}

登录后复制

注意事项

正确使用 WaitGroup 要注意以下几点：

避免重复调用 Done：每个 goroutine 只应调用一次 Done，否则会导致 panic。
Add 的值不能为负：除了初始设置外，Add传入负数会引发错误。
确保 Wait 在所有 Add 之后调用：如果 Wait 先于 Add 执行，可能无法正确捕获所有任务。
传递 WaitGroup 应使用指针：在函数间传递时，应传指针以避免副本问题。

常见误用与修复

一个常见错误是将 wg 以值方式传入函数，导致每个 goroutine 操作的是副本：

<code>// 错误示例
go func(wg sync.WaitGroup) { // 值传递，复制了wg
    defer wg.Done()
}(wg)

登录后复制

应改为指针传递：

go func(wg *sync.WaitGroup) {
    defer wg.Done()
}(&wg)

登录后复制

基本上就这些。只要掌握 Add、Done、Wait 的配合，就能安全地协调并发任务的完成。

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