2

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

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


iuoui
120 声望10 粉丝