0

0

Golangsync.WaitGroup并发任务同步技巧

P粉602998670

P粉602998670

发布时间:2025-09-17 15:28:01

|

356人浏览过

|

来源于php中文网

原创

WaitGroup通过计数器协调并发任务,Add在启动前增加计数,Done在协程结束时减1,Wait阻塞主协程直至计数归零,需避免在goroutine中调用Add或重复使用未重置的WaitGroup,结合context可实现超时控制,确保程序健壮性。

golangsync.waitgroup并发任务同步技巧

在Go语言中,sync.WaitGroup 是处理并发任务同步最常用的工具之一。它能确保主协程等待所有子协程完成后再继续执行,避免了因主程序提前退出而导致子任务未完成的问题。掌握其正确用法,对编写稳定高效的并发程序至关重要。

理解 WaitGroup 的基本机制

WaitGroup 本质上是一个计数器,用于等待一组协程结束。它有三个核心方法:Add(delta int)Done()Wait()

  • Add:增加计数器的值,通常在启动协程前调用,表示要等待的任务数量。
  • Done:在协程内部调用,将计数器减1,通常用 defer 触发。
  • Wait:阻塞主协程,直到计数器归零。

使用时需注意:Add 应在 goroutine 启动前调用,否则可能引发竞态条件。

典型使用模式

最常见的场景是批量启动多个 goroutine 并等待它们完成。例如发起多个HTTP请求或并行处理数据块。

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

示例代码:
var wg sync.WaitGroup
for i := 0; i < 5; i++ {
    wg.Add(1)
    go func(id int) {
        defer wg.Done()
        // 模拟任务执行
        fmt.Printf("Worker %d done\n", id)
    }(i)
}
wg.Wait()
fmt.Println("All workers finished")

这里每个 goroutine 执行完都会调用 Done,主程序在 Wait 处阻塞直到全部完成。

避免常见错误

WaitGroup 虽然简单,但误用容易导致程序死锁或 panic。

KPPW客客出品专业威客系统
KPPW客客出品专业威客系统

客客出品专业威客系统英文名称KPPW,也是keke produced professional witkey的缩写。KPPW是一款基于PHP+MYSQL技术构架的威客系统,积客客团队多年实践和对威客模式商业化运作的大量调查分析而精心策划研发,是您轻松搭建威客网站的首选利器。KPPW针对威客任务和商品交易模式进行了细致的分析,提供完善威客任务流程控制解决方案,并将逐步分享威客系统专业化应用作为我们的

下载
  • 不要在 goroutine 中调用 Add:这可能导致 Wait 已开始而 Add 还未执行,造成漏计数。
  • 确保 Done 被调用且仅调用一次:多次调用会导致 panic,遗漏则使程序永远阻塞。
  • 不要重复使用未重置的 WaitGroup:如需复用,应配合 sync.Pool 或重新声明。

若任务数量动态变化,建议在外部确定总数后再调用 Add,而不是在循环内边加边启。

与 Context 配合实现超时控制

WaitGroup 本身不支持超时,但可结合 context 实现更安全的等待。

通过 select 监听 context.Done(),可在指定时间内等待任务完成,超时则放弃。

示例:
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
defer cancel()

var wg sync.WaitGroup ch := make(chan struct{})

go func() { for i := 0; i < 3; i++ { wg.Add(1) go func(id int) { defer wg.Done() time.Sleep(time.Duration(rand.Intn(3)) * time.Second) fmt.Printf("Task %d completed\n", id) }(i) } wg.Wait() close(ch) }()

select { case <-ch: fmt.Println("All tasks completed") case <-ctx.Done(): fmt.Println("Timeout, some tasks may not finish") }

这种方式提升了程序健壮性,防止无限等待。

基本上就这些。只要遵循 Add 在前、Done 配合 defer、避免协程内修改计数的原则,WaitGroup 就能可靠地帮你管理并发任务生命周期。

相关专题

更多
string转int
string转int

在编程中,我们经常会遇到需要将字符串(str)转换为整数(int)的情况。这可能是因为我们需要对字符串进行数值计算,或者需要将用户输入的字符串转换为整数进行处理。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

338

2023.08.02

int占多少字节
int占多少字节

int占4个字节,意味着一个int变量可以存储范围在-2,147,483,648到2,147,483,647之间的整数值,在某些情况下也可能是2个字节或8个字节,int是一种常用的数据类型,用于表示整数,需要根据具体情况选择合适的数据类型,以确保程序的正确性和性能。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

542

2024.08.29

c++怎么把double转成int
c++怎么把double转成int

本专题整合了 c++ double相关教程,阅读专题下面的文章了解更多详细内容。

53

2025.08.29

C++中int的含义
C++中int的含义

本专题整合了C++中int相关内容,阅读专题下面的文章了解更多详细内容。

197

2025.08.29

Go中Type关键字的用法
Go中Type关键字的用法

Go中Type关键字的用法有定义新的类型别名或者创建新的结构体类型。本专题为大家提供Go相关的文章、下载、课程内容,供大家免费下载体验。

234

2023.09.06

go怎么实现链表
go怎么实现链表

go通过定义一个节点结构体、定义一个链表结构体、定义一些方法来操作链表、实现一个方法来删除链表中的一个节点和实现一个方法来打印链表中的所有节点的方法实现链表。

446

2023.09.25

go语言编程软件有哪些
go语言编程软件有哪些

go语言编程软件有Go编译器、Go开发环境、Go包管理器、Go测试框架、Go文档生成器、Go代码质量工具和Go性能分析工具等。本专题为大家提供go语言相关的文章、下载、课程内容,供大家免费下载体验。

249

2023.10.13

0基础如何学go语言
0基础如何学go语言

0基础学习Go语言需要分阶段进行,从基础知识到实践项目,逐步深入。php中文网给大家带来了go语言相关的教程以及文章,欢迎大家前来学习。

698

2023.10.26

AO3中文版入口地址大全
AO3中文版入口地址大全

本专题整合了AO3中文版入口地址大全,阅读专题下面的的文章了解更多详细内容。

1

2026.01.21

热门下载

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

精品课程

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

共162课时 | 12.6万人学习

C# 教程
C# 教程

共94课时 | 7.1万人学习

C++教程
C++教程

共115课时 | 13万人学习

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

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