分布式信号量需借助外部存储实现,Redis+Lua通过原子脚本保障计数安全,etcd则利用CAS和租约实现强一致性带过期的信号量。

Go 本身没有内置的分布式信号量,sync.Mutex 和 semaphore.Weighted 都只作用于单机进程内。要跨多台机器协调资源配额(比如限制 10 个服务实例总共只能同时处理 5 个支付请求),必须借助外部存储做原子计数和状态同步。
用 Redis + Lua 实现原子 acquire/release
Redis 的 INCR/DECR 本身不带条件判断,直接用会超卖;必须用 Lua 脚本把“检查剩余数 + 自增/自减”打包成原子操作。这是最常用、最轻量的方案。
- 脚本逻辑:读取当前计数
current,若current 则INCR并返回 1;否则返回 0 - Go 中调用:
redis.Client.Eval(ctx, luaScript, []string{key}, limit),返回结果为int64类型,非零表示获取成功 - 注意设置 key 过期时间(
EXPIRE),避免客户端崩溃后锁永远不释放;建议在acquire成功后立刻执行EXPIRE - 释放时不能只
DECR,要先GET确认 key 存在且值 > 0,再DECR,否则可能把其他客户端的计数减成负数
使用 etcd 实现带租约的分布式信号量
etcd 的 CompareAndSwap (CAS) 和租约(lease)机制天然适合实现带自动过期的信号量,比 Redis 更强调一致性,但延迟略高。
使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器,提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象,支持命名依赖等
- 核心是维护一个 key(如
/semaphore/payment),value 是当前已占用数量(字符串格式) -
acquire:用cmp比较当前值是否 then 执行Put把值 +1;失败则重试或返回错误 - 必须绑定 lease:创建 lease 后,所有
Put操作带上LeaseID,这样客户端宕机后 lease 过期,key 自动删除,避免死锁 - 不要用
Get+Put两步操作 —— 中间可能被其他 client 修改,必须用Txn原子事务
避免常见坑:超时、重入、误释放
分布式环境下,网络分区、GC STW、goroutine panic 都可能导致信号量状态错乱,光靠“拿到就加、释放就减”不够。
- acquire 必须设超时(
context.WithTimeout),否则阻塞在 Redis/Etcd 请求上会拖垮整个服务 - 同一个 goroutine 重复
acquire同一把信号量,会导致计数虚高;应通过map[goroutineID]semKey或上下文标记防重入 - 释放操作不能依赖 defer —— 如果 acquire 失败,defer 仍会执行,造成负计数;应只在 acquire 返回 true 后才注册释放逻辑
- 推荐封装成结构体,把 key、limit、client、leaseID 封装在一起,避免参数散落各处
真正难的不是写 acquire/release 函数,而是定义清楚“谁负责清理”和“超时后如何兜底”。比如一个 acquire 成功但后续业务 panic,又没来得及释放,就得靠 lease 自动回收或后台巡检任务补偿 —— 这部分逻辑往往比信号量本身更重。
golang免费学习笔记(深入):立即使用
在学习笔记中,你将探索golang的核心概念和高级技巧!










