基于 Redis 的分布式信号量 Spring Boot 集成 Redisson 使用 Semaphore 控制并发访问数量

前言

pom

yml

Controller

演示

注意

前言

工作中开发过一个服务，这里记作A服务，主要功能是配置，部署以及调用云函数。其中配置云函数的功能里，有一个配置项是并发数，意思是同一时间最多能有多少个请求调用已部署的云函数。

客户端通过调用A服务，再由A服务去调用云函数，从而实现客户端请求云函数的功能。在调用云函数时，是要根据配置的并发数控制请求的数量。如果时单机架构的话，直接使用JDK中的 Semaphore 就能够实现并发线程控制了。但是项目的架构是微服务架构，所以需要使用分布式信号量才行。

项目中有使用 Redis，所以可以基于 Redis 来实现分布式信号量，使用 Spring Boot 集成 Redisson 使用 RSemaphore 就能够轻松实现分布式信号量了。

pom

        <dependency><groupId>org.redisson</groupId><artifactId>redisson-spring-boot-starter</artifactId><version>3.11.4</version></dependency>

yml

application.yml

spring:redis:redisson:config: "classpath:redisson.yml"server:port: 8888

redisson.yml

{"singleServerConfig":{"idleConnectionTimeout":10000,"pingTimeout":1000,"connectTimeout":10000,"timeout":3000,"retryAttempts":3,"retryInterval":1500,"subscriptionsPerConnection":5,"clientName":null,"address": "redis://localhost:6379","subscriptionConnectionMinimumIdleSize":1,"subscriptionConnectionPoolSize":50,"connectionMinimumIdleSize":32,"connectionPoolSize":64,"database":0},"threads":0,"nettyThreads":0,"codec":{"class":"org.redisson.codec.JsonJacksonCodec"},"transportMode":"NIO"
}

Controller

以下示例代码演示如何使用 Redisson 中的 RSemaphore

初始化了一个 semaphore 的信号量，并设置其 permits 为 1，即最多一个线程能获取许可

接口 /redisson/semaphore 逻辑中，首先会尝试获取许可，拿到许可后才能往下执行业务逻辑，若是超过1秒还没获取到许可，则说明当前存在线程正在处理，就直接返回提示信息”服务器繁忙，请稍后再试“

@RestController
@RequestMapping("/redisson")
public class RedissonController implements InitializingBean {@Resourceprivate RedissonClient client;private RSemaphore semaphore;@Overridepublic void afterPropertiesSet() {semaphore = client.getSemaphore("semaphore");semaphore.trySetPermits(1);}/*** semaphore*/@GetMapping("/semaphore")public String semaphore() throws InterruptedException {if (!semaphore.tryAcquire(1, TimeUnit.SECONDS)) {return "服务器繁忙，请稍后再试";}try {//处理业务逻辑System.out.println("processing");Thread.sleep(1000 * 4);} catch (InterruptedException e) {throw new RuntimeException(e);} finally {semaphore.release();}return "成功";}
}