Golang中如何为ServiceDM缓存查询方法实现线程安全预检查以规避竞态条件
本地缓存类方法的并发安全改造方案
你原来的代码除了并发问题,还有个基础笔误:赋值和返回值的地方误用了类型名ServiceDM,而非当前接收者实例dm,实际编译就会报错。抛开笔误不说,这个带缓存预检查的方法在并发场景下实际有两个致命风险:
- 无锁读写指针字段,直接触发Go内存模型定义的data race,极端场景下会读到未完成赋值的脏指针
- 所有并发协程都会同时通过缓存非空的预检查,重复调用下游服务,轻则造成下游服务瞬时压力飙升,重则把服务打挂
方案1:读写锁+双检查模式(通用适配,无特殊业务约束)
这个是适配场景最广的实现,不管你要不要做缓存失效、拉取失败重试,都能直接用,核心逻辑是:
- 缓存读取阶段用读锁,不阻塞其他协程读缓存
- 缓存未命中要写值的时候,升级为写锁,并且拿到锁之后再二次检查一次缓存值,防止锁等待期间已经有协程完成了缓存写入,重复执行远端拉取逻辑
对应代码示例:
type ServiceDM struct { mu sync.RWMutex ServiceData *XXType } func (dm *ServiceDM) A() (*XXType, error) { // 第一步:快速读缓存 dm.mu.RLock() cached := dm.ServiceData dm.mu.RUnlock() if cached != nil { return cached, nil } // 缓存未命中,加写锁 dm.mu.Lock() defer dm.mu.Unlock() // 双检查:拿到锁之后再确认一次,避免重复拉取 if dm.ServiceData != nil { return dm.ServiceData, nil } // 原有业务逻辑:调用下游、组装数据 resp, err := fetchRemoteData() if err != nil { return nil, err } dm.ServiceData = resp return dm.ServiceData, nil }
这个方案的实现成本极低,没有任何第三方依赖,只要逻辑写对,完全可以从根上避免竞态和缓存击穿问题,是生产环境最常用的实现方式。
方案2:sync.Once 实现(仅适配初始化只跑一次的场景)
如果你的业务逻辑里,ServiceData一旦拉取成功就永久有效,不需要主动清空缓存、也不需要拉取失败后重试,那可以直接用标准库的sync.Once替换手写的锁逻辑,代码更简洁:
type ServiceDM struct { once sync.Once ServiceData *XXType initErr error } func (dm *ServiceDM) A() (*XXType, error) { dm.once.Do(func() { resp, err := fetchRemoteData() if err != nil { dm.initErr = err return } dm.ServiceData = resp }) return dm.ServiceData, dm.initErr }
注意这个方案的边界:
sync.Once的执行逻辑不管成功失败都只会跑一次,如果你的业务需要在拉取失败后下次调用自动重试,或者要定时失效缓存重新拉取,这个方案就不适用了。
常见踩坑提醒
不要图效率只用原子操作(比如sync/atomic.Pointer)替换原来的字段读写:这种方式只能解决指针读写的内存可见性问题,避免data race,但完全没法解决缓存未命中时,多个协程同时读到底层值为nil、重复打下游的击穿问题,没法覆盖所有并发风险。
内容的提问来源于stack exchange,提问作者Frank
相关产品推荐
相关产品推荐

