mybatis-plus 新增/修改如何实现自动填充指定字段

2023-05-28 原文

mybatis-plus 新增/修改自动填充指定字段

1.需要修改的字段在模型类上添加

@TableField(fill = FieldFill.xxx) 注解。

FieldFill的选项：

在这里插入图片描述

哪个字段在什么时候填充需要手动设置注解

在这里插入图片描述

2.新建一个MetaObjectHandler的实现类

MyMetaObjectHandler

package com.demo.mp.config;
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;
import java.util.Objects;
@Component // 注入bean
public class MyMetaObjectHandler implements MetaObjectHandler {
  /**
   * @Description: 新增时填充
   *
   * @auther: zpq
   * @date: 2020/11/2 2:26 下午
   */
  @Override
  public void insertFill(MetaObject metaObject) {
    // this.setFieldValByName(字段名, 内容, metaObject：用这个对象进行填充用的，称为元数据对象);
    this.setFieldValByName("createBy", "admin", metaObject);
    // ================setFieldValByName 方法内的内容，有兴趣可以进去看一看，中文注释的==================================================
    /// **
    // * 通用填充
    // *
    // * @param fieldName  java bean property name
    // * @param fieldVal   java bean property value
    // * @param metaObject meta object parameter
    // */
    // default MetaObjectHandler setFieldValByName(String fieldName, Object fieldVal, MetaObject
    // metaObject) {
    //  if (Objects.nonNull(fieldVal) && metaObject.hasSetter(fieldName)) {
    //    metaObject.setValue(fieldName, fieldVal);
    //  }
    //  return this;
    // }
  }
  /**
   * * @Description: 更新时填充
   *
   * @auther: zpq
   * @date: 2020/11/2 11:37 上午
   */
  @Override
  public void updateFill(MetaObject metaObject) {
    this.setFieldValByName("updateBy", "admin", metaObject);
  }
}

完成之后就可以使用了

注意：当你设置了逻辑删除，在逻辑删除的时候，是不会进行修改填充的，这个需要注意一下。

通用字段自动填充的最佳实践总结

在进行持久层数据维护（新增或修改）的时候，我们通常需要记录一些非业务字段，比如：create_time、update_time、update_by、create_by等用来维护数据记录的创建时间、修改时间、修改人、创建人等信息。通常情况下我们需要对这些字段进行手动赋值。赋值的过程也比较冗余，都是重复操作。

通常是为create_time赋值为系统的当前时间、update_time赋值为系统修改操作执行时的当前时间。
create_by(创建人)、update_by(修改人)赋值为当前的登录用户的用户名

xxxYyyZzz.setUpdateBy("zimug");    //数据记录更新操作人
xxxYyyZzz.setUpdateTime(new Date());   //数据记录更新操作的时间

Mybatis plus 为我们提供了一种一劳永逸的自动化赋值方式。

一、调整数据库表结构

以mysql数据库环境下的xxx_yyy_zzz表为例，在原有的表字段的基础上，添加下面的四个通用数据维护字段。

ALTER TABLE `xxx_yyy_zzz` ADD COLUMN `create_by` VARCHAR(64) NOT NULL COMMENT '本条记录创建人';
ALTER TABLE `xxx_yyy_zzz` ADD COLUMN `create_time` DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '本条记录创建时间';
ALTER TABLE `xxx_yyy_zzz` ADD COLUMN `update_by` VARCHAR(64) NOT NULL COMMENT '本条记录修改人';
ALTER TABLE `xxx_yyy_zzz` ADD COLUMN `update_time` DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '本条记录的修改时间';

二、通用维护信息父类-自动赋值的字段

既然我们对某一张表的数据进行新增创建、修改信息的维护，我们的实体类也要做必要的调整。为了避免为每一个实体类都加上这四个成员变量，我们定义一个父类BaseColumns。

@Data
public class BaseColumns {
  /**
   * 本条记录创建人，insert操作的时候自动为该字段赋值
   */
  @TableField(fill = FieldFill.INSERT 
  private String createBy;
  /**
   * 本条记录创建时间，insert操作的时候自动为该字段赋值
   */
  @TableField(fill = FieldFill.INSERT)
  private LocalDateTime  createTime;
  /**
   * 本条记录更新人，insert或update操作的时候自动为该字段赋值，select = false
   */
  @TableField(fill = FieldFill.INSERT_UPDATE,select = false)  
  private String updateBy;
  /**
   * 本条记录更新时间，insert或update操作的时候自动为该字段赋值，select = false
   */
  @TableField(fill = FieldFill.INSERT_UPDATE,select = false)
  private LocalDateTime updateTime;
}

fill = FieldFill.INSERT表示insert操作的时候自动为该字段赋值
fill = FieldFill.INSERT_UPDATE表示nsert或update操作的时候自动为该字段赋值
select = false表示在使用Mybatis Wrapper条件构造器进行查询的时候，不查询这个属性对应的数据库字段。数据修改时间操作人通常对于运维更有意义，所以通常不需要展示在web页面上，所以通常select查询的时候不包含它。（这个内容与我们本机的字段自动填充没有太直接的联系，但是在实际应用中是有意义的）

三、实体类的实现

下文实体类XxxYyyZzz对应数据库中的xxx_yyy_zzz表，除了以上四个通用字段，xxx_yyy_zzz表还包含其他的业务字段。

@Data
@EqualsAndHashCode(callSuper = true)
public class XxxYyyZzz extends BaseColumns {
    //其他的属性字段
}

四、自动赋值的规则

@Component
public class MybastisColumnsHandler implements MetaObjectHandler {
    @Resource
    private JwtTokenUtil jwtTokenUtil;  //我的工具类，用于从Token令牌中获取登陆人信息
    //设置数据新增时候的，字段自动赋值规则
    @Override
    public void insertFill(MetaObject metaObject) {
      this.strictInsertFill(metaObject, "createTime", LocalDateTime.class, LocalDateTime.now());
      this.strictInsertFill(metaObject, "updateTime", LocalDateTime.class, LocalDateTime.now());
      this.strictInsertFill(metaObject, "createBy", String.class, jwtTokenUtil.getUsernameFromToken());
      this.strictUpdateFill(metaObject, "updateBy", String.class, jwtTokenUtil.getUsernameFromToken());
    }
    //设置数据修改update时候的，字段自动赋值规则
    @Override
    public void updateFill(MetaObject metaObject) {
      this.strictUpdateFill(metaObject, "updateTime", LocalDateTime.class, LocalDateTime.now());
      this.strictUpdateFill(metaObject, "updateBy", String.class, jwtTokenUtil.getUsernameFromToken());
    }
}

在数据新增的时候，自动为createTime、updateTime、createBy、updateBy赋值，即数据初始化。

在数据修改的时候，自动为updateTime、updateBy赋值。

JwtTokenUtil是我写的一个工具类，从当前的登录用户JWT Token中获取当前登录用户的用户名。（你的系统里面获取当前登录用户名方法和我的不一样，但总之可以获取到）

五、实现效果

比如在进行数据更新的时候，下面的这两行代码就不需要写了，由updateFill(MetaObject metaObject)自动完成

//xxxYyyZzz.setUpdateBy("zimug");    //数据记录更新操作人
//xxxYyyZzz.setUpdateTime(new Date());   //数据记录更新操作的时间
xxxYyyZzzMapper.updateById(xxxYyyZzz);

同理，在数据insert操作的时候，insertFill(MetaObject metaObject)将被自动执行。

以上为个人经验，希望能给大家一个参考，也希望大家多多支持Devmax。