Go sync.Once介绍

1,519 阅读1分钟

gosync.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里面的函数已经执行完毕,所以才会有这样的机制。

如果开始有两个并发调用,那么这种机制可以保证第二个调用不会立即返回,会在获取锁的时候阻塞,等第一个调用执行完毕之后,第二个调用进行二次校验之后就直接返回了。