共计 1872 个字符，预计需要花费 5 分钟才能阅读完成。
在上一篇文章《Week03: Go 并发编程 (七) 深刻了解 errgroup》当中看 errgourp 源码的时候咱们发现最初返回 err 是通过 once 来只保障返回一个非 nil 的值的，本文就来看一下 Once 的应用与实现
案例
once 的应用很简略
`func main() {`
`var (`
`o sync.Once`
`wg sync.WaitGroup`
`)`
`for i := 0; i < 10; i++ {`
`wg.Add(1)`
`go func(i int) {`
`defer wg.Done()`
`o.Do(func() {`
`fmt.Println("once", i)`
`})`
`}(i)`
`}`
`wg.Wait()`
`}`
输入
`❯ go run ./main.go`
`once 9`
源码剖析
`type Once struct {`
`done uint32`
`m Mutex`
`}`
done 用于断定函数是否执行，如果不为 0 会间接返回
`func (o *Once) Do(f func()) {`
`// Note: Here is an incorrect implementation of Do:`
`//`
`// if atomic.CompareAndSwapUint32(&o.done, 0, 1) {`
`// f()`
`// }`
`//`
`// Do guarantees that when it returns, f has finished.`
`// This implementation would not implement that guarantee:`
`// given two simultaneous calls, the winner of the cas would`
`// call f, and the second would return immediately, without`
`// waiting for the first's call to f to complete.`
`// This is why the slow path falls back to a mutex, and why`
`// the atomic.StoreUint32 must be delayed until after f returns.`
`if atomic.LoadUint32(&o.done) == 0 {`
`// Outlined slow-path to allow inlining of the fast-path.`
`o.doSlow(f)`
`}`
`}`
看 go 的源码真的能够学到很多货色，在这里还给出了很容易犯错的一种实现
`if atomic.CompareAndSwapUint32(&o.done, 0, 1) {`
`f()`
`}`
如果这么实现最大的问题是，如果并发调用，一个 goroutine 执行，另外一个不会等正在执行的这个胜利之后返回，而是间接就返回了，这就不能保障传入的办法肯定会先执行一次了 所以回头看官网的实现
`if atomic.LoadUint32(&o.done) == 0 {`
`// Outlined slow-path to allow inlining of the fast-path.`
`o.doSlow(f)`
`}`
会先判断 done 是否为 0，如果不为 0 阐明还没执行过，就进入 doSlow
`func (o *Once) doSlow(f func()) {`
`o.m.Lock()`
`defer o.m.Unlock()`
`if o.done == 0 {`
`defer atomic.StoreUint32(&o.done, 1)`
`f()`
`}`
`}`
在 doSlow 当中应用了互斥锁来保障只会执行一次
总结
- Once 保障了传入的函数只会执行一次，这罕用在单例模式，配置文件加载，初始化这些场景下
- 然而须要留神。Once 是不能复用的，只有执行过了，再传入其余的办法也不会再执行了
- 并且 Once.Do 在执行的过程中如果 f 呈现 panic，前面也不会再执行了
参考文献
- https://pkg.go.dev/sync#Once
- 6.2 同步原语与锁
mohuishou
lailin.xyz 的博客账号，关注但不限于 Go 开发，云原生，K8s 等
36 篇原创内容
公众号
👆 关注我，_ 一起在常识的陆地漫游 _
转发，点赞，在看就是对我最大的激励
点击“浏览原文”查看参考文献等信息
正文完