前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >redis实现网关限流(限制API调用次数1000次/分)

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

作者头像
java思维导图
发布2020-06-28 16:48:21
1.4K0
发布2020-06-28 16:48:21
举报
文章被收录于专栏:java思维导图java思维导图

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

代码语言:javascript
复制
<dependency>
 <groupId>org.springframework.boot</groupId>
 <artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>
<dependency>
 <groupId>org.apache.commons</groupId>
 <artifactId>commons-pool2</artifactId>
</dependency>

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

代码语言:javascript
复制
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          # 连接超时时间

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

方法一:
代码语言:javascript
复制
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次。否则,就清空列表之前的值

方法二:
代码语言:javascript
复制
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

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2020-06-23,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 java思维导图 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 方法一:
  • 方法二:
相关产品与服务
云数据库 Redis
腾讯云数据库 Redis(TencentDB for Redis)是腾讯云打造的兼容 Redis 协议的缓存和存储服务。丰富的数据结构能帮助您完成不同类型的业务场景开发。支持主从热备,提供自动容灾切换、数据备份、故障迁移、实例监控、在线扩容、数据回档等全套的数据库服务。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档