Golang协程同步：WaitGroup使用全解析

**Golang协程同步技巧：WaitGroup使用详解** 在Golang并发编程中，`sync.WaitGroup`是至关重要的协程同步工具。本文深入解析`WaitGroup`的原理与使用，通过`Add`、`Done`和`Wait`方法，实现主协程等待多个子协程完成任务。文章详细介绍了`WaitGroup`的工作机制，即通过维护一个计数器来控制协程的同步。`Add(n)`用于增加计数器，表示需要等待n个协程；`Done()`在协程完成时将计数器减1；`Wait()`则阻塞主协程，直到计数器归零。本文还提供了实用的代码示例，展示如何使用`WaitGroup`启动多个协程并确保它们全部完成后主程序才退出，有效避免任务丢失，提升Golang并发程序的稳定性和可靠性。

WaitGroup用于等待协程完成，通过Add、Done、Wait方法实现同步。示例中启动3个协程，主协程等待其全部完成。

在Go语言中，协程（goroutine）是实现并发的核心机制。但多个协程并行执行时，主程序可能在协程完成前就退出，导致任务丢失。为解决这个问题，sync.WaitGroup 是最常用的协程同步工具。

WaitGroup 是什么？

WaitGroup 用于等待一组协程完成任务。它内部维护一个计数器，通过三个方法控制流程：

• Add(n)：增加计数器，表示要等待 n 个协程
• Done()：计数器减1，通常在协程末尾调用
• Wait()：阻塞主协程，直到计数器归零

基本使用示例

下面是一个简单的例子，启动3个协程并发执行任务，主协程等待它们全部完成：

package main

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

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done() // 任务完成，计数器减1
    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 

输出结果会是三个 worker 依次打印开始和完成，最后输出 "All workers finished"。

常见使用技巧和注意事项

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

• 确保每个 Add 对应一个 Done，否则 Wait 会一直阻塞
• 避免在协程中调用 Add，可能导致主协程还没 Wait 就结束
• 通常将 WaitGroup 指针传入协程函数，避免值拷贝
• 使用 defer wg.Done() 可确保即使发生 panic 也能正确计数

错误示例（可能导致死锁）：

go func() {
    wg.Add(1) // ❌ 不应在协程内 Add
    defer wg.Done()
    // ...
}()

正确做法是在 go 之前调用 Add：

wg.Add(1)
go func() {
    defer wg.Done()
    // 任务逻辑
}()

实际应用场景

WaitGroup 常用于批量处理任务，比如并发请求多个API、并行处理文件、爬虫抓取等。

例如，同时请求多个外部服务：

urls := []string{"http://example1.com", "http://example2.com"}

var wg sync.WaitGroup
for _, url := range urls {
    wg.Add(1)
    go func(u string) {
        defer wg.Done()
        resp, err := http.Get(u)
        if err != nil {
            fmt.Printf("Error: %s\n", err)
            return
        }
        fmt.Printf("Success: %s, Status: %s\n", u, resp.Status)
        resp.Body.Close()
    }(url)
}
wg.Wait()
fmt.Println("All requests completed")

基本上就这些。WaitGroup 简单高效，是Go并发编程中最基础也最重要的同步手段之一。用好它，就能安全地管理多个协程的生命周期。

今天关于《Golang协程同步：WaitGroup使用全解析》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！