自定义注解的使用

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()};
}

你可能感兴趣的:(java,jvm,spring)