sync.Once
前言
本次的代码是基于go version go1.13.15 darwin/amd64
sync.Once的作用
根据名字就大致能猜到这个函数的作用,就是使用sync.once
的对象只能执行一次。
我们在errgroup
就能看到它的身影
type Group struct {
cancel func()
wg sync.WaitGroup
errOnce sync.Once
err error
}
他保证了,只会记录第一个出错的goroutine
的错误信息
实现原理
// Once is an object that will perform exactly one action.
type Once struct {
// 0未执行,1执行了
done uint32
// 互斥锁
m Mutex
}
里面就一个对外的函数
func (o *Once) Do(f func()) {
// 原子的读取done的值,如果为0代表onec第一次的执行还没有出发
if atomic.LoadUint32(&o.done) == 0 {
// 执行
o.doSlow(f)
}
}
func (o *Once) doSlow(f func()) {
// 加锁
o.m.Lock()
defer o.m.Unlock()
// 判断done变量为0表示还没执行第一次
if o.done == 0 {
// 计数器原子的加一
defer atomic.StoreUint32(&o.done, 1)
// 执行传入的函数
f()
}
}
总结
1、总体上也是很简单一个计数器,一把互斥锁,通过atomic.LoadUint32
的原子读取技术器中的值;
2、如果计数器中的值为0表示还没有执行;
3、加锁,执行传入的函数,然后通过atomic.StoreUint32
原子的对计数器的值进行加一操作;
4、完成。