itbaima-study/SpringBoot-Vue-Template-Jwt
1
0
Fork 0
mirror of https://github.com/itbaima-study/SpringBoot-Vue-Template-Jwt.git synced 2025-05-18 05:21:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复Redis限流中increment概率性重新创建一个无过期时间的新键值对（刚好过期）此时会导致用户永久被封禁（永远返回403或是401）

Browse Source
This commit is contained in:
柏码の讲师 2024-02-22 23:24:16 +08:00
parent 9eff280059
commit 8636be9a48
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
my-project-backend/src/main/java/com/example/utils/FlowUtils.java
View File

@ -72,8 +72,12 @@ public class FlowUtils {
* @return 是否通过限流检查 * @return 是否通过限流检查
*/ */
private boolean internalCheck(String key, int frequency, int period, LimitAction action){ private boolean internalCheck(String key, int frequency, int period, LimitAction action){
if (Boolean.TRUE.equals(template.hasKey(key))) { String count = template.opsForValue().get(key);
Long value = Optional.ofNullable(template.opsForValue().increment(key)).orElse(0L); if (count != null) {
long value = Optional.ofNullable(template.opsForValue().increment(key)).orElse(0L);
int c = Integer.parseInt(count);
if(value != c + 1)
template.expire(key, period, TimeUnit.SECONDS);
return action.run(value > frequency); return action.run(value > frequency);
} else { } else {
template.opsForValue().set(key, "1", period, TimeUnit.SECONDS); template.opsForValue().set(key, "1", period, TimeUnit.SECONDS);