自定义mybatis-plus的自动填充功能,根据注解填值

转载标明出处~~

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());
				}
			}
		}
	}
}

代码可能还需要优化，但能解决根据固定字段名去填值的限制性，例子中判断的是时间类型，也可以做其他扩充，如有更好的方法望不吝赐教！！！
一个具有注脚的文本。