java redis 限流_redis实现网关限流(限制API调用次数1000次/分)

本文介绍了如何使用Java和Redis实现网关限流,以限制API在一分钟内的调用次数不超过1000次。文章详细讲解了两种实现方式:方法一是利用Redis的List存储调用时间,当List长度达到1000时检查最后两个时间戳的差值以判断是否超出限流阈值;方法二是通过设置带过期时间的Key记录调用次数,存在一定的误差。这两种方法各有优缺点,可以根据实际需求选择合适的方法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

添加maven依赖,使用springboot2.x版本

org.springframework.boot

spring-boot-starter-data-redis

org.apache.commons

commons-pool2

添加redis配置进application.yml,springboot2.x版本的redis是使用lettuce配置的

spring:

redis:

database: 0

host: localhost

port: 6379

lettuce: # 这里标明使用lettuce配置

pool:

max-active: 8 # 连接池最大连接数

max-wait: -1ms # 连接池最大阻塞等待时间(使用负值表示没有限制

max-idle: 5 # 连接池中的最大空闲连接

min-idle: 0 # 连接池中的最小空闲连接

timeout: 10000ms # 连接超时时间

使用redis作限流器有两种写法

方法一:

Long size = redisTemplate.opsForList().size("apiRequest");

if (size < 1000) {

redisTemplate.opsForList().leftPush("apiRequest", System.currentTimeMillis());

} else {

Long start = (Long) redisTemplate.opsForList().index("apiRequest", -1);

if ((System.currentTimeMillis() - start) < 60000) {

throw new RuntimeException("超过限流阈值");

} else {

redisTemplate.opsForList().leftPush("apiRequest", System.currentTimeMillis());

redisTemplate.opsForList().trim("apiRequest", -1, -1);

}

}

核心思路:用一个list来存放一串值,每次请求都把当前时间放进,如果列表长度为1000,那么调用就是1000次。如果第1000次调用时的当前时间和最初的时间差小于60s,那么就是1分钟里调用超1000次。否则,就清空列表之前的值

方法二:

Integer count = (Integer) redisTemplate.opsForValue().get("apiKey");

Integer integer = Optional.ofNullable(count).orElse(0);

if (integer > 1000) {

throw new RuntimeException("超过限流阈值");

}

if (redisTemplate.getExpire("apiKey", TimeUnit.SECONDS).longValue() < 0) {

redisTemplate.multi();

redisTemplate.opsForValue().increment("apiKey", 1);

redisTemplate.expire("apiKey", 60, TimeUnit.SECONDS);

redisTemplate.exec();

} else {

redisTemplate.opsForValue().increment("apiKey", 1);

}

核心思路:设置key,过期时间为1分钟,其值是api这分钟内调用次数

对比:方法一耗内存,限流准确。方法二结果有部分误差,只限制key存在的这一分钟内调用次数低于1000次,不代表任意时间段的一分钟调用次数低于1000

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值