注解配合拦截器使用

小灰灰 2023-01-15 03:44 126阅读 0赞

package com.picc.vx.portal.annotation;
    
    import java.lang.annotation.ElementType;
    import java.lang.annotation.Retention;
    import java.lang.annotation.RetentionPolicy;
    import java.lang.annotation.Target;
    
    /**
     * 记录请求日志
     */
    @Target({ElementType.METHOD, ElementType.TYPE})
    @Retention(RetentionPolicy.RUNTIME)
    public @interface LogRecord {
        /**
         * 功能名称
         */
        String modelName();
    
        /**
         * 操作类型
         */
        String operate();
    
        /**
         * 参数位置 as[0].name[0].
         */
        int num() default 0;
    
        /**
         * 是否数组
         */
        boolean isArray() default false;
    
        /**
         * 提取名称
         */
        String name() default "name";
    
    
    }

package com.picc.vx.portal.interceptor;
    
    import com.alibaba.fastjson.JSON;
    import com.picc.vx.portal.annotation.LogRecord;
    import org.springframework.core.MethodParameter;
    import org.springframework.lang.Nullable;
    import org.springframework.web.method.HandlerMethod;
    import org.springframework.web.multipart.MultipartFile;
    import org.springframework.web.multipart.MultipartHttpServletRequest;
    import org.springframework.web.multipart.MultipartResolver;
    import org.springframework.web.multipart.support.StandardServletMultipartResolver;
    import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;
    
    import javax.servlet.http.HttpServletRequest;
    import javax.servlet.http.HttpServletResponse;
    import java.lang.reflect.Field;
    import java.util.HashMap;
    import java.util.Map;
    
    /**
     * 日志记录拦截器
     */
    public class LogRecordInterceptor extends HandlerInterceptorAdapter {
    
        @Override
        public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
            System.out.println("这是进入前.....");
            return true;
        }
    
        /**
         * 处理器处理之前调用
         */
        @Override
        public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, @Nullable Exception ex) {
            HandlerMethod handlerMethod = (HandlerMethod) handler;
            LogRecord logRecord = findLog(handlerMethod);
            if (null == logRecord) {
                return;
            }
            String name = logRecord.name();
            String value = "";
            String requestBody = getFormParam(request);
            MethodParameter[] parameters = handlerMethod.getMethodParameters();
            MethodParameter parameter = parameters[logRecord.num()];
            Field[] fields = parameter.getParameter().getType().getDeclaredFields();
            Object o = JSON.parseArray(requestBody, parameter.getParameter().getType());
            try {
                for (Field field : fields) {
                    if (name.equals(field.getName())) {
                        value = field.get(o).toString();
    
                    }
                }
            } catch (IllegalAccessException e) {
                e.printStackTrace();
            }
    
    
            //如果没有添加权限注解则直接跳过允许访问
            if (logRecord == null) {
                return;
            }
    
            //获取注解中的值
            String modelName = logRecord.modelName();
            String operate = logRecord.operate();
    
            String content = "【某某某】" + operate + "【" + modelName + "】" + value;
    
            System.out.println("信息：" + content);
    
            return;
        }
    
        /**
         * 根据handlerMethod返回注解信息
         *
         * @param handlerMethod 方法对象
         * @return Log注解
         */
        private LogRecord findLog(HandlerMethod handlerMethod) {
            //在方法上寻找注解
            LogRecord logRecord = handlerMethod.getMethodAnnotation(LogRecord.class);
            if (logRecord == null) {
                //在类上寻找注解
                logRecord = handlerMethod.getBeanType().getAnnotation(LogRecord.class);
            }
    
            return logRecord;
        }
    
        private String getFormParam(HttpServletRequest request) {
            MultipartResolver resolver = new StandardServletMultipartResolver();
            MultipartHttpServletRequest mRequest = resolver.resolveMultipart(request);
    
            Map<String, Object> param = new HashMap<>();
            Map<String, String[]> parameterMap = mRequest.getParameterMap();
            if (!parameterMap.isEmpty()) {
                param.putAll(parameterMap);
            }
            Map<String, MultipartFile> fileMap = mRequest.getFileMap();
            if (!fileMap.isEmpty()) {
                for (Map.Entry<String, MultipartFile> fileEntry : fileMap.entrySet()) {
                    MultipartFile file = fileEntry.getValue();
                    param.put(fileEntry.getKey(), file.getOriginalFilename() + "(" + file.getSize() + " byte)");
                }
            }
            return toJson(param);
        }
    
        private static String toJson(Object object) {
            return JSON.toJSONStringWithDateFormat(object, "yyyy-MM-dd HH:mm:ss");
        }
    
    }

package com.picc.vx.portal.config;
    
    import com.picc.vx.portal.interceptor.LogRecordInterceptor;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
    import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
    
    /**
     *  拦截器配置类
     */
    @Configuration
    public class InterceptorConfig implements WebMvcConfigurer {
    
        @Override
        public void addInterceptors(InterceptorRegistry registry) {
            //权限验证拦截
            registry.addInterceptor(new LogRecordInterceptor()).addPathPatterns("com/picc/vx/portal/ctrl/**");
    
        }
    }