// 多次调用仅执行一次指定的函数
f func (o *Once) Do(f func())
我们写一段代码来测试一下sync.Once的功能,我们再协程中进行调用观察调用次数,执行后可以发现init只打印了一次
func Test(){
fmt.Println("init")
}
func main() {
var once sync.Once
for i:=0 ;i<10;i++{
//多次调用执行一次
go once.Do(Test)
//Test()
}
time.Sleep(time.Second*2)
}
我们可以看看Once的源码
once的源码逻辑也很简单,done的值,初始值为0表示还未执行过,1表示已经执行过。在调用中也添加了锁避免出现并发问题。 当 done==1表示已经执行过了,直接结束返回
package sync
import (
"sync/atomic"
)
type Once struct {
done uint32
m Mutex
}
func (o *Once) Do(f func()) {
if atomic.LoadUint32(&o.done) == 0 {
// Outlined slow-path to allow inlining of the fast-path.
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()
}
}