Mybatis-Plus自动填充的实现示例

在常用业务中有些属性需要配置一些默认值，MyBatis-Plus提供了实现此功能的插件。在这里修改user表添加 create_time 字段和 update_time 字段，在User类中添加对应属性。

1、为需要自动填充的属性添加注解 @TableField

提供了4种自动填充策略：DEFAULT，默认不处理。INSERT，插入填充字段。UPDATE，更新填充字段。INSERT_UPDATE，插入和更新填充字段。

@Data
public class User {
  private Long id;
  private String name;
  private Integer age;
  private String email;

  @TableField(fill = FieldFill.INSERT)
  private Date createTime;
  @TableField(fill = FieldFill.INSERT_UPDATE)
  private Date updateTime;
}

2、实现字段填充控制器，编写自定义填充规则

实现 MetaObjectHandler 接口，实现 insertFill 和 updateFill 方法，此处的 create_time 和update_time字段需要插入时填充值， 只有 update_time 字段在修改时需要填充，所以策略如下。

//需要将自定义填充控制器注册为组件

@Component
public class MyMetaObjectHandler implements MetaObjectHandler {

  private static final Logger LOGGER= LoggerFactory.getLogger(MyMetaObjectHandler.class);

  //insert操作时要填充的字段
  @Override
  public void insertFill(MetaObject metaObject) {
    LOGGER.info("start insert fill ...");
    //根据属性名字设置要填充的值
    this.setFieldValByName("createTime",new Date(),metaObject);
    this.setFieldValByName("updateTime",new Date(),metaObject);
  }

  //update操作时要填充的字段
  @Override
  public void updateFill(MetaObject metaObject) {
    LOGGER.info("start insert fill ...");
    this.setFieldValByName("updateTime",new Date(),metaObject);
  }
}

3、插入数据测试

@RunWith(SpringRunner.class)
@SpringBootTest
public class CRUDTest {
  @Autowired
  private UserMapper userMapper;

  @Test
  public void testInsert(){
    User user = new User();
    user.setName("jack11");
    user.setAge(20);
    user.setEmail("4849111@qq.com");

    int result= userMapper.insert(user);
    System.out.println(result);
    System.out.println(user);
  } 
}

4、修改数据测试

@Test
public void testUpdate(){
  User user = new User();
  user.setId(2L);
  user.setName("Jackie");
  int result = userMapper.updateById(user);
  System.out.println(result);
}

一次插入数据后，create_time和update_time都被填充了设置的时间，做update操作后只有update_time的进行了填充修改。

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持小牛知识库。