自定义注解的使用

秒速五厘米 2023-10-09 17:43 80阅读 0赞

1.首先定义一个自定义注解

import java.lang.annotation.*;
    
    @Target(ElementType.METHOD)  表明这个注解运行在一个方法上
    @Retention(RetentionPolicy.RUNTIME) 由JVM 加载，包含在类文件中，在运行时可以被获取到
    @Documented  表明该注解标记的元素可以被Javadoc 或类似的工具文档化
    public @interface CheckToken {
    }

2.定义一个aop,每次在这个注解出现时来拦截这个注解并进行逻辑检验

@Component
    @Aspect

在此类加上这两个注解表明这是一个aop并加载到spring容器中

@Before("@annotation(com.datanew.reimburse.sign.CheckToken)")
    public Object checked(ProceedingJoinPoint pjp){
    
        String methodName = pjp.getSignature().getName();
        String clazzName = pjp.getTarget().getClass().getSimpleName();
    
        //获取请求参数：
        MethodSignature ms = (MethodSignature) pjp.getSignature();
        //获取请求参数类型
        String[] parameterNames = ms.getParameterNames();
        //获取请求参数值
        Object[] parameterValues = pjp.getArgs();
        String[] results = getParams(parameterNames, parameterValues);
        String params = results[0];
    
        CheckToken signVerifyAnon = AnnotationUtil.getAnnotation(((MethodSignature) pjp.getSignature()).getMethod(), CheckToken.class);
    
        if(signVerifyAnon != null){
    
            if("looks".equals(methodName)){
    
                throw new SignAuthFailedException("请求错误");
            }
        }
    
        return null;
    }
    
    protected String[] getParams(String[] parameterNames, Object[] parameterValues) {
        StringBuilder sb = new StringBuilder("{");
        //组合请求参数，进行日志打印
        if (parameterNames != null && parameterNames.length > 0) {
            for (int i = 0; i < parameterNames.length; i++) {
                if (parameterNames[i].equals("bindingResult")) {
                    break;
                }
                if ((parameterValues[i] instanceof HttpServletRequest) || (parameterValues[i] instanceof HttpServletResponse)) {
                    sb.append("\"").
                            append(parameterNames[i]).append("\":").append(parameterValues[i]);
                } else {
                    sb.append("\"").
                            append(parameterNames[i]).append("\":")
                            .append(JSONUtil.toJsonStr(parameterValues[i]));
                }
            }
        }
        sb.append("}");
        return new String[]{sb.toString()};
    }