手摸手Go 单例模式与sync.Once-CSDN博客

本文链接：https://blog.csdn.net/sydhappy/article/details/114650578

I leave uncultivated today, was precisely yesterday perishes tomorrow which person of the body implored。

单例模式作为一个较为常见的设计模式，他的定义也很简单，将类的实例化限制为一个单个实例。在Java的世界里，你可能需要从懒汉模式、双重检查锁模式、饿汉模式、静态内部类、枚举等方式中选择一种手动撸一遍代码，但是他们操作起来很容易一不小心就会出现bug。而在Go里，内建提供了保证操作只会被执行一次的sync.Once，操作起来及其简单。

基本使用

在开发过程中需要单例模式的场景比较常见，比如web开发过程中，不可避免的需要跟DB打交道，而DB管理器初始化通常需要保证有且仅发生一次。那么使用sync.Once实现起来就比较简单了。

var manager *DBManager
var once sync.Once
func GetDBManager()*DBManager{
  once.DO(func(){
    manager = &DBManager{}
    manager.initDB(config)
  })
  return manager
}

可以看到仅仅需要once.DO(func(){...})即可, 开发者只需要关注自己的初始化程序即可，单例由sync.Once来保证，极大降低了开发者的心智负担。

sync.Once源码分析

数据结构

sync.Once结构也比较简单，只有一个uint32字段和一个互斥锁Mutex。

// 一旦使用不允许被拷贝
type Once struct {
  // done表示当前的操作是否已经被执行 0表示还没有 1表示已经执行
  // done属性放在结构体的第一位，是因为它在hot path中使用
  // hot path在每个调用点会被内联。
  // 将done放在结构体首位，像amd64/386等架构上可以允许更多的压缩指令
  // 并且在其他架构上更少的指令去计算偏移量
 done uint32
 m    Mutex
}

sync.Once的核心原理，是利用sync.Mutex和atomic包的原子操作来完成。done表示是否成功完成一次执行。存在两个状态：

0 表示当前sync.Once 的第一次DO操作尚未成功
1 表示当前sync.Once 的第一次DO操作已经完成

每次DO方法调用都会去检查done的值，如果为1则啥也不做；如果为0则进入doSlow流程，doSlow很巧妙的先使用sync.Mutex。这样如果并发场景，只有一个goroutine会抢到锁执行下去，其他goroutine则阻塞在锁上，这样的好处是如果拿到锁的那个goroutine失败，其他阻塞在锁上的goroutine就是预备队替补上去。确保sync.Once有且仅成功执行一次的语义。

once flow graph

好了，接下来看源码

操作方法

Do

Do执行函数f当且仅当对应sync.Once实例第一次调用Do。换句话说，给定var once Once,如果once.Do(f)被调用了多次，,尽管f在每次调用的值均不同，但只有第一次调用会执行f。如果需要每个函数都执行，则需要新的sync.Once实例。

// Do的作用主要是针对初始化且有且只能执行一次的场景。因为Do直到f返回才返回，
// 所以如果f内调用Do则会导致死锁
// 如果f执行过程中panic了 那么Do任务f已经执行完毕 未来再次调用不会再执行f
func (o *Once) Do(f func()) {
 if atomic.LoadUint32(&o.done) == 0 {//判断f是否被执行
  // 可能会存在并发 进入slow-path
  o.doSlow(f)
 }
}

注释里提到了一种不正确的Do的实现

if atomic.CompareAndSwapUint32(&o.done, 0, 1) {
  f()
}

这种实现不正确的原因在于，无法保证f()有且仅执行一次的语义。因为使用直接CAS来解决问题，如果同时有多个goroutine竞争执行Do那么是能保证有且仅有一个goroutine会得到执行机会，其他goroutine只能默默离开。

但是如果获得执行机会的goroutine执行失败了，那么以后f()就在也没有执行机会了。

那么我们来看看官方的实现方式

func (o *Once) doSlow(f func()) {
 o.m.Lock()
 defer o.m.Unlock()
 if o.done == 0 {//二次判断f是否已经被执行
  defer atomic.StoreUint32(&o.done, 1)
  f()
 }
}

官方的做法就是如果多个goroutine都来竞争Do，那么先让一个goroutine拿到sync.Mutex的锁，其他的goroutine先不着急让他们直接返回，而是都先阻塞在sync.Mutex上。如果那个拿到锁的goroutine很不幸执行f()失败了，那么defer o.m.Unlock()操作会立刻唤醒阻塞的goroutine接着尝试执行直到成功为止。执行成功后通过defer atomic.StoreUint32(&o.done, 1)来将执行f()的大门给关闭上。