Go语言中是否存在类似Java ConcurrentMap.computeIfAbsent的并发Map函数？

我当初在找Go里类似Java ConcurrentMap.computeIfAbsent的功能时，也和你一样，翻遍了标准库还有不少缓存库，最后发现sync.Map最接近需求，但很快就碰到了那个头疼的问题——它的LoadOrStore方法没办法做延迟计算！

你说得没错，func (m *Map) LoadOrStore(key, value interface{}) (actual interface{}, loaded bool)这个方法要求你先把要存储的value创建好再传进去，不管对应的key是不是已经存在。如果创建value的成本很高（比如要发起数据库查询、做复杂的计算或者IO操作），那这种方式就太浪费资源了——明明key已经存在，却白做了一堆无用功，这和Java里只在key不存在时才执行计算逻辑的computeIfAbsent完全不一样。

那怎么解决这个问题呢？其实我们可以基于sync.Map封装一个自己的延迟计算版本，核心思路是双重检查加锁，确保只有当key真正不存在的时候，才会执行计算逻辑。给你个示例实现：

import "sync"

type LazySyncMap struct {
    inner sync.Map
    mutex sync.Mutex
}

// LoadOrCompute 类似Java的computeIfAbsent，仅当key不存在时执行compute函数生成值
func (m *LazySyncMap) LoadOrCompute(key interface{}, compute func() interface{}) (interface{}, bool) {
    // 第一步：先无锁尝试加载，快速返回已存在的值
    if val, exists := m.inner.Load(key); exists {
        return val, true
    }

    // 第二步：加锁，避免多个goroutine同时执行compute
    m.mutex.Lock()
    defer m.mutex.Unlock()

    // 第三步：再次检查，防止加锁期间已有其他goroutine写入了该key
    if val, exists := m.inner.Load(key); exists {
        return val, true
    }

    // 第四步：执行计算并存储
    computedVal := compute()
    m.inner.Store(key, computedVal)
    return computedVal, false
}

这个实现的好处是：

• 大部分情况下（key已存在）不需要加锁，保持了sync.Map的高性能
• 只有当key确实不存在时，才会执行compute函数，避免了不必要的资源消耗
• 加锁后的双重检查能防止多个goroutine重复计算同一个key的值

当然，如果你对性能有更高的要求，还可以考虑用更细粒度的锁（比如分片锁），但上面这个版本对于大多数场景来说已经足够好用了。

内容的提问来源于stack exchange，提问作者hussachai