当前位置：首页 > news >正文

mybatis-plus公共字段自动填充fillStrategy()方法和strictFill()方法

news 2024/9/25 16:37:06

前言：在mybatis-plus的公共字段自动填充功能中有多种填充方式，以下将通过源码对各种填充方法进行剖析

相信了解mybatis-plus公共字段自动填充功能的同学对MetaObjectHandler接口并不陌生，我们把MetaObjectHandler接口整理一下，只关注他的注入字段部分（结论在下面）

public interface MetaObjectHandler {default MetaObjectHandler setFieldValByName(String fieldName, Object fieldVal, MetaObject metaObject) {if (Objects.nonNull(fieldVal) && metaObject.hasSetter(fieldName)) {metaObject.setValue(fieldName, fieldVal);}return this;}default MetaObjectHandler strictInsertFill(TableInfo tableInfo, MetaObject metaObject, List<StrictFill<?, ?>> strictFills) {return strictFill(true, tableInfo, metaObject, strictFills);}default MetaObjectHandler strictUpdateFill(TableInfo tableInfo, MetaObject metaObject, List<StrictFill<?, ?>> strictFills) {return strictFill(false, tableInfo, metaObject, strictFills);}default MetaObjectHandler strictFill(boolean insertFill, TableInfo tableInfo, MetaObject metaObject, List<StrictFill<?, ?>> strictFills) {if ((insertFill && tableInfo.isWithInsertFill()) || (!insertFill && tableInfo.isWithUpdateFill())) {strictFills.forEach(i -> {final String fieldName = i.getFieldName();final Class<?> fieldType = i.getFieldType();tableInfo.getFieldList().stream().filter(j -> j.getProperty().equals(fieldName) && fieldType.equals(j.getPropertyType()) &&((insertFill && j.isWithInsertFill()) || (!insertFill && j.isWithUpdateFill()))).findFirst().ifPresent(j -> strictFillStrategy(metaObject, fieldName, i.getFieldVal()));});}return this;}default MetaObjectHandler fillStrategy(MetaObject metaObject, String fieldName, Object fieldVal) {if (getFieldValByName(fieldName, metaObject) == null) {setFieldValByName(fieldName, fieldVal, metaObject);}return this;}default MetaObjectHandler strictFillStrategy(MetaObject metaObject, String fieldName, Supplier<?> fieldVal) {if (metaObject.getValue(fieldName) == null) {Object obj = fieldVal.get();if (Objects.nonNull(obj)) {metaObject.setValue(fieldName, obj);}}return this;}
}

此时有6个方法可以进行字段填充

setFieldValByName()
strictInsertFill()
strictUpdateFill()
strictFill()
fillStrategy()
strictFillStrategy()

我们逐个进行分析

~~setFieldValByName()~~
当要注入的值fieldVal不为空，则注入，被fillStrategy()方法代替
~~strictInsertFill()~~
~~strictUpdateFill()~~
以上两个方法都是直接封装后调用strictFille()，可以忽略
strictFill()
校验实体表里的字段与要注入的公共字段的数据的名称和类型是否一致，再调用strictFillStrategy()方法进行填充
fillStrategy()
在直接调用setFieldValByName()的基础上，做了一层安全校验
若元数据的该字段为null，且fieldVal不为空才注入
~~strictFillStrategy()~~
若元数据的该字段为null，且fieldVal不为空才注入，被strictFill()方法代替

由于fillStrategy()和strictFillStrategy()比较相似，可以一下比较这两个方法

default MetaObjectHandler fillStrategy(MetaObject metaObject, String fieldName, Object fieldVal) {//把getFieldValByName()和setFieldValByName()方法体内的代码进行替换并简化if (metaObject.getValue(fieldName) == null) {if (Objects.nonNull(fieldVal)) {metaObject.setValue(fieldName, fieldVal);}}return this;
}default MetaObjectHandler strictFillStrategy(MetaObject metaObject, String fieldName, Supplier<?> fieldVal) {if (metaObject.getValue(fieldName) == null) {Object obj = fieldVal.get();//不同点if (Objects.nonNull(obj)) {metaObject.setValue(fieldName, obj);}}return this;
}