【MyBatis-Plus】實現字段自動填充功能

在項目中，我們有一些公共的字段需要做修改
如：

1. gmt_create：創建時間
2. creator_id：創建人
3. gmt_modified：修改時間
4. modifier_id：修改人

這時候我們可以采用 MyBatis-Plus 中的字段自動填充功能去實現

思路：抽取公用字段封裝到BaseEntity類中，再將使用到此公共字段的類繼承基類，最后由 MyBatis-Plus 幫我們實現自動填充，這樣我們便可以在service服務類中減少一定代碼重復量！

二、實現

1. 實體類

@Data
@TableName("t_user")
public class User extends BaseEntity<User> {

    private static final long serialVersionUID = 1L;

    @TableId(value = "id", type = IdType.AUTO)
    private Integer id;
 
    @TableField("username")
    private String username;
    
    @Override
    protected Serializable pkVal() {
        return this.id;
    }
}

2. 公用字段 - 使用注解填充字段，如：@TableField(fill = FieldFill.INSERT)

@Getter
@Setter
public abstract class BaseEntity<T extends Model> extends Model {
    /**
     * 創建日期 - 現在時表示主動創建
     */
    @TableField(value = "gmt_create", fill = FieldFill.INSERT)
    private Date gmtCreate;
    /**
     * 修改時間 - 過去分詞表示被動更新
     */
    @TableField(value = "gmt_modified", fill = FieldFill.INSERT_UPDATE)
    private Date gmtModified;
}

3. 自定義MyMetaObjectHandler字段自動填充處理類繼承MetaObjectHandler

/**
 *  <p> MyBatisPlus自定義字段自動填充處理類 - 實體類中使用 @TableField注解 </p>
 *
 * @description: 注意前端傳值時要為null
 * @author: zhengqing
 * @date: 2019/8/18 0018 1:46
 */
@Component
public class MyMetaObjectHandler extends MetaObjectHandler {

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

    /**
     * 創建時間
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        LOG.info(" -------------------- start insert fill ...  --------------------");
        if (metaObject.hasGetter("gmtCreate") && metaObject.hasGetter("gmtModified")) {
            setFieldValByName("gmtCreate", new Date(), metaObject);
            setFieldValByName("gmtModified", new Date(), metaObject);
        }
    }

    /**
     * 最后一次更新時間
     */
    @Override
    public void updateFill(MetaObject metaObject) {
        LOG.info(" -------------------- start update fill ...  --------------------");
        if (metaObject.hasGetter("et.gmtModified")) {
            setFieldValByName("gmtModified", new Date(), metaObject);
        }
    }

mybatis-plus以前的填充策略是：

public enum FieldFill {
    DEFAULT(0, "默認不處理"),
    INSERT(1, "插入填充字段"),
    UPDATE(2, "更新填充字段"),
    INSERT_UPDATE(3, "插入和更新填充字段");
    /**
     * 主鍵
     */
    private final int key;
    /**
     * 描述
     */
    private final String desc;
    FieldFill(final int key, final String desc) {
        this.key = key;
        this.desc = desc;
    }
    public static FieldFill getIgnore(int key) {
        FieldFill[] fis = FieldFill.values();
        for (FieldFill fi : fis) {
            if (fi.getKey() == key) {
                return fi;
            }
        }
        return FieldFill.DEFAULT;
    }
    public int getKey() {
        return this.key;
    }
    public String getDesc() {
        return this.desc;
    }
}

現在只能看到枚舉類型。