go
中sync.Once
也是一个工作中会用到的一个类型,它是开箱即用的
var once sync.Once
once.Do(func(){
fmt.Println("do once")
})
只要声明改类型的once
变量,就可以直接使用它的Do
方法,Do
方法的参数是一个无参数,无返回的函数。
它有一个特性,就是不管调用Do
方法多少次,里面的函数只会执行一次。
我们可以看下它的实现源码,非常精简
// Once is an object that will perform exactly one action.
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()
}
}
我们可以看到源码中Do
函数里面有一个类似于双重检查的操作,然后在doSlow
函数里面才真正的加了锁,最后结束才修改了done
的值
这是因为在设计sync.Once
的时候,要实现函数返回时,Do
里面的函数已经执行完毕,所以才会有这样的机制。
如果开始有两个并发调用,那么这种机制可以保证第二个调用不会立即返回,会在获取锁的时候阻塞,等第一个调用执行完毕之后,第二个调用进行二次校验之后就直接返回了。