【Golang】sync.Once的使用

MaybeHC

发布于 2024-04-23 19:41:39

810

发布于 2024-04-23 19:41:39

单次执行

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

// 多次调用仅执行一次指定的函数
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()
	}
}

本文参与?腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2024-04-23，如有侵权请联系 cloudcommunity@tencent.com 删除

安全