Initial project

src/main/java/com/imyeyu/spring/annotation/RequestRateLimitAbstractInterceptor.java

@@ -0,0 +1,56 @@
+package com.imyeyu.spring.annotation;
+
+import jakarta.servlet.http.HttpServletRequest;
+import jakarta.servlet.http.HttpServletResponse;
+import org.springframework.lang.NonNull;
+import org.springframework.web.method.HandlerMethod;
+import org.springframework.web.servlet.HandlerInterceptor;
+
+/**
+ * 抽象访问频率限制，具体子类实现
+ *
+ * @author 夜雨
+ * @version 2021-08-16 18:07
+ */
+public abstract class RequestRateLimitAbstractInterceptor implements HandlerInterceptor {
+
+	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);
+}