如何在Golang中针对特定userID使用Mutex避免余额更新竞态

完全可以通过为每个userID创建独立互斥锁的方式解决这个问题，这种粒度的锁既能保证同一用户的余额更新操作串行执行、避免竞态条件，又不会限制不同用户的并发访问，是进程内处理这类场景的常用方案。

具体实现方式 #

可以用Go标准库的sync.Map来存储每个userID对应的sync.Mutex，sync.Map本身就是为并发场景设计的键值存储，适合动态管理不同用户的锁：

import "sync"

type BalanceHandler struct {
  repo      Repository
  provider  Provider
  muMap     sync.Map // key: int64(userID), value: *sync.Mutex
}

func(h *BalanceHandler) AddToBalance(userID int64, amount int64){
  // 获取或创建当前userID对应的互斥锁
  mu, _ := h.muMap.LoadOrStore(userID, &sync.Mutex{})
  mutex := mu.(*sync.Mutex)
  mutex.Lock()
  defer mutex.Unlock() // 确保函数退出时解锁

  // 同一userID的请求会串行执行以下逻辑
  user := h.repo.GetUser(userID)
  bal := h.provider.GetBalance(user.Email())
  newBalance := bal + amount
  h.provider.UpdateBalance(user.Email(), newBalance)
}

注意事项 #

• 内存占用：如果系统用户量极大且长期不清理，sync.Map会积累大量锁对象，若有需要可以定期清理长时间未活跃用户的锁（比如结合过期时间机制），不过大多数业务场景下无需额外处理。
• 分布式场景限制：如果你的服务是分布式部署（多实例运行），进程内的互斥锁就失效了，这时候需要改用分布式锁（比如基于Redis、ZooKeeper）或者直接依赖数据库的行级锁来保证跨实例的并发安全。
• 锁的粒度：用userID作为锁的粒度是合理的，避免了全局锁导致的并发性能下降，平衡了安全性和效率。

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