Go 标准库源码分析 - sync 的 WaitGroup

2023-12-31 11:48

本文主要是介绍Go 标准库源码分析 - sync 的 WaitGroup,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

WaitGroup常用于多个goroutine协作,主要功能是阻塞等待一组goroutine完成。 

一、数据结构

type WaitGroup struct {noCopy noCopystate1 [3]uint32    // 用于存放任务计数器、等待者计数器和信号量 
}

WaitGroup采用64位的值来保存计数器,其中高32位为任务计数器,低32位为等待者计数器,另外用32位的值保存信号量。

WaitGroup在使用时需要64位的计数器进行原子操作,这要求计数器的地址是64位对齐的。如果是64位的操作系统,可直接取数组前两位元素作为计数器;如果是32位操作系统,state1可能不是64位对齐的地址,则将数组第一个元素作为信号量而后两个元素为计数器。

func (wg *WaitGroup) state() (statep *uint64, semap *uint32) {if uintptr(unsafe.Pointer(&wg.state1))%8 == 0 {return (*uint64)(unsafe.Pointer(&wg.state1)), &wg.state1[2]} else {return (*uint64)(unsafe.Pointer(&wg.state1[1])), &wg.state1[0]}
}

 

二、使用方法

1. Add

func (wg *WaitGroup) Add(delta int) {// 获取计数器和信号量的地址statep, semap := wg.state()// 增加任务数,为原子操作state := atomic.AddUint64(statep, uint64(delta)<<32)// 任务计数器v := int32(state >> 32)// 等待者计数器w := uint32(state)// 任务数不能为负if v < 0 {panic("sync: negative WaitGroup counter")}// 在Add前执行Waitif w != 0 && delta > 0 && v == int32(delta) {panic("sync: WaitGroup misuse: Add called concurrently with Wait")}if v > 0 || w == 0 {return}// 执行至此,说明任务数为0,且等待者数不为0// 原计数器与现在的不同,可能是Add和Wait同时执行if *statep != state {panic("sync: WaitGroup misuse: Add called concurrently with Wait")}// 将计数器归零*statep = 0// 唤醒等待者for ; w != 0; w-- {runtime_Semrelease(semap, false, 0)}
}

2. Done

func (wg *WaitGroup) Done() {// 执行Add方法将任务数减1wg.Add(-1)
}

3. Wait

func (wg *WaitGroup) Wait() {// 获取计数器和信号量地址statep, semap := wg.state()for {// 原子读计数器的值state := atomic.LoadUint64(statep)// 任务计数器v := int32(state >> 32)// 等待计数器w := uint32(state)if v == 0 {// 任务数为0,直接返回return}// 任务数不为0时,原子增加等待者数if atomic.CompareAndSwapUint64(statep, state, state+1) {// 挂起当前goroutineruntime_Semacquire(semap)// 在Add方法里,唤醒等待者前会将计数器归零,因此若此处的statep不为0,则说明又调用了Add或Wait方法,导致panicif *statep != 0 {panic("sync: WaitGroup is reused before previous Wait has returned")}return}}
}

 

这篇关于Go 标准库源码分析 - sync 的 WaitGroup的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/555947

相关文章

GO语言短变量声明的实现示例

《GO语言短变量声明的实现示例》在Go语言中,短变量声明是一种简洁的变量声明方式,使用:=运算符,可以自动推断变量类型,下面就来具体介绍一下如何使用,感兴趣的可以了解一下... 目录基本语法功能特点与var的区别适用场景注意事项基本语法variableName := value功能特点1、自动类型推

GO语言中函数命名返回值的使用

《GO语言中函数命名返回值的使用》在Go语言中,函数可以为其返回值指定名称,这被称为命名返回值或命名返回参数,这种特性可以使代码更清晰,特别是在返回多个值时,感兴趣的可以了解一下... 目录基本语法函数命名返回特点代码示例命名特点基本语法func functionName(parameters) (nam

Go之errors.New和fmt.Errorf 的区别小结

《Go之errors.New和fmt.Errorf的区别小结》本文主要介绍了Go之errors.New和fmt.Errorf的区别,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考... 目录error的基本用法1. 获取错误信息2. 在条件判断中使用基本区别1.函数签名2.使用场景详细对

Go语言连接MySQL数据库执行基本的增删改查

《Go语言连接MySQL数据库执行基本的增删改查》在后端开发中,MySQL是最常用的关系型数据库之一,本文主要为大家详细介绍了如何使用Go连接MySQL数据库并执行基本的增删改查吧... 目录Go语言连接mysql数据库准备工作安装 MySQL 驱动代码实现运行结果注意事项Go语言执行基本的增删改查准备工作

Go中select多路复用的实现示例

《Go中select多路复用的实现示例》Go的select用于多通道通信,实现多路复用,支持随机选择、超时控制及非阻塞操作,建议合理使用以避免协程泄漏和死循环,感兴趣的可以了解一下... 目录一、什么是select基本语法:二、select 使用示例示例1:监听多个通道输入三、select的特性四、使用se

Go语言使用Gin处理路由参数和查询参数

《Go语言使用Gin处理路由参数和查询参数》在WebAPI开发中,处理路由参数(PathParameter)和查询参数(QueryParameter)是非常常见的需求,下面我们就来看看Go语言... 目录一、路由参数 vs 查询参数二、Gin 获取路由参数和查询参数三、示例代码四、运行与测试1. 测试编程路

Go语言使用net/http构建一个RESTful API的示例代码

《Go语言使用net/http构建一个RESTfulAPI的示例代码》Go的标准库net/http提供了构建Web服务所需的强大功能,虽然众多第三方框架(如Gin、Echo)已经封装了很多功能,但... 目录引言一、什么是 RESTful API?二、实战目标:用户信息管理 API三、代码实现1. 用户数据

Android 缓存日志Logcat导出与分析最佳实践

《Android缓存日志Logcat导出与分析最佳实践》本文全面介绍AndroidLogcat缓存日志的导出与分析方法,涵盖按进程、缓冲区类型及日志级别过滤,自动化工具使用,常见问题解决方案和最佳实... 目录android 缓存日志(Logcat)导出与分析全攻略为什么要导出缓存日志?按需过滤导出1. 按

Go语言网络故障诊断与调试技巧

《Go语言网络故障诊断与调试技巧》在分布式系统和微服务架构的浪潮中,网络编程成为系统性能和可靠性的核心支柱,从高并发的API服务到实时通信应用,网络的稳定性直接影响用户体验,本文面向熟悉Go基本语法和... 目录1. 引言2. Go 语言网络编程的优势与特色2.1 简洁高效的标准库2.2 强大的并发模型2.

深入理解go中interface机制

《深入理解go中interface机制》本文主要介绍了深入理解go中interface机制,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学... 目录前言interface使用类型判断总结前言go的interface是一组method的集合,不