实战指南:手把手带你写一个用在分布式数据库连接池(Connection Pool)中的原生 Semaphore 控制器

秋萱同学_1263

秋萱同学_1263

2026-06-02

1006人浏览

原创

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

实战指南:手把手带你写一个用在分布式数据库连接池(connection pool)中的原生 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(原子扣减 + 设置过期)

AIHub
AIHub

一款AI开发辅助工具,主要用于一站式人工智能工具集合网站,收录了国内外1500+AI工具和网站,每日持续更新,适合需要提升相关任务效率的用户。

下载
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 耗时,绘制热力图;超过阈值(如剩余

不复杂但容易忽略。

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

redis分布式锁

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
java
java

Java是一个通用术语,用于表示Java软件及其组件,包括“Java运行时环境 (JRE)”、“Java虚拟机 (JVM)”以及“插件”。php中文网还为大家带了Java相关下载资源、相关课程以及相关文章等内容,供大家免费下载使用。

2023.06.15

11557

6

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

2023.07.05

8662

9

java自学难吗
java自学难吗

Java自学并不难。Java语言相对于其他一些编程语言而言,有着较为简洁和易读的语法,本专题为大家提供java自学难吗相关的文章,大家可以免费体验。

2023.07.31

7452

8

java配置jdk环境变量
java配置jdk环境变量

Java是一种广泛使用的高级编程语言,用于开发各种类型的应用程序。为了能够在计算机上正确运行和编译Java代码,需要正确配置Java Development Kit(JDK)环境变量。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.01

1224

3

java保留两位小数
java保留两位小数

Java是一种广泛应用于编程领域的高级编程语言。在Java中,保留两位小数是指在进行数值计算或输出时,限制小数部分只有两位有效数字,并将多余的位数进行四舍五入或截取。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.02

988

3

java基本数据类型
java基本数据类型

java基本数据类型有:1、byte;2、short;3、int;4、long;5、float;6、double;7、char;8、boolean。本专题为大家提供java基本数据类型的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

1476

5

java有什么用
java有什么用

java可以开发应用程序、移动应用、Web应用、企业级应用、嵌入式系统等方面。本专题为大家提供java有什么用的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

3029

5

java在线网站
java在线网站

Java在线网站是指提供Java编程学习、实践和交流平台的网络服务。近年来,随着Java语言在软件开发领域的广泛应用,越来越多的人对Java编程感兴趣,并希望能够通过在线网站来学习和提高自己的Java编程技能。php中文网给大家带来了相关的视频、教程以及文章,欢迎大家前来学习阅读和下载。

2023.08.03

20291

3

配置java环境变量
配置java环境变量

配置Java环境变量是为了让操作系统能够识别和使用Java的相关命令和功能。本专题为大家提供配置java环境变量相关文章,帮助大家解决问题。

2023.08.03

1335

8

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.7万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 134.1万人学习