公共字段自动填充统一处理

公共字段自动填充

• 通过AOP面向切面编程统一处理
• 技术要点：枚举，注解，AOP,反射

具体步骤

1. 自定义注解，用于标识需要公共字段填充的方法

   • 创建自定义注解@AutoFill

     package com.sky.annotation;
     
     import com.sky.enumeration.OperationType;
     
     import java.lang.annotation.ElementType;
     import java.lang.annotation.Retention;
     import java.lang.annotation.RetentionPolicy;
     import java.lang.annotation.Target;
     
     @Target({ElementType.METHOD})
     @Retention(RetentionPolicy.RUNTIME)
     public @interface AutoFill {
         //数据库操作
         OperationType value();
     }
     

2. 自定义切面类，同意拦截加入了AtuoFill注解方法，通过反射为公共字段赋值

   • 创建切面类AutoFillAspect

     package com.sky.aspect;
     
     import com.sky.annotation.AutoFill;
     import com.sky.constant.AutoFillConstant;
     import com.sky.context.BaseContext;
     import com.sky.enumeration.OperationType;
     import lombok.extern.slf4j.Slf4j;
     import org.aspectj.lang.JoinPoint;
     import org.aspectj.lang.Signature;
     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 java.lang.reflect.InvocationTargetException;
     import java.lang.reflect.Method;
     import java.time.LocalDateTime;
     
     /**
      * 自定义切面实现公共字段填充逻辑
      */
     @Aspect
     @Component
     @Slf4j
     public class AutoFillAspect {
         /**
          * 切入点
          */
         @Pointcut("execution(* com.sky.mapper.*.*(..))&& @annotation(com.sky.annotation.AutoFill)")
         public void  autoFillPointCut(){}
         /**
          * 前置通知
          */
         @Before("autoFillPointCut()")
         public void autoFill(JoinPoint joinPoint) throws NoSuchMethodException, InvocationTargetException, IllegalAccessException {
             //获取当前被拦截的方法的数据操作类型
             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){
                 Method setCreateTime = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_CREATE_TIME, LocalDateTime.class);
                 Method setCreateUser = 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);
                 setCreateUser.invoke(entity,currentId);
                 setUpdateTime.invoke(entity,now);
                 setUpdateUser.invoke(entity,currentId);
     
     
             }else if (operationType == OperationType.UPDATE){
     
             }
     
         }
     }
     

3. 在Mapper的方法上加入@AutoFill注解

   @AutoFill(value = OperationType.INSERT)
   @Insert("insert into employee (name,username,password,phone,sex,id_number,status,create_time,update_time,create_user,update_user)"
   +"values"+
   "(#{name},#{username},#{password},#{phone},#{sex},#{idNumber},#{status},#{createTime},#{updateTime},#{createUser},#{updateUser})")
   void insert(Employee employee);