opensource/RuoYi-Vue-Plus
2
0
Fork 1
mirror of https://gitee.com/dromara/RuoYi-Vue-Plus.git synced 2026-09-21 02:25:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

抽取通用变量 以及格式修改

Browse Source
This commit is contained in:
wdtt 2023-02-26 20:08:58 +08:00
parent 560cb1d95a
commit 94ca60250a
2 changed files with 24 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ruoyi-common/src/main/java/com/ruoyi/common/annotation/RateLimiter.java
View File

@ -16,7 +16,7 @@ import java.lang.annotation.*;
public @interface RateLimiter {
/**
* 限流key,支持使用Spring el表达式来动态获取方法上的参数值
* 格式类似于 #{#code} 请注意格式！！
* 格式类似于 #code.id #{#code}
*/
String key() default CacheConstants.RATE_LIMIT_KEY;

42
ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/RateLimiterAspect.java
View File

@ -14,6 +14,10 @@ import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.reflect.MethodSignature;
import org.redisson.api.RateType;
import org.springframework.core.DefaultParameterNameDiscoverer;
import org.springframework.core.ParameterNameDiscoverer;
import org.springframework.expression.EvaluationContext;
import org.springframework.expression.ExpressionParser;
import org.springframework.expression.ParserContext;
import org.springframework.expression.common.TemplateParserContext;
import org.springframework.expression.spel.standard.SpelExpressionParser;
import org.springframework.expression.spel.support.StandardEvaluationContext;
@ -31,6 +35,16 @@ import java.lang.reflect.Method;
@Component
public class RateLimiterAspect {
//定义spel表达式解析器
private final ExpressionParser parser = new SpelExpressionParser();
//定义spel解析模版
private final ParserContext parserContext = new TemplateParserContext();
//定义spel上下文对象进行解析
private final EvaluationContext context = new StandardEvaluationContext();
//方法参数解析器
private final ParameterNameDiscoverer pnd = new DefaultParameterNameDiscoverer();
@Before("@annotation(rateLimiter)")
public void doBefore(JoinPoint point, RateLimiter rateLimiter) throws Throwable {
int time = rateLimiter.time();
@ -51,9 +65,9 @@ public class RateLimiterAspect {
}
log.info("限制令牌 => {}, 剩余令牌 => {}, 缓存key => '{}'", count, number, combineKey);
} catch (Exception e) {
if(e instanceof ServiceException){
if (e instanceof ServiceException) {
throw e;
}else{
} else {
throw new RuntimeException("服务器限流异常，请稍候再试");
}
}
@ -65,32 +79,23 @@ public class RateLimiterAspect {
MethodSignature signature = (MethodSignature) point.getSignature();
Method method = signature.getMethod();
Class<?> targetClass = method.getDeclaringClass();
if(StringUtils.startsWith(key,"#{") && StringUtils.endsWith(key,"}")){
//判断是否是spel格式
if (StringUtils.containsAny(key, "#")) {
//获取参数值
Object[] args = point.getArgs();
//获取方法上参数的名称
/**
* 这里的DefaultParameterNameDiscoverers是Spring自带的解析器
* 在SpringMvc获取参数的过程中也有用到
*/
String[] parameterNames = new DefaultParameterNameDiscoverer().getParameterNames(method);
//使用Spring的el表达式解析
SpelExpressionParser parser = new SpelExpressionParser();
//注入到上下文对象中进行解析
StandardEvaluationContext context = new StandardEvaluationContext();
String[] parameterNames = pnd.getParameterNames(method);
for (int i = 0; i < parameterNames.length; i++) {
context.setVariable(parameterNames[i],args[i]);
context.setVariable(parameterNames[i], args[i]);
}
//定义解析模版
TemplateParserContext parserContent = new TemplateParserContext();
//解析返回给key
try {
key = parser.parseExpression(key,parserContent).getValue(context,String.class) + "";
}catch (Exception e){
key = parser.parseExpression(key, parserContext).getValue(context, String.class) + ":";
} catch (Exception e) {
throw new ServiceException("限流key解析异常!请联系管理员!");
}
}
StringBuilder stringBuffer = new StringBuilder();
StringBuilder stringBuffer = new StringBuilder(key);
if (rateLimiter.limitType() == LimitType.IP) {
// 获取请求ip
stringBuffer.append(ServletUtils.getClientIP()).append("-");
@ -99,7 +104,6 @@ public class RateLimiterAspect {
stringBuffer.append(RedisUtils.getClient().getId()).append("-");
}
stringBuffer.append(targetClass.getName()).append("-").append(method.getName());
stringBuffer.append(":").append(key);
return stringBuffer.toString();
}
}