0

0

sync.WaitGroup在Golang中如何等待所有goroutine完成任务

P粉602998670

P粉602998670

发布时间:2025-08-30 09:46:01

|

549人浏览过

|

来源于php中文网

原创

sync.WaitGroup通过计数器协调goroutine完成。1. 初始化后,在启动goroutine前调用Add增加计数;2. 每个goroutine结束时调用Done减少计数,推荐使用defer wg.Done()确保执行;3. 主goroutine调用Wait阻塞,直到计数器归零。常见陷阱是Add调用时机错误,必须在go语句前执行,否则可能导致竞态条件或死锁。最佳实践包括:始终在启动goroutine前Add,使用defer wg.Done()保证通知,避免复制WaitGroup。

sync.waitgroup在golang中如何等待所有goroutine完成任务

sync.WaitGroup
在Golang中主要通过一个内部计数器来协调goroutine的完成。当需要等待一组goroutine时,我们会在启动它们之前调用
Add
方法增加计数,每个goroutine完成任务后调用
Done
方法递减计数。主goroutine则通过调用
Wait
方法阻塞,直到这个计数器归零,表示所有被追踪的goroutine都已完成。

当我们在Golang中处理并发任务时,经常会遇到一个场景:启动多个goroutine去执行一些工作,然后主goroutine需要等待所有这些子goroutine都完成,才能继续执行后续逻辑,或者安全地退出程序。

sync.WaitGroup
就是为此而生的。

它的核心机制其实很简单:

  1. 初始化:声明一个
    var wg sync.WaitGroup
  2. 计数器增加:在启动任何需要等待的goroutine之前,调用
    wg.Add(delta int)
    。这个
    delta
    通常是你即将启动的goroutine数量。每调用一次
    Add(1)
    ,内部计数器就加一。
  3. 任务完成通知:在每个goroutine内部,当其任务即将完成时,调用
    wg.Done()
    。这会将内部计数器减一。一个非常好的实践是使用
    defer wg.Done()
    ,这样即使goroutine内部发生panic,
    Done()
    也能被执行,避免主goroutine永远等待。
  4. 等待:在主goroutine中,调用
    wg.Wait()
    。这个方法会阻塞当前goroutine,直到
    WaitGroup
    的内部计数器变为零。

来看个例子,这能让你更直观地理解:

立即学习go语言免费学习笔记(深入)”;

稿定AI绘图
稿定AI绘图

稿定推出的AI绘画工具

下载
package main

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

func worker(id int, wg *sync.WaitGroup) {
    // 确保无论如何,这个goroutine完成时都会通知WaitGroup
    defer wg.Done() 
    fmt.Printf("Worker %d starting...\n", id)
    time.Sleep(time.Duration(id) * time.Second) // 模拟工作
    fmt.Printf("Worker %d finished.\n", id)
}

func main() {
    var wg sync.WaitGroup
    numWorkers := 3

    fmt.Println("Main: Launching workers...")
    for i := 1; i <= numWorkers; i++ {
        wg.Add(1) // 每启动一个worker,计数器加1
        go worker(i, &wg)
    }

    fmt.Println("Main: Waiting for all workers to complete...")
    wg.Wait() // 阻塞直到所有worker都调用了Done()

    fmt.Println("Main: All workers completed. Exiting.")
}

运行这段代码,你会看到主goroutine在启动所有

worker
之后,会一直等到所有
worker
都打印出"finished"才继续执行
"Main: All workers completed. Exiting."
。这正是我们想要的效果。

我个人在使用

WaitGroup
时,最常犯的一个小错误,或者说容易忽略的细节,就是
Add
的时机。它必须在对应的
go
语句之前调用。如果你在
go worker(i, &wg)
之后才
wg.Add(1)
,那么在某些极端情况下,如果
worker
执行得特别快,可能在
Add
被调用之前就已经
Done
了,这会导致计数器不匹配,
Wait
可能会提前返回或者永远等待。所以,顺序很重要。

Golang
sync.WaitGroup
有哪些常见的陷阱和最佳实践?

在使用

sync.WaitGroup
时,虽然它看起来直观,但

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

178

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

226

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

337

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

208

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

388

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

194

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

189

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

192

2025.06.17

c++主流开发框架汇总
c++主流开发框架汇总

本专题整合了c++开发框架推荐,阅读专题下面的文章了解更多详细内容。

97

2026.01.09

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Go 教程
Go 教程

共32课时 | 3.6万人学习

Go语言实战之 GraphQL
Go语言实战之 GraphQL

共10课时 | 0.8万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号