springboot自定义注解带参,添加日志功能

比眉伴天荒 2022-09-05 12:50 107阅读 0赞

自定义注解代码:

import java.lang.annotation.*;
    
    
    @Target({ElementType.METHOD, ElementType.TYPE})
    @Retention(RetentionPolicy.RUNTIME)
    @Documented
    public @interface Xxxx{
        /**
         * 模块名称
         *
         * @return
         */
        String modelName() default "";
    
        /**
         * 功能名称
         *
         * @return
         */
        String methodName() default "";
    }

配合对接AOP代码:

import com.alibaba.fastjson.JSON;
    import lombok.extern.slf4j.Slf4j;
    import org.aspectj.lang.JoinPoint;
    import org.aspectj.lang.annotation.Aspect;
    import org.aspectj.lang.annotation.Before;
    import org.aspectj.lang.annotation.Pointcut;
    import org.aspectj.lang.reflect.MethodSignature;
    import org.springframework.stereotype.Component;
    import org.springframework.web.multipart.MultipartFile;
    
    import javax.servlet.http.HttpServletRequest;
    import javax.servlet.http.HttpServletResponse;
    import java.lang.reflect.Method;
    import java.util.Arrays;
    
    
    @Aspect
    @Component
    @Slf4j
    public class XxxxAop {
    
        @Pointcut("@annotation(自定义注解路径)")
        public void Xxxx() {
        }
    
        @Before("Xxxx()")
        public void doBefore(JoinPoint joinPoint) throws Exception {
            // 方法描述
            MethodSignature signature = (MethodSignature) joinPoint.getSignature();
            Method methodFromClass = joinPoint.getSignature()
    			.getDeclaringType()
    			.getDeclaredMethod(joinPoint.getSignature().getName(), signature.getParameterTypes());
            TempLog des = methodFromClass.getDeclaredAnnotation(自定义注解名.class);
            String[] parameterNames = signature.getParameterNames();
            Object[] args = joinPoint.getArgs();
            String userId = "", ecCode = "", params = "";
            for (Object arg : args) {
                if (arg instanceof SessionMap) {
    			    //解析用户信息
                    userId = ((SessionMap) arg).getSsoTokenInfo().getUserId();
                    ecCode = ((SessionMap) arg).getSsoTokenInfo().getEcCode();
                    params += "用户信息,";
                } else if (arg instanceof MultipartFile) {
                    params = params + ((MultipartFile) arg).getOriginalFilename() + ",";
                } else if (arg instanceof HttpServletResponse) {
                    params = params + "HttpServletResponse ,";
                } else if (arg instanceof HttpServletRequest) {
                    params = params + "HttpServletRequest ,";
                } else {
                    if (arg == null || arg.equals("")) {
                        params = params + "null,";
                    } else {
                        params = params + JSON.toJSON(arg) + ",";
                    }
                }
            }
            log.info("|#{} |#{} |#用户 EC_CODE = {} , USER_ID = {} |# 参数名称 {} ,参数 [{}]"
                    , des.modelName()
                    , des.methodName()
                    , ecCode
                    , userId
                    , Arrays.toString(parameterNames)
                    , params);
        }
    
    }

使用自定义注解:

@TempLog(modelName = ".....", methodName = ".....")
        @RequestMapping(value = "/A", method = RequestMethod.POST)
        public Result A(@RequestBody SessionMap sessionMap) {
           
        }