Golang Once源码解析

Posted 2020-10-31

sync.Once可以实现单例模式，确保sync.Once.Do(f func())只会被执行一次，可以初始化某个实例单例。

针对Golang 1.9的sync.Once，与Golang 1.10一样。 源代码位置：sync\once.go。

结构体

Once结构体定义如下：

type Once struct {
    m    Mutex
    done uint32   // 初始值为0表示还未执行过，1表示已经执行过
}

Do

func (o *Once) Do(f func()) {
    // done==1表示已经执行过了，直接结束返回
    if atomic.LoadUint32(&o.done) == 1 {
        return
    }
    // 锁住对象，避免并发问题
    o.m.Lock()
    defer o.m.Unlock()
    if o.done == 0 {
        // 先将done设置为1，再执行f函数
        defer atomic.StoreUint32(&o.done, 1)
        f()
    }
}

需要注意的是执行f函数是同步进行的，也就是说可能存在阻塞问题。