转载标明出处~~
mybatis-plus的自动填充功能不做过多介绍(官网传送门 ==>),不过官网给出的例子是根据createTime 、updateTime 等固定字段名去判断填充的。有一定的局限性,在某些情况下,却不太适用,例如如下情况:
A.java
/**
* 创建时间
*/
@TableField(value = "aasy04a040",fill=FieldFill.INSERT)
@ApiModelProperty(value = "创建时间")
private LocalDateTime acreateTime;
/**
* 更新时间
*/
@TableField(value = "aasy04a050",fill=FieldFill.INSERT_UPDATE)
@ApiModelProperty(value = "更新时间")
private LocalDateTime aupdateTime;
B.java
/**
* 创建时间
*/
@ApiModelProperty(value = "创建时间")
@TableField(value = "bbsy08a100",fill =FieldFill.INSERT)
private Date bcreateTime;
/**
* 更新时间
*/
@ApiModelProperty(value = "更新时间")
@TableField(value = "bbsy08a110",fill = FieldFill.INSERT_UPDATE)
private Date bupdateTime;
A和B类都有创建时间和更新时间字段, 但字段名却不相同,肯定自动填充失效啊。。。。。
看了一下午代码之后(瞌睡了俩小时***),想出了如下方法,直接上代码!!!
/**
* @Author ZhaoJianTao
* @Description 根据注解自动填充值
* @CreationDate 2021/3/6
*/
@Slf4j
@Component
public class MyMetaObjectHandler implements MetaObjectHandler {
@Override
public void insertFill(MetaObject metaObject) {
log.info("start insert fill ....");
Class<?> originalClass = metaObject.getOriginalObject().getClass();
Field[] field = originalClass.getDeclaredFields();
for (int i = 1; i < field.length; i++) {
String fieldName=field[i].getName();
String type = field[i].getType().getName();
TableField ann = field[i].getAnnotation(TableField.class);
if(ann!=null && (FieldFill.INSERT_UPDATE.equals(ann.fill()) || FieldFill.INSERT.equals(ann.fill()))){
if("java.util.Date".equals(type)){
this.strictInsertFill(metaObject, fieldName, Date.class, new Date());
}else if("java.time.LocalDateTime".equals(type)){
this.strictInsertFill(metaObject, fieldName, LocalDateTime.class, LocalDateTime.now());
}
}
}
}
@Override
public void updateFill(MetaObject metaObject) {
log.info("start update fill ....");
Class<?> originalClass = metaObject.getOriginalObject().getClass();
Field[] field = originalClass.getDeclaredFields();
for (int i = 1; i < field.length; i++) {
String fieldName=field[i].getName();
String type = field[i].getType().getName();
TableField ann = field[i].getAnnotation(TableField.class);
if(ann!=null && (FieldFill.INSERT_UPDATE.equals(ann.fill()) || FieldFill.UPDATE.equals(ann.fill()))){
if("java.util.Date".equals(type)){
this.strictUpdateFill(metaObject, fieldName, Date.class, new Date());
}else if("java.time.LocalDateTime".equals(type)){
this.strictUpdateFill(metaObject, fieldName, LocalDateTime.class, LocalDateTime.now());
}
}
}
}
}
代码可能还需要优化,但能解决根据固定字段名去填值的限制性,例子中判断的是时间类型,也可以做其他扩充,如有更好的方法望不吝赐教!!!
一个具有注脚的文本。