mybatis-plus公共字段自动填充

1.对mybaits-plus自动生成代码模板有疑问的同学,请看我的另一篇博客

https://blog.csdn.net/qq_26182739

2.这里讲解怎么结合自动策略配置,自动填充类似于update_time,update_name,update_id这些公共字段,每次更新或者插入的 时候不用我们再进行同样的赋值操作

在自定义配置模板中加入如下代码:

// 策略配置
StrategyConfig strategy = new StrategyConfig();
// 表的自动填充字段
List tableFills = new ArrayList<>();
TableFill updateTime = new TableFill("update_time", FieldFill.INSERT_UPDATE);
TableFill updateName = new TableFill("update_name", FieldFill.INSERT_UPDATE);
TableFill updateId = new TableFill("update_id", FieldFill.INSERT_UPDATE);
tableFills.add(updateTime);
tableFills.add(updateName);
tableFills.add(updateId);
strategy.setTableFillList(tableFills);

以上代码可以在生成entity的时候自动配置 属性fill

属性fill选择对应策略,该申明告知 Mybatis-Plus 需要预留注入 SQL 字段,生成的entity的字段配置如下所示:

@ApiModelProperty(value = "更新人")
@TableField(value = "update_id", fill = FieldFill.INSERT_UPDATE)
private String updateId;

@ApiModelProperty(value = "更新人名称")
@TableField(value = "update_name", fill = FieldFill.INSERT_UPDATE)
private String updateName;

@ApiModelProperty(value = "更新时间")
@TableField(value = "update_time", fill = FieldFill.INSERT_UPDATE)
private Date updateTime;

 FieldFill的对应策略

public enum FieldFill {
    /**
     * 默认不处理
     */
    DEFAULT,
    /**
     * 插入填充字段
     */
    INSERT,
    /**
     * 更新填充字段
     */
    UPDATE,
    /**
     * 插入和更新填充字段
     */
    INSERT_UPDATE
}

然后按照官网的方法自定义实现类 MyMetaObjectHandler

注意:加入get set的判断,避免某些表只有create相关字段,没有update相关字段时出现如下异常:

 org.apache.ibatis.reflection.ReflectionException: There is no getter for property named 'updateTime'

@Component
public class MetaObjectHandlerConfig implements MetaObjectHandler {
    @Override
    public void insertFill(MetaObject metaObject) {
        Object createTime = null;
        Object updateTime = null;
        if(metaObject.hasGetter("createTime") ){
            createTime = getFieldValByName("createTime", metaObject);
        }
        if(metaObject.hasGetter("updateTime")){
            updateTime = getFieldValByName("updateTime", metaObject);
        }
        Object id = getFieldValByName("id", metaObject);
        Date now = new Date();
        if (createTime == null && metaObject.hasSetter("createTime")){
            setFieldValByName("createTime",now, metaObject);
        }
        if (updateTime == null && metaObject.hasSetter("updateTime")){
            setFieldValByName("updateTime",now, metaObject);
        }
        if (id == null){
            setFieldValByName("id", SnowFlakeIDGenerator.getInstance().nextId(), metaObject);
        }
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        if(metaObject.hasSetter("updateTime")){
            setFieldValByName("updateTime", new Date(), metaObject);
        }
    }
}

 赶紧用起来吧!

你可能感兴趣的:(Springboot)