65 lines
1.8 KiB
Java
65 lines
1.8 KiB
Java
package com.imyeyu.spring.annotation;
|
|
|
|
import jakarta.servlet.http.HttpServletRequest;
|
|
import jakarta.servlet.http.HttpServletResponse;
|
|
import lombok.NoArgsConstructor;
|
|
import org.springframework.lang.NonNull;
|
|
import org.springframework.web.method.HandlerMethod;
|
|
import org.springframework.web.servlet.HandlerInterceptor;
|
|
|
|
/**
|
|
* 抽象访问频率限制，具体子类实现
|
|
*
|
|
* @author 夜雨
|
|
* @version 2021-08-16 18:07
|
|
*/
|
|
@NoArgsConstructor
|
|
public abstract class RequestRateLimitAbstractInterceptor implements HandlerInterceptor {
|
|
|
|
/**
|
|
* 构建接口标识
|
|
*
|
|
* @param handlerMethod 方法信息
|
|
* @return 接口标识
|
|
*/
|
|
protected String buildId(HandlerMethod handlerMethod) {
|
|
return handlerMethod.getMethod().getDeclaringClass().getSimpleName() + "." + handlerMethod.getMethod().getName();
|
|
}
|
|
|
|
/**
|
|
* 处理前
|
|
*
|
|
* @param req 请求
|
|
* @param resp 返回
|
|
* @param handler 处理方法
|
|
* @return true 为通过
|
|
*/
|
|
@Override
|
|
public boolean preHandle(@NonNull HttpServletRequest req,
|
|
@NonNull HttpServletResponse resp,
|
|
@NonNull Object handler) {
|
|
// 方法注解
|
|
if (handler instanceof HandlerMethod handlerMethod) {
|
|
RequestRateLimit requestRateLimit = handlerMethod.getMethodAnnotation(RequestRateLimit.class);
|
|
if (requestRateLimit == null) {
|
|
return true;
|
|
}
|
|
// 频率限制
|
|
return beforeRun(req, resp, buildId(handlerMethod), requestRateLimit.lifeCycle(), requestRateLimit.value());
|
|
}
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* 接口访问前触发
|
|
*
|
|
* @param req 请求
|
|
* @param resp 返回
|
|
* @param id 方法
|
|
* @param cycle 生命周期
|
|
* @param limit 访问限制
|
|
* @return true 为通过
|
|
*/
|
|
public abstract boolean beforeRun(HttpServletRequest req, HttpServletResponse resp, String id, int cycle, int limit);
|
|
}
|