From 3372040097ad2c01aeb6fd6485e89f19bf81b316 Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期三, 18 三月 2026 17:02:34 +0800
Subject: [PATCH] #
---
src/main/java/com/zy/common/config/AdminInterceptor.java | 97 ++++++++++++++++++++++++++++++++++++------------
1 files changed, 73 insertions(+), 24 deletions(-)
diff --git a/src/main/java/com/zy/common/config/AdminInterceptor.java b/src/main/java/com/zy/common/config/AdminInterceptor.java
index a9badbf..03c488d 100644
--- a/src/main/java/com/zy/common/config/AdminInterceptor.java
+++ b/src/main/java/com/zy/common/config/AdminInterceptor.java
@@ -8,6 +8,7 @@
import com.zy.common.utils.Http;
import com.zy.system.entity.*;
import com.zy.system.service.*;
+import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.lang.Nullable;
@@ -19,12 +20,20 @@
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import java.lang.reflect.Method;
+import java.util.Date;
+import java.util.LinkedHashMap;
+import java.util.Map;
/**
* Created by vincent on 2019-06-13
*/
@Component
+@Slf4j
public class AdminInterceptor implements HandlerInterceptor {
+
+ private static final String ATTR_USER_ID = "userId";
+ private static final String ATTR_OPERATE_LOG = "operateLog";
+ private static final Long SUPER_USER_ID = 9527L;
@Value("${super.pwd}")
private String superPwd;
@@ -45,6 +54,8 @@
if (handler instanceof org.springframework.web.servlet.resource.ResourceHttpRequestHandler) {
return true;
}
+ ManagerAuth annotation = resolveManagerAuth(handler);
+ boolean requiresCheck = annotation != null && annotation.value().equals(ManagerAuth.Auth.CHECK);
// super璐﹀彿
String token = request.getHeader("token");
if (token!=null) {
@@ -57,32 +68,43 @@
return false;
}
if ("super".equals(deToken.substring(13))) {
- request.setAttribute("userId", 9527);
+ request.setAttribute(ATTR_USER_ID, SUPER_USER_ID);
+ if (requiresCheck) {
+ cacheOperateLog(request, annotation.memo(), SUPER_USER_ID);
+ }
return true;
}
}
}
// 璺ㄥ煙璁剧疆
// response.setHeader("Access-Control-Allow-Origin", "*");
- HandlerMethod handlerMethod = (HandlerMethod) handler;
- Method method = handlerMethod.getMethod();
- if (method.isAnnotationPresent(ManagerAuth.class)){
- ManagerAuth annotation = method.getAnnotation(ManagerAuth.class);
- if (annotation.value().equals(ManagerAuth.Auth.CHECK)){
- return check(request, response, annotation.memo());
- }
+ if (!(handler instanceof HandlerMethod)) {
+ return true;
+ }
+ if (requiresCheck){
+ return check(request, response, annotation.memo());
}
return true;
}
@Override
public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, @Nullable ModelAndView modelAndView) {
-// Object obj = request.getAttribute("operateLog");
-// if (obj instanceof OperateLog) {
-// OperateLog operate = (OperateLog) obj;
-// operate.setResponse(String.valueOf(response.getStatus()));
-// operateLogService.save(operate);
-// }
+ }
+
+ @Override
+ public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler,
+ @Nullable Exception ex) {
+ Object obj = request.getAttribute(ATTR_OPERATE_LOG);
+ if (!(obj instanceof OperateLog)) {
+ return;
+ }
+ OperateLog operateLog = (OperateLog) obj;
+ operateLog.setResponse(buildResponseContent(response, ex));
+ try {
+ operateLogService.save(operateLog);
+ } catch (Exception saveEx) {
+ log.warn("淇濆瓨鎿嶄綔鏃ュ織澶辫触, uri={}", request.getRequestURI(), saveEx);
+ }
}
private boolean check(HttpServletRequest request, HttpServletResponse response, String memo) {
@@ -106,17 +128,9 @@
Http.response(response, BaseRes.LIMIT);
return false;
}
- // 鎿嶄綔鏃ュ織
- OperateLog operateLog = new OperateLog();
- operateLog.setAction(Cools.isEmpty(memo)?request.getRequestURI():memo);
- operateLog.setIp(request.getRemoteAddr());
- operateLog.setUserId(user.getId());
- operateLog.setRequest(JSON.toJSONString(request.getParameterMap()));
- operateLog.setResponse(JSON.toJSONString(request.getRequestURI()));
- operateLogService.save(operateLog);
// 璇锋眰缂撳瓨
- request.setAttribute("userId", user.getId());
-// request.setAttribute("operateLog", operateLog);
+ request.setAttribute(ATTR_USER_ID, user.getId());
+ cacheOperateLog(request, memo, user.getId());
return true;
} catch (Exception e){
Http.response(response, BaseRes.DENIED);
@@ -154,4 +168,39 @@
}
+ private ManagerAuth resolveManagerAuth(Object handler) {
+ if (!(handler instanceof HandlerMethod)) {
+ return null;
+ }
+ HandlerMethod handlerMethod = (HandlerMethod) handler;
+ Method method = handlerMethod.getMethod();
+ if (!method.isAnnotationPresent(ManagerAuth.class)) {
+ return null;
+ }
+ return method.getAnnotation(ManagerAuth.class);
+ }
+
+ private void cacheOperateLog(HttpServletRequest request, String memo, Long userId) {
+ if (userId == null || request.getAttribute(ATTR_OPERATE_LOG) != null) {
+ return;
+ }
+ OperateLog operateLog = new OperateLog();
+ operateLog.setAction(Cools.isEmpty(memo) ? request.getRequestURI() : memo);
+ operateLog.setIp(request.getRemoteAddr());
+ operateLog.setUserId(userId);
+ operateLog.setRequest(JSON.toJSONString(request.getParameterMap()));
+ operateLog.setCreateTime(new Date());
+ request.setAttribute(ATTR_OPERATE_LOG, operateLog);
+ }
+
+ private String buildResponseContent(HttpServletResponse response, Exception ex) {
+ Map<String, Object> result = new LinkedHashMap<>();
+ result.put("status", response.getStatus());
+ if (ex != null) {
+ result.put("error", ex.getClass().getName());
+ result.put("message", ex.getMessage());
+ }
+ return JSON.toJSONString(result);
+ }
+
}
--
Gitblit v1.9.1