Go 并发实战 -- sync Once
Go 并发实战 -- sync Once
邹志全
发布于 2019-07-31 10:59:55
发布于 2019-07-31 10:59:55
前言
Once是一个非常实用的API，它保证了一个事情仅做一次，这个在许多场景非常有用，所以Once也是go提供的为数不多的API之一。
语法基础
Once的语法非常简单，整个API就提供了一个Do函数，Do函数接受的是一个函数对象，通过once.Do可以保证这个事情仅做一次，来看一个demo：
func main() {
var once = sync.Once{}
once.Do(say)
once.Do(say)
once.Do(say)
once.Do(say)
}
func say() {
fmt.Println("say once")
}输出：
image.png
源码实现
// Once is an object that will perform exactly one action.
type Once struct {
m Mutex // 一把锁
done uint32 // 是否已经做过的标记
}
func (o *Once) Do(f func()) {
// 如果已经做过了直接返回，这里使用atmoic保证并发场景下的安全性
if atomic.LoadUint32(&o.done) == 1 {
return
}
// 这里有一个先判断后操作的竞态条件，所以锁了一下
o.m.Lock()
defer o.m.Unlock()
if o.done == 0 {
// 将状态标记为已完成，这个过程还是在锁内部
defer atomic.StoreUint32(&o.done, 1)
f()
}
}ps：源码中有一个细节点，如果声明了多个defer，按照定义顺序，倒序执行，这也就是为什么最终把状态置为已完成是发生在锁内部了。 关于Once的内容挺简单的，暂时先介绍这么多。
本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2019.07.10 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录