Golang 中的 WaitGroup 有什么用_Golang 并发协程等待机制详解

WaitGroup用于等待多个goroutine完成，通过Add增加计数、Done减少计数、Wait阻塞直到计数为零，确保主协程在所有子任务结束后再继续执行。

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

WaitGroup 的核心作用

当启动多个 goroutine 做并行任务时，主协程通常不能立即退出，否则会终止所有子协程。WaitGroup 提供了一种机制，让主协程“等待”所有子任务完成后再继续或退出。

其本质是一个计数器：

• 每启动一个 goroutine，调用 Add(1) 增加计数
• 每个 goroutine 执行完后，调用 Done() 减少计数
• 主协程调用 Wait() 阻塞，直到计数归零

基本使用方法

下面是一个典型的 WaitGroup 使用示例：

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

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)
    }

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

输出结果：

有道小P

有道小P，新一代AI全科学习助手，在学习中遇到任何问题都可以问我。

64

查看详情

Worker 1 starting
Worker 2 starting
Worker 3 starting
Worker 1 done
Worker 2 done
Worker 3 done
All workers finished

使用注意事项

WaitGroup 虽然简单，但使用不当容易出错。以下几点需特别注意：

• Add 的值不能为负数：否则会 panic
• 确保 Done 调用次数与 Add 一致：漏调或多调都会导致死锁或 panic
• WaitGroup 不是可复制类型：不要将其作为参数值传递，应传指针
• Add 应在 Wait 之前调用：如果在 Wait 后 Add，可能引发竞态条件

常见误用与修复

错误写法：在 goroutine 内部才 Add

for i := range tasks {
    go func() {
        wg.Add(1)     // 错误！Add 在 goroutine 内
        defer wg.Done()
        // ...
    }()
}
wg.Wait()

这可能导致主协程已进入 Wait，而 Add 还未执行，从而漏记协程。

正确做法：在 goroutine 启动前调用 Add

for i := range tasks {
    wg.Add(1)
    go func() {
        defer wg.Done()
        // ...
    }()
}
wg.Wait()

基本上就这些。WaitGroup 是 Go 并发编程中最基础也最实用的同步工具之一，掌握它的使用和边界条件，能有效避免程序提前退出或死锁问题。关键是控制好计数的增减时机，保证一致性。不复杂但容易忽略细节。

以上就是Golang 中的 WaitGroup 有什么用_Golang 并发协程等待机制详解的详细内容，更多请关注php中文网其它相关文章！