手记

【九月打卡】第17天 go--waitGroup、Once

课程名称:深入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的核心逻辑是

  1. 当前协程获取互斥锁

  2. 执行传入的函数

  3. 运行延迟调用,将成员变量done改为1

  4. 其他协程看到done为1,不再执行


0人推荐
随时随地看视频
慕课网APP