golang 系列:sync.Once 讲解

发布时间 2023-06-09 22:14:13作者: 技术颜良

sync.Once 介绍

之前提到过 Go 的并发辅助对象:WaitGroup。同样的, sync.Once 也是 Go 官方的一并发辅助对象,它能够让函数方法只执行一次,达到类似 init 函数的效果。我们来看看它的简单用法:

func main() {
 var once sync.Once
 onceFunc := func() {
  fmt.Println("Only once")
 }

 for i := 0; i < 10; i++ {
  once.Do(onceFunc)
 }
}

这里执行后我们将只看到一次 Only once 的打印信息,这就是 sync.Once 的一次性效果。

sync.Once 源码

我们来看下 sync.Once 的源码:

type Once struct {
 done uint32
 m    Mutex
}

func (o *Once) Do(f func()) {
    // 原子加载标识值,判断是否已被执行过
 if atomic.LoadUint32(&o.done) == 0 {
  o.doSlow(f)
 }
}

func (o *Once) doSlow(f func()) { // 还没执行过函数
 o.m.Lock()
 defer o.m.Unlock()
 if o.done == 0 { // 再次判断下是否已被执行过函数
  defer atomic.StoreUint32(&o.done, 1) // 原子操作:修改标识值
  f() // 执行函数
 }
}

从上面可以分析出,sync.Once 是通过对一个标识值,原子性的修改和加载,来减少锁竞争的。