package com.ruoyi.framework.aspectj;
|
|
import com.ruoyi.common.annotation.RateLimiter;
|
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;
|
import org.aspectj.lang.annotation.Aspect;
|
import org.aspectj.lang.annotation.Before;
|
import org.aspectj.lang.annotation.Pointcut;
|
import org.aspectj.lang.reflect.MethodSignature;
|
import org.redisson.api.RateType;
|
import org.springframework.stereotype.Component;
|
|
import java.lang.reflect.Method;
|
|
/**
|
* 限流处理
|
*
|
* @author Lion Li
|
*/
|
@Slf4j
|
@Aspect
|
@Component
|
public class RateLimiterAspect {
|
|
// 配置织入点
|
@Pointcut("@annotation(com.ruoyi.common.annotation.RateLimiter)")
|
public void rateLimiterPointCut() {
|
}
|
|
@Before("rateLimiterPointCut()")
|
public void doBefore(JoinPoint point) throws Throwable {
|
RateLimiter rateLimiter = getAnnotationRateLimiter(point);
|
int time = rateLimiter.time();
|
int count = rateLimiter.count();
|
String combineKey = getCombineKey(rateLimiter, point);
|
try {
|
RateType rateType = RateType.OVERALL;
|
if (rateLimiter.limitType() == LimitType.CLUSTER) {
|
rateType = RateType.PER_CLIENT;
|
}
|
long number = RedisUtils.rateLimiter(combineKey, rateType, count, time);
|
if (number == -1) {
|
throw new ServiceException("访问过于频繁，请稍后再试");
|
}
|
log.info("限制令牌 => {}, 剩余令牌 => {}, 缓存key => '{}'", count, number, combineKey);
|
} catch (ServiceException e) {
|
throw e;
|
} catch (Exception e) {
|
throw new RuntimeException("服务器限流异常，请稍后再试");
|
}
|
}
|
|
/**
|
* 是否存在注解，如果存在就获取
|
*/
|
private RateLimiter getAnnotationRateLimiter(JoinPoint joinPoint) {
|
Signature signature = joinPoint.getSignature();
|
MethodSignature methodSignature = (MethodSignature) signature;
|
Method method = methodSignature.getMethod();
|
|
if (method != null) {
|
return method.getAnnotation(RateLimiter.class);
|
}
|
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){
|
// 获取客户端实例id
|
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();
|
}
|
|
}
|
