GolangWaitGroup同步goroutine实战教程

本文是一篇关于 Golang `sync.WaitGroup` 的实战指南，深入探讨了如何在 Go 语言中使用 `WaitGroup` 实现 goroutine 的同步。`WaitGroup` 通过内部计数器，允许主协程等待一组并发的 goroutine 完成任务，避免程序提前退出。文章详细介绍了 `WaitGroup` 的基本机制，包括 `Add`、`Done` 和 `Wait` 方法的使用，并提供了一个并发下载文件的示例，展示了如何正确地增加计数、减少计数以及阻塞主协程直到所有 goroutine 完成。此外，文章还强调了使用 `WaitGroup` 的注意事项，例如确保 `Done` 调用次数与 `Add` 一致，以及避免复制 `WaitGroup`。最后，还介绍了 `WaitGroup` 与通道结合的扩展用法，以便在等待完成的同时收集结果，帮助开发者编写更可靠的并发程序。

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 并发编程中最基础也最常用的同步工具之一，掌握其正确用法对编写可靠的并发程序至关重要。

本篇关于《GolangWaitGroup同步goroutine实战教程》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！