redis實現閘道器限流(限制API呼叫次數1000次/分)
阿新 • • 發佈:2019-09-27
- 新增maven依賴,使用springboot2.x版本
<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>
- 新增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次,不代表任意時間段的一分鐘呼叫次數低於1