Go WaitGroup 和 Cond 实现原理

最新推荐文章于 2025-01-10 13:32:17 发布

程序猿加瓦

最新推荐文章于 2025-01-10 13:32:17 发布

阅读量319

点赞数

分类专栏： Java 文章标签： java 开发语言

本文链接：https://blog.youkuaiyun.com/weixin_62421895/article/details/126482045

版权

本文探讨了Go语言中的WaitGroup和Cond原语。WaitGroup用于等待一组Goroutine的完成，其核心是计数器和信号量管理。而Cond则允许Goroutine在满足特定条件时被阻塞和唤醒，它依赖于锁机制来确保线程安全。文章详细阐述了这两个概念的底层数据结构及使用方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

WaitGroup

概念

Go标准库提供了WaitGroup原语, 可以用它来等待一批 Goroutine 结束

底层数据结构

// A WaitGroup must not be copied after first use.
type WaitGroup struct {
 noCopy noCopy
 state1 [3]uint32
}

其中 noCopy 是 golang 源码中检测禁止拷贝的技术。如果程序中有 WaitGroup 的赋值行为，使用 go vet 检查程序时，就会发现有报错。但需要注意的是，noCopy 不会影响程序正常的编译和运行。

state1主要是存储着状态和信号量，状态维护了 2 个计数器，一个是请求计数器counter ，另外一个是等待计数器waiter（已调用 WaitGroup.Wait 的 goroutine 的个数）

当数组的首地址是处于一个8字节对齐的位置上时，那么就将这个数组的前8个字节作为64位值使用表示状态，后4个字节作为32位值表示信号量(semaphore)；同理如果首地址没有处于8字节对齐的位置上时，那么就将前4个字节作为semaphore，后8个字节作为64位数值。