使用 Redisson 实现分布式并发限流

使用 Redisson 实现简单的分布式并发限流

在开发高并发的分布式应用时,控制同时处理的任务数量至关重要。不同于限制每秒请求数(QPS),我们有时需要限制系统的并发任务数,以避免系统资源过度占用。Redisson 提供的 RPermitExpirableSemaphore 是一个理想的解决方案,它可以帮助我们实现分布式并发限流,并且具备自动回收超时许可和程序崩溃后释放资源的能力。

为什么选择 Redisson 的 RPermitExpirableSemaphore?

  1. 分布式控制:支持跨多个服务实例共享同一个信号量,确保所有实例都能公平竞争获取许可。
  2. 自动回收:为每个许可设置过期时间,超时后自动释放,防止死锁。
  3. 资源安全:即使服务崩溃,未释放的许可也能被自动清理,保证系统稳定运行。

快速入门

添加依赖

首先,在你的项目中添加 Redisson 的 Maven 依赖:

xml 复制代码
<!-- Maven -->
<dependency>
    <groupId>org.redisson</groupId>
    <artifactId>redisson</artifactId>
    <version>3.21.1</version>
</dependency>

基本用法

以下是一个简单的例子,展示了如何使用 RPermitExpirableSemaphore 来限制并发任务的数量:

java 复制代码
import org.redisson.Redisson;
import org.redisson.api.RPermitExpirableSemaphore;
import org.redisson.api.RedissonClient;
import org.redisson.config.Config;

public class ConcurrentTaskLimiter {

    private final RedissonClient redissonClient;

    public ConcurrentTaskLimiter(String redisAddress) {
        Config config = new Config();
        config.useSingleServer().setAddress(redisAddress);
        this.redissonClient = Redisson.create(config);
    }

    public void executeWithLimit(Runnable task, int maxConcurrentTasks, long leaseTime, TimeUnit unit) throws InterruptedException {
        RPermitExpirableSemaphore semaphore = redissonClient.getPermitExpirableSemaphore("mySemaphore");
        semaphore.trySetPermits(maxConcurrentTasks); // 设置最大并发任务数
        
        String permitId = semaphore.acquire(leaseTime, unit); // 尝试获取许可
        if (permitId != null) {
            try {
                task.run(); // 执行任务
            } finally {
                semaphore.release(permitId); // 确保许可被释放
            }
        } else {
            System.out.println("Failed to acquire permit within timeout.");
        }
    }

    public static void main(String[] args) throws InterruptedException {
        ConcurrentTaskLimiter limiter = new ConcurrentTaskLimiter("redis://127.0.0.1:6379");
        
        Runnable task = () -> {
            System.out.println("Executing task...");
            try {
                Thread.sleep(5000); // 模拟耗时操作
            } catch (InterruptedException e) {
                Thread.currentThread().interrupt();
            }
        };

        // 允许最多3个并发任务,每个许可的有效期为10秒
        limiter.executeWithLimit(task, 3, 10, TimeUnit.SECONDS);
    }
}

关键点解释

  • 初始化 Redisson 客户端:通过配置连接到 Redis 服务器。
  • 设置最大并发任务数 :调用 trySetPermits 方法设置信号量的最大许可数。
  • 获取许可执行任务 :调用 acquire 方法尝试获取许可,如果成功,则执行任务;完成后必须调用 release 方法释放许可。
相关推荐
黄雪超8 分钟前
Kafka——消费者组消费进度监控都怎么实现?
大数据·分布式·kafka
月亮不月亮25 分钟前
月亮商场购物打折Java
java·eclipse
guozhetao34 分钟前
【ST表、倍增】P7167 [eJOI 2020] Fountain (Day1)
java·c++·python·算法·leetcode·深度优先·图论
技术思考者35 分钟前
基础很薄弱如何规划考研
java·经验分享·考研
●VON1 小时前
重生之我在暑假学习微服务第二天《MybatisPlus-下篇》
java·学习·微服务·架构·mybatis-plus
老华带你飞1 小时前
口腔助手|口腔挂号预约小程序|基于微信小程序的口腔门诊预约系统的设计与实现(源码+数据库+文档)
java·数据库·微信小程序·小程序·论文·毕设·口腔小程序
java叶新东老师1 小时前
RabbitMq 常用命令和REST API
分布式·rabbitmq·ruby
hqxstudying1 小时前
J2EE模式---服务层模式
java·数据库·后端·spring·oracle·java-ee
GM_8281 小时前
【最新最完整】SpringAI-1.0.0开发MCP Server,搭建MCP Client 实战笔记(进阶+详细+完整代码)
java·后端·ai编程·springai·mcp
都叫我大帅哥1 小时前
Java DelayQueue:时间管理大师的终极武器
java