课程名称:深入Go底层原理,重写Redis中间件实战
课程章节:6-9
课程讲师:Moody
课程内容:
※ sync.WaitGroup
可以等待一组协程执行完并释放,一个比较常见的场景是批量发出RPC或者HTTP请求
type waitGroup struct{
noCopy noCopy
state1 [3]unit32
}
noCopy是一种特殊的结构,表示该结构体不会被copy和再赋值,避免出现问题。底层是sync.noCopy 结构体,编译期间会进行检查。
state1 存储着状态和信号量。32位机器是 waiter counter sema。64位机器是 sema waiter counter。
sync.WaitGroup 暴露了三个接口,分别是 Add,Wait,Done。
Add方法主要是更sate1里面的counter,增加一个等待协程就+1,每次一个协程完成任务就调用Done,Done底层也是Add(-1)。
Wait会阻塞调用这个方法的协程,当counter>0的时候,调用runtime.sync_runtime_Semacquire让当前协程休眠,当counter减少到0的时候,就会调用,sync.runtime_Semrelease 唤醒那些等待的协程。
※ Once
Once结构体可以保证调用某段代码只执行一次。每一个sync.Once结构体对外唯一暴露的方法是Do,该方法会接收一个入参为空的函数。
如果传入的函数在前面的Do已经执行过,会直接返回。
如果没有执行过,会调用 sync.Once.doSlow 执行传入的函数
doSlow的核心逻辑是
当前协程获取互斥锁
执行传入的函数
运行延迟调用,将成员变量done改为1
其他协程看到done为1,不再执行