sync-once-source-code-analysis

golang once rwmutex 源码分析

Posted by Mickey on November 4, 2018

sync.Once源码分析

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函数是同步进行的,也就是说可能存在阻塞问题。