Browse Source
Merge branch 'refs/heads/milestone-20251104-现金管理二期' into sunjiabei/feature-20251021102635-银行接口
sunjiabei/feature-20251021102635-银行接口
Merge branch 'refs/heads/milestone-20251104-现金管理二期' into sunjiabei/feature-20251021102635-银行接口
sunjiabei/feature-20251021102635-银行接口
16 changed files with 174 additions and 39 deletions
-
6pom.xml
-
39src/main/java/com/example/demo/config/RateLimitUtil.java
-
54src/main/java/com/example/demo/controller/cash/CashRefundController.java
-
1src/main/java/com/example/demo/domain/entity/User.java
-
5src/main/java/com/example/demo/domain/vo/cash/CashCollection.java
-
3src/main/java/com/example/demo/domain/vo/cash/CashRecordDone.java
-
2src/main/java/com/example/demo/domain/vo/coin/GoldUser.java
-
3src/main/java/com/example/demo/domain/vo/coin/RechargeActivity.java
-
2src/main/java/com/example/demo/mapper/cash/CashCollectionMapper.java
-
7src/main/java/com/example/demo/serviceImpl/bean/BeanConsumeServiceImpl.java
-
20src/main/java/com/example/demo/serviceImpl/cash/CashAuditServiceImpl.java
-
5src/main/java/com/example/demo/serviceImpl/cash/CashCollectionServiceImpl.java
-
34src/main/java/com/example/demo/serviceImpl/cash/CashRefundServiceImpl.java
-
3src/main/java/com/example/demo/serviceImpl/cash/MessageServiceImpl.java
-
2src/main/resources/cashMapper/CashCollectionMapper.xml
-
17src/main/resources/cashMapper/CashRefundMapper.xml
@ -0,0 +1,39 @@ |
|||||
|
package com.example.demo.config; |
||||
|
|
||||
|
import com.google.common.cache.Cache; |
||||
|
import com.google.common.cache.CacheBuilder; |
||||
|
import java.util.concurrent.TimeUnit; |
||||
|
|
||||
|
/** |
||||
|
* 本地限流工具类(基于Guava Cache) |
||||
|
*/ |
||||
|
public class RateLimitUtil { |
||||
|
// 缓存:key=限流标识(用户ID/IP),value=占位符(无实际意义) |
||||
|
private static final Cache<String, Object> RATE_LIMIT_CACHE = CacheBuilder.newBuilder() |
||||
|
.expireAfterWrite(3, TimeUnit.SECONDS) // 3秒后自动过期(时间窗口) |
||||
|
.maximumSize(5000) // 最大缓存容量(防止内存溢出,根据用户量调整) |
||||
|
.build(); |
||||
|
|
||||
|
/** |
||||
|
* 判断是否允许请求 |
||||
|
* @param key 限流唯一标识(如用户ID、IP) |
||||
|
* @return true=允许请求,false=限流中 |
||||
|
*/ |
||||
|
public static boolean isAllowed(String key) { |
||||
|
// 1. 尝试从缓存获取key,存在则说明3秒内已请求过(限流) |
||||
|
if (RATE_LIMIT_CACHE.getIfPresent(key) != null) { |
||||
|
return false; |
||||
|
} |
||||
|
// 2. 缓存中不存在,存入缓存(占位符用new Object()即可) |
||||
|
RATE_LIMIT_CACHE.put(key, new Object()); |
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 手动移除限流标识(如接口执行失败时,释放限流) |
||||
|
* @param key 限流唯一标识 |
||||
|
*/ |
||||
|
public static void removeKey(String key) { |
||||
|
RATE_LIMIT_CACHE.invalidate(key); |
||||
|
} |
||||
|
} |
||||
Write
Preview
Loading…
Cancel
Save
Reference in new issue