【Golang】sync.Once的使用

单次执行

• Once的作用是多次调用但只执行一次，Once只有一个方法，Once.Do()，向Do传入一个函 数，这个函数在第一次执行Once.Do()的时候会被调用
• 以后再执行Once.Do()将没有任何动作，即使传入了其他的函数，也不会被执行，如果要执 行其它函数，需要重新创建一个Once对象。
• Once可以安全的再多个协程中并行使用，是协程安全的。

代码语言：javascript复制

// 多次调用仅执行一次指定的函数
f func (o *Once) Do(f func())

代码测试

我们写一段代码来测试一下sync.Once的功能，我们再协程中进行调用观察调用次数，执行后可以发现init只打印了一次

代码语言：javascript复制

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表示已经执行过了，直接结束返回

代码语言：javascript复制

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()
	}
}

安全 测试 函数 协程 源码

0 人点赞