分布式数据库连接池不能直接用JDK的Semaphore,因其本地线程安全特性无法跨进程协同计数,导致全局连接超限;需基于Redis/etcd实现带租约、原子性、幂等性的分布式Semaphore控制器。

为什么分布式数据库连接池不能直接用 JDK 的 Semaphore
本地线程安全的 Semaphore 在单机场景下表现良好,但放到分布式数据库连接池中会立刻失效——多个应用实例各自维护一套计数器,根本无法协同控制全局连接数。比如 A 实例 acquire 了 5 个许可,B 实例也 acquire 了 5 个,而实际数据库只允许 8 个并发连接,结果就是连接被打满、拒绝服务甚至雪崩。
核心目标:让 Semaphore “跨进程可见”
真正的分布式 Semaphore 控制器,本质是把“许可计数 + 原子操作”从内存搬到共享存储,并保证高可用与低延迟。关键不在于重写算法,而在于选对协调底座和封装边界:
- 底层必须是强一致、低延迟的分布式键值存储:推荐 Redis(配合 Lua 脚本保证原子性)或 etcd(利用 Compare-and-Swap + TTL 自动续期);ZooKeeper 因性能和运维成本已不推荐用于高频许可调度。
- 许可不是静态配额,而是带租约(lease)的动态资源:每次 acquire 必须附带超时时间(如 30 秒),避免客户端崩溃后许可永久泄露;release 需幂等,支持自动过期回收。
-
不暴露底层存储细节给上层连接池:控制器对外 API 应完全兼容
Semaphore接口(acquire()/tryAcquire()/release()),内部异步刷新本地缓存 + 后台保活,降低调用开销。
手写一个基于 Redis 的分布式 Semaphore 控制器(精简可运行版)
以下为生产可用的核心逻辑(使用 Jedis + Lua 脚本),不含重试、熔断、指标埋点等增强功能,但已覆盖一致性、租约、幂等三大要点:
// Lua 脚本:acquire.lua(原子扣减 + 设置过期)
local key = KEYS[1]
local permits = tonumber(ARGV[1])
local lease = tonumber(ARGV[2])
local current = tonumber(redis.call('GET', key) or '0')
if current >= permits then
redis.call('INCRBY', key, -permits)
redis.call('EXPIRE', key, lease)
return 1
else
return 0
end
// Java 控制器片段(关键逻辑)
public class DistributedSemaphore {
private final JedisPool jedisPool;
private final String key;
private final int maxPermits;
private final int leaseSeconds;
public boolean tryAcquire(int permits) {
try (Jedis jedis = jedisPool.getResource()) {
Object result = jedis.eval(
ScriptLoader.load("acquire.lua"),
Collections.singletonList(key),
Arrays.asList(String.valueOf(permits), String.valueOf(leaseSeconds))
);
return Long.parseLong(result.toString()) == 1;
}
}
public void release(int permits) {
try (Jedis jedis = jedisPool.getResource()) {
// 使用 INCRBY 安全释放,无需检查 key 是否存在
jedis.incrBy(key, permits);
// 不重置过期时间:由下次 acquire 或后台心跳维持
}
}
}
注意:maxPermits 需预先通过 SETNX 初始化(如 SETNX pool:db1:sem 100),且所有实例共用同一 key;leaseSeconds 应略大于连接最大执行时间,防止误回收。
集成到连接池前必须做的三件事
- 兜底降级开关:当 Redis 不可用时,自动切到本地限流(如 Guava RateLimiter),并告警,而不是直接阻塞或抛异常。
- 连接生命周期绑定许可:在连接 create 时 acquire,在 close 时 release;若连接因网络中断未正常 close,依赖租约自动回收,避免死锁。
-
监控许可水位与等待队列:采集
GET pool:db1:sem值 + 拦截 acquire 耗时,绘制热力图;超过阈值(如剩余
不复杂但容易忽略。











