Once是golang官方提供的一种单例模式的实现。其代码实现非常简单,但是也有不少值得学习的地方。
Go1.19
Once Struct
type Once struct {
done uint32
m Mutex
}
一个uint32变量和一个互斥锁
Do
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()
}
}
首先一个fast path判断done是否被改变了,如果被改变后不为0,则直接结束返回 。如果未改变,就去slow path,加锁去执行f()。
这里有个几个问题需要解释一下:
- LoadUint32能不能换成CAS?
不能。源码里面comments有解释,如果两个goroutine并发请求,CAS没有成功的那一个,就会立刻结束返回,而CAS成功的那一个可能f()还没执行完,未CAS成功的那一个goroutine就不一定能获取拿到f()执行后的结果 - doSlow()当中o.done==0获取done的时候为什么不用LoadUint32 ?是否会有data race导致多次初始化f()?
不会有data race,因为执行到这里的时候,m.Lock()拿到锁后,后面执行代码一定是串行的。所以o.done==0读操作和atomic.StoreUint32(&o.done, 1)写操作是串行的,不会存在datarace。而o.done == 0和Do里面**atomic.LoadUint32(&o.done) == 0 **都是读取操作,所以也不会有data race。