Golang WaitGroup同步goroutine实战教程

WaitGroup用于等待多个goroutine完成，通过Add增加计数、Done减少计数、Wait阻塞直到计数为零，确保主协程正确同步子任务。

在Go语言中，WaitGroup 是 sync 包提供的一个同步原语，用于等待一组并发的 goroutine 完成任务。它非常适合用于主协程需要等待多个子协程执行完毕后再继续的场景。下面通过实际例子说明如何正确使用 WaitGroup 实现多个 goroutine 的同步。

WaitGroup 基本机制

WaitGroup 内部维护一个计数器：

• Add(n)：增加计数器的值，表示要等待 n 个 goroutine
• Done()：在每个 goroutine 结束时调用，相当于计数器减1
• Wait()：阻塞主协程，直到计数器归零

典型使用模式是：主协程启动多个 goroutine 前调用 Add，每个子协程结束前调用 Done，主协程调用 Wait 等待所有完成。

基本使用示例

以下是一个简单的并发下载模拟，使用 WaitGroup 等待所有任务完成：

package main
import (
"fmt"
"sync"
"time"
)
func downloadFile(id int, wg sync.WaitGroup) {
defer wg.Done() // 任务完成，计数器减1
fmt.Printf("开始下载文件 %d\n", id)
time.Sleep(2  time.Second) // 模拟耗时操作
fmt.Printf("完成下载文件 %d\n", id)
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 3; i++ {
    wg.Add(1)           // 增加等待计数
    go downloadFile(i, &wg)
}

wg.Wait() // 阻塞，直到所有 goroutine 调用 Done
fmt.Println("所有文件下载完成")
}

输出结果会显示三个任务并发执行，最后统一输出“所有文件下载完成”。

常见使用注意事项

使用 WaitGroup 时有几个关键点需要注意，避免出现死锁或 panic：

• 确保 Done 调用次数与 Add 一致：少调用会导致 Wait 永不返回，多调用会 panic
• 通常使用 defer wg.Done()：保证即使发生 panic 也能正确释放计数
• WaitGroup 不应被复制：传递给 goroutine 时应传指针
• Add 应在 Wait 之前调用：否则可能引发竞态条件

结合通道的扩展用法

有时除了等待完成，还需要收集结果。可以结合 channel 和 WaitGroup 使用：

func processTasks(tasks []int) []string {
    var wg sync.WaitGroup
    results := make(chan string, len(tasks))
for _, task := range tasks {
    wg.Add(1)
    go func(t int) {
        defer wg.Done()
        result := fmt.Sprintf("处理任务 %d 完成", t)
        results <- result
    }(task)
}

go func() {
    wg.Wait()
    close(results)
}()

var output []string
for res := range results {
    output = append(output, res)
}
return output
}

这种方式既能并发执行，又能安全地收集所有结果，避免主协程提前退出导致 goroutine 被中断。

基本上就这些。WaitGroup 是 Go 并发编程中最基础也最常用的同步工具之一，掌握其正确用法对编写可靠的并发程序至关重要。