From dcaeb18870ecdf319dbdb03820405a745d4eeb32 Mon Sep 17 00:00:00 2001
From: 疯狂的狮子li <15040126243@163.com>
Date: 星期五, 17 九月 2021 17:22:45 +0800
Subject: [PATCH] add 增加 集群限流功能
---
ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/RateLimiterAspect.java | 28 ++++++++++++++++++++++------
1 files changed, 22 insertions(+), 6 deletions(-)
diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/RateLimiterAspect.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/RateLimiterAspect.java
index e63e47b..93502c0 100644
--- a/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/RateLimiterAspect.java
+++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/RateLimiterAspect.java
@@ -4,6 +4,7 @@
import com.ruoyi.common.enums.LimitType;
import com.ruoyi.common.exception.ServiceException;
import com.ruoyi.common.utils.RedisUtils;
+import com.ruoyi.common.utils.ServletUtils;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.Signature;
@@ -34,20 +35,19 @@
@Before("rateLimiterPointCut()")
public void doBefore(JoinPoint point) throws Throwable {
RateLimiter rateLimiter = getAnnotationRateLimiter(point);
- String key = rateLimiter.key();
int time = rateLimiter.time();
int count = rateLimiter.count();
-
+ String combineKey = getCombineKey(rateLimiter, point);
try {
RateType rateType = RateType.OVERALL;
- if (rateLimiter.limitType() == LimitType.IP) {
+ if (rateLimiter.limitType() == LimitType.CLUSTER) {
rateType = RateType.PER_CLIENT;
}
- // 杩斿洖 false 璇存槑 鑾峰彇浠ょ墝澶辫触
- if (!RedisUtils.rateLimiter(key, rateType, count, time)) {
+ long number = RedisUtils.rateLimiter(combineKey, rateType, count, time);
+ if (number == -1) {
throw new ServiceException("璁块棶杩囦簬棰戠箒锛岃绋嶅悗鍐嶈瘯");
}
- log.info("闄愬埗璇锋眰'{}',缂撳瓨key'{}'", count, key);
+ log.info("闄愬埗浠ょ墝 => {}, 鍓╀綑浠ょ墝 => {}, 缂撳瓨key => '{}'", count, number, combineKey);
} catch (ServiceException e) {
throw e;
} catch (Exception e) {
@@ -69,4 +69,20 @@
return null;
}
+ public String getCombineKey(RateLimiter rateLimiter, JoinPoint point) {
+ StringBuilder stringBuffer = new StringBuilder(rateLimiter.key());
+ if (rateLimiter.limitType() == LimitType.IP) {
+ // 鑾峰彇璇锋眰ip
+ stringBuffer.append(ServletUtils.getClientIP()).append("-");
+ } else if (rateLimiter.limitType() == LimitType.CLUSTER){
+ // 鑾峰彇瀹㈡埛绔疄渚媔d
+ stringBuffer.append(RedisUtils.getClientId()).append("-");
+ }
+ MethodSignature signature = (MethodSignature) point.getSignature();
+ Method method = signature.getMethod();
+ Class<?> targetClass = method.getDeclaringClass();
+ stringBuffer.append(targetClass.getName()).append("-").append(method.getName());
+ return stringBuffer.toString();
+ }
+
}
--
Gitblit v1.9.3