菜品管理——公共字段自动填充

问题分析

业务表中存在一些公共字段，导致存在冗余代码，后期维护不便

实现思路

技术点：枚举  注解 AOP  反射

• 自定义注解 AutoFill，用于标识需要进行公共字段自动填充的方法
• 自定义切面类AutoFillAspect ，统一拦截加入了AutoFill注解的方法，通过反射为公共字段赋值
• 在Mapper的方法上加入AutoFill注解

代码开发

自定义注解 AutoFill，用于标识需要进行公共字段自动填充的方法

在common下添加一个包

自定义切面类AutoFillAspect ，统一拦截加入了AutoFill注解的方法，通过反射为公共字段赋值

在common下添加一个包

自定义切面类AutoFillAspect

在Mapper的方法上加入AutoFill注解

小测试

咱们通过前后端联调的方式进行测试

首先在前端进行修改

 记得提前打断点，我们会发现程序执行到log处

此时发现拦截到的就是update方法

编写前置通知

ctrl +alt +B 点击Signature

 转换为MethodSignature（其子接口）

得

getDeclaredMethod 抛出异常

invoke 抛出异常 

修改以后

常量类

代码直接用这些可能会容易出错，而且不规范

所以我们直接使用common中的常量类

 切面类代码

@Before("autoFillPiontCut()")
    public void autoFill(JoinPoint joinPoint){
        log.info("开始进行公共字段的自动填充....");
        //获取当前被拦截到的方法上的数据库操作类型
        MethodSignature signature = (MethodSignature)joinPoint.getSignature();//方法签名对象
        AutoFill autoFill = signature.getMethod().getAnnotation(AutoFill.class);//获得方法上的注解对象
        OperationType operationType = autoFill.value();//获得数据库的操作类型

        //获取被拦截的方法的参数————实体对象
        Object[] args = joinPoint.getArgs();
        if(args == null || args.length == 0){
            return;//防止出现空指针
        }
        Object entity = args[0];

        //准备赋值的数据
        LocalDateTime now = LocalDateTime.now();
        Long currentId = BaseContext.getCurrentId();

        //根据不同的操作类型，通过 反射 为对应的属性赋值
        if(operationType == OperationType.INSERT){
            //为4个公共字段赋值
            try {
                Method setCreateTime = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_CREATE_TIME, LocalDateTime.class);
                Method setCreteUser = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_CREATE_USER, Long.class);
                Method setUpdateTime = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_UPDATE_TIME, LocalDateTime.class);
                Method setUpdateUser = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_UPDATE_USER, long.class);

                //通过反射为对象属性赋值
                setCreateTime.invoke(entity,now);
                setCreteUser.invoke(entity,currentId);
                setUpdateTime.invoke(entity,now);
                setUpdateUser.invoke(entity,currentId);
            } catch (Exception e) {
                throw new RuntimeException(e);
            }

        } else if (operationType == OperationType.UPDATE) {
            //为2个公共字段赋值
            try {
                Method setUpdateTime = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_UPDATE_TIME, LocalDateTime.class);
                Method setUpdateUser = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_UPDATE_USER, long.class);

                //通过反射为对象属性赋值
                setUpdateTime.invoke(entity,now);
                setUpdateUser.invoke(entity,currentId);
            } catch (Exception e) {
                throw new RuntimeException(e);
            }
        }


    }

在mapper加入相应的注解

在切面已经为公共属性赋值，那么其他地方就不需要赋值

功能测试

后端运行，进行前后端联调

发现确实拦截到了update方法

没毛病