Golang 并发原语 Sync Once

HyggeBest2025-08-26 15:27

Once 保证操作只执行一次

适用于一些单例、配置的加载等

内部数据结构

go 复制代码

type Once struct {
    done atomic.Uint32
    m    Mutex
}

done 用于标识操作是否已经完成只有0或1
m 互斥锁

核心方法

Do

scss 复制代码

func (o *Once) Do(f func()) {
    if o.done.Load() == 0 {
       o.doSlow(f)
    }
}
func (o *Once) doSlow(f func()) {
    o.m.Lock()
    defer o.m.Unlock()
    if o.done.Load() == 0 {
       defer o.done.Store(1)
       f()
    }
}

原子加载done值是否为0
如果等于0说明f func()并没有被初始化
进入doSlow方法
加锁，然后再次原子加载done值是否为0。这里进行可一个double check,确保当前done的值没有被修改
执行f func(),然后将done标记为1

once这个并发原底层代码非常简洁，也很好理解。