Golang协程同步方法 sync.WaitGroup实践-Golang-PHP中文网

Golang协程同步方法 sync.WaitGroup实践

P粉602998670

发布： 2025-08-29 11:36:02

原创

718人浏览过

首先初始化WaitGroup，再通过Add增加计数，每个goroutine执行完调用Done，主线程调用Wait阻塞直至所有任务完成。

golang协程同步方法 sync.waitgroup实践

在Go语言中，sync.WaitGroup 是一种常用的协程同步机制，用于等待一组并发的goroutine执行完成。它特别适用于主线程需要等待多个子任务结束的场景，比如批量发起网络请求、并行处理数据等。使用 WaitGroup 可以避免程序提前退出，确保所有协程都执行完毕。

WaitGroup 核心方法

WaitGroup 提供了三个主要方法：

Add(delta int)：增加计数器的值，通常用于添加要等待的goroutine数量。
Done()：将计数器减1，一般在goroutine结束时调用，相当于 Add(-1)。
Wait()：阻塞当前协程，直到计数器变为0。

基本使用示例

下面是一个简单的例子，展示如何使用 WaitGroup 等待多个goroutine完成：

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)           // 每启动一个goroutine，计数器加1
        go worker(i, &wg)
    }

    wg.Wait() // 等待所有goroutine完成
    fmt.Println("All workers finished")
}

登录后复制

输出结果：

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

法语写作助手

法语助手旗下的AI智能写作平台，支持语法、拼写自动纠错，一键改写、润色你的法语作文。

查看详情

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

使用注意事项

虽然 WaitGroup 使用简单，但有几个关键点需要注意，避免常见错误：

避免Add负数或超出范围：在调用 Add 时传入负数会引发 panic，必须确保 Add 的值合理。
确保 Done 被调用：每个 Add 对应一次 Done，否则 Wait 会一直阻塞，导致死锁。
传递指针：WaitGroup 应以指针形式传递给goroutine，否则值拷贝会导致各个协程操作的是不同的实例。
Add 应在 go 之前调用：如果在goroutine内部才 Add，可能主线程的 Wait 已经开始，导致遗漏。

实际应用场景

WaitGroup 常用于并行任务处理。例如，从多个API获取数据：

func fetchAllData() {
    var wg sync.WaitGroup
    urls := []string{"http://a.com", "http://b.com", "http://c.com"}

    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
            }
            defer resp.Body.Close()
            fmt.Printf("Fetched %s\n", u)
        }(url)
    }

    wg.Wait()
    fmt.Println("All requests completed")
}

登录后复制

这段代码并发请求多个URL，并等待全部完成后再继续。

基本上就这些。WaitGroup 是Go中最基础也最实用的同步工具之一，掌握它的正确用法对编写可靠的并发程序至关重要。不复杂但容易忽略细节。

以上就是Golang协程同步方法 sync.WaitGroup实践的详细内容，更多请关注php中文网其它相关文章！