Go sync.Once介绍

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

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

有疑问加站长微信联系（非本文作者）