no message

yangyang

2025-07-15 ff370ab5aa1f046824e73e1cafab1be6acb13780

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package com.zy.asrs.openapi.config; 
 
import com.zy.asrs.common.sys.entity.OperateLog; 
import com.zy.asrs.common.sys.service.OperateLogService; 
import com.zy.asrs.framework.annotations.AppAuth; 
import org.springframework.beans.factory.annotation.Autowired; 
import org.springframework.lang.Nullable; 
import org.springframework.stereotype.Component; 
import org.springframework.web.method.HandlerMethod; 
import org.springframework.web.servlet.ModelAndView; 
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter; 
 
import javax.servlet.http.HttpServletRequest; 
import javax.servlet.http.HttpServletResponse; 
import java.lang.reflect.Method; 
 
@Component 
public class AdminInterceptor extends HandlerInterceptorAdapter { 
 
    @Autowired 
    private OperateLogService operateLogService; 
 
    @Override 
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { 
        cors(response); 
        if (handler instanceof org.springframework.web.servlet.resource.ResourceHttpRequestHandler) { 
            return true; 
        } 
        // 跨域设置 
        // response.setHeader("Access-Control-Allow-Origin", "*"); 
        HandlerMethod handlerMethod = (HandlerMethod) handler; 
        Method method = handlerMethod.getMethod(); 
        if (method.isAnnotationPresent(AppAuth.class)){ 
            AppAuth annotation = method.getAnnotation(AppAuth.class); 
            if (annotation.value().equals(AppAuth.Auth.CHECK)){ 
                request.setAttribute("appAuth", 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) throws Exception { 
//        Object r = request.getAttribute("cool-response"); 
    } 
 
    /** 
     * 跨域 
     */ 
    public static void cors(HttpServletResponse response){ 
        // 跨域设置 
        response.setHeader("Access-Control-Allow-Origin", "*"); 
        response.setHeader("Access-Control-Allow-Credentials", "true"); 
        response.setHeader("Access-Control-Allow-Methods", "*"); 
        response.setHeader("Access-Control-Allow-Headers", "Content-Type,Access-Token"); 
        response.setHeader("Access-Control-Expose-Headers", "*"); 
 
    } 
 
}