Golang协程同步怎么做 sync.WaitGroup实践-Golang-PHP中文网

Golang协程同步怎么做 sync.WaitGroup实践

P粉602998670

发布： 2025-08-25 12:15:01

原创

733人浏览过

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

golang协程同步怎么做 sync.waitgroup实践

在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 <= 3; i++ {
        wg.Add(1)           // 每启动一个协程，计数+1
        go worker(i, &wg)
    }

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

登录后复制

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

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

常见使用技巧和注意事项

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

Vheer

AI图像处理平台

125

查看详情

确保每个 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协程同步怎么做 sync.WaitGroup实践的详细内容，更多请关注php中文网其它相关文章！