JAVA中自定义注解和AOP使用

问题:有时候想对传入的参数,进行一些校验,并且方法比较多的情况下,可以参考使用注解或者直接AOP切点进行校验或者逻辑处理。

1 定义注解

@Inherited
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD})
public @interface RuntimeStatus {
   /**
     * 区分 project\stage\exam流程处理
     *
     * @return
     */
    RuntimeStatusMeta key();

    /**
     * 指定参数值
     *
     * @return
     */
    String propertyName() default "";

    /**
     * 指定参数是否是对象
     *
     * @return
     */
    boolean isArgObj() default false;
}

2 定义AOP

@Aspect
@Component
@Slf4j
public class RuntimeStatusAop {

    // cloud.core.annotion.RuntimeStatus:RuntimeStatus包地址

    @Pointcut(value = "@annotation(cloud.core.annotion.RuntimeStatus)")
    private void cutPermission() {

    }

// 直接使用AOP切面
//    @Pointcut("execution(public * cloud.controller.ExamController.*(..))")
//    private void examState() {
//
//    }

    @Before("cutPermission()")
    public void doSession(JoinPoint point) {
// 获取注解值
        MethodSignature methodSignature = (MethodSignature) point.getSignature();
        Method method = methodSignature.getMethod();
        
        RuntimeStatus runtimeStatus = method.getAnnotation(RuntimeStatus.class);
        RuntimeStatusMeta meta = runtimeStatus.key();
        String propertyName = runtimeStatus.propertyName();

        Object value = null;
        // 请求的方法参数值
        Object[] args = point.getArgs();

        // 获取ID
        if (runtimeStatus.isArgObj()) {
            value = PropertyUtil.getFieldValueByName(args[0], propertyName);
        } else {
            // 请求的方法参数名称
            LocalVariableTableParameterNameDiscoverer parameterNameDiscoverer = new LocalVariableTableParameterNameDiscoverer();
            String[] paramNames = parameterNameDiscoverer.getParameterNames(method);
            if (args != null && paramNames != null) {
                for (int i = 0; i < args.length; i++) {
//                    params += "  " + paramNames[i] + ": " + args[i];
                    if (paramNames[i].equals(propertyName)) {
                        value = args[i];
                        break;
                    }
                }
            }
        }
        log.info("propertyName => {}", propertyName);
        log.info("meta => {}", meta.getName() + "、" + meta.getCode());
        log.info("value => {}",value);
        // 获取了参数值,接下来逻辑处理
        。。。

    }

说明:注解和AOP各有优劣势,AOP表达式直接切点的话,可以使用在无参数传入或者参数具备同样格式的方式;注解的话适用于需要处理传入参数,并且参数名称或则对象不统一的情况;

3 使用

(1) 如果传入的是基本类型的参数,直接PropertyName 直接写参数名称

    @RequestMapping(value = "/delById", method = RequestMethod.POST)
    @RuntimeStatus(key = RuntimeStatusMeta.EXAM, propertyName = "id")
    public ResultUtil delById(int id) {
    
    }

(2) 如果传入的是对象类型参数,标记isArgObj为true,然后指定对象里面的属性名称,这样就可以在AOP里,单独对是对象参数获取其属性值

    @RequestMapping(value = "/update", method = RequestMethod.POST)
    @RuntimeStatus(key = RuntimeStatusMeta.EXAM,isArgObj = true,propertyName = "id")
    public ResultUtil update(@RequestBody Exam obj) {
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容