关于golang:Go并发编程八-深入理解-syncOnce

43次阅读

共计 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,前面也不会再执行了

参考文献

  1. https://pkg.go.dev/sync#Once
  2. 6.2 同步原语与锁

mohuishou

lailin.xyz 的博客账号,关注但不限于 Go 开发,云原生,K8s 等

36 篇原创内容

公众号

👆 关注我,_ 一起在常识的陆地漫游 _

转发,点赞,在看就是对我最大的激励

点击“浏览原文”查看参考文献等信息

正文完
 0