需求

     使用 MyBatis 插入或修改某条记录时，能自动设置数据表里的 create_time 和 update_time 字段，即自动给实体类对象的 createTime 和 updateTime 属性赋值。（如果使用 MyBatis-Plus，该功能很容易实现，现在针对的场景是仅使用 MyBatis）

解决方案

​    使用AOP的原理，在执行新增或修改前，用反射获取方法的第一个参数，给它的 createTime 和 updateTime 属性赋值 。（约定：方法的第一个参数是要操作的实体类）

代码

1、自定义注解：

/**
 * 标识 需要自动填充的方法
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface AutoFill {
    OperationType value();
}

2、枚举类：指定方法类型是 insert 还是 update，insert 操作需要 同时设置 createTime 和 updateTime，update操作只设置 updateTime 。

/**
 * 操作类型
 */
public enum OperationType {
    UPDATE,
    INSERT
}

3、切面：

/**
 * 自定义切面 自动填充公共字段
 */
@Aspect
@Component
public class AutoFillAspect {

    /**
     * 切入点
     *   * com.example.mapper.*.*(..)  需要能定位到mapper接口的位置
     *   com.example.autofill.AutoFill  是上面写的自定义注解的路径
     */
    @Pointcut("execution(* com.example.mapper.*.*(..)) && @annotation(com.example.autofill.AutoFill)")
    public void autoFillPointCut() {
    }


    /**
     * 前置通知，insert update之前 填充时间
     */
    @Before("autoFillPointCut()")
    public void autoFill(JoinPoint joinPoint) throws Exception {
        //获取到当前被拦截的方法上的数据库操作类型
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();//方法签名对象
        AutoFill autoFill = signature.getMethod().getAnnotation(AutoFill.class);//获得方法上的注解对象
        OperationType operationType = autoFill.value();//获得数据库操作类型   insert  update
        // insert  update  方法参数是实体类 需要放在第一个
        // 获取实体参数对象
        Object[] args = joinPoint.getArgs();
        if (args == null || args.length == 0) {
            return;
        }
        Object entity = args[0];
        // 为 createTime 和 updateTime 属性赋值
        LocalDateTime localDateTime = LocalDateTime.now();
        if (operationType == OperationType.INSERT) {
            // insert 操作 两个时间都赋值
            Method setCreateTime = entity.getClass().getMethod(SET_CREATE_TIME, LocalDateTime.class);
            Method setUpdateTime = entity.getClass().getMethod(SET_UPDATE_TIME, LocalDateTime.class);
            setCreateTime.invoke(entity, localDateTime);
            setUpdateTime.invoke(entity, localDateTime);
        } else if (operationType == OperationType.UPDATE) {
            // update 操作 只设置更新时间
            Method setUpdateTime = entity.getClass().getMethod(SET_UPDATE_TIME, LocalDateTime.class);
            setUpdateTime.invoke(entity, localDateTime);
        }
    }
}

测试案例

实体类：

import lombok.Data;
import java.time.LocalDateTime;

@Data
public class User{
    private Long id;
    private String name;
    private LocalDateTime createTime;
    private LocalDateTime updateTime;
}

mapper接口：

/**
 * 添加用户
 */
@AutoFill(value = OperationType.INSERT)
Integer addUser(User user);

xml文件：

<insert id="addUser" parameterType="user" useGeneratedKeys="true" keyProperty="id">
    insert into user(name, create_time, update_time)
    values (#{name}, #{createTime}, #{updateTime})
</insert>

测试方法：

@Resource
private UserMapper userMapper;

void test(){
    User user = new User();
    user.setName("张三");  // 无需手动设置 createTime 和 updateTime 属性
    userMapper.addUser(user);
}