mybatis-plus框架的拓展包，在框架原有基础上做了进一步的轻度封装，增强内容

package com.tangzc.mpe.base; import com.baomidou.mybatisplus.annotation.FieldFill; import com.baomidou.mybatisplus.annotation.TableField; import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler; import com.tangzc.mpe.annotation.DefaultValue; import com.tangzc.mpe.annotation.FillData; import com.tangzc.mpe.annotation.FillTime; import com.tangzc.mpe.annotation.OptionDate; import com.tangzc.mpe.annotation.OptionUser; import com.tangzc.mpe.annotation.handler.AutoFillHandler; import com.tangzc.mpe.annotation.handler.FieldDateTypeHandler; import com.tangzc.mpe.magic.util.AnnotatedElementUtilsPlus; import com.tangzc.mpe.magic.util.SpringContextUtil; import lombok.extern.slf4j.Slf4j; import org.apache.ibatis.reflection.MetaObject; import org.springframework.beans.factory.NoSuchBeanDefinitionException; import org.springframework.beans.factory.NoUniqueBeanDefinitionException; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; import org.springframework.context.event.ContextRefreshedEvent; import org.springframework.context.event.EventListener; import org.springframework.core.annotation.AnnotatedElementUtils; import java.lang.reflect.Field; import java.math.BigDecimal; import java.text.ParseException; import java.text.SimpleDateFormat; import java.time.LocalDate; import java.time.LocalDateTime; import java.time.ZoneId; import java.time.format.DateTimeFormatter; import java.util.ArrayList; import java.util.Arrays; import java.util.Date; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.Objects; import java.util.Optional; import java.util.function.Function; import java.util.stream.Collectors; /** * 数据审计插件 * * @author don */ @Slf4j //@Component @ConditionalOnMissingBean(MetaObjectHandler.class) public class AutoFillMetaObjectHandler implements MetaObjectHandler { @Override public void insertFill(MetaObject metaObject) { optionFill(metaObject, FieldFill.INSERT); } @Override public void updateFill(MetaObject metaObject) { optionFill(metaObject, FieldFill.UPDATE); } private void optionFill(MetaObject metaObject, FieldFill option) { Object object = metaObject.getOriginalObject(); Class<?> clazz = object.getClass(); Field[] declaredFields = clazz.getDeclaredFields(); List<Field> fields = new ArrayList<>(Arrays.asList(declaredFields)); for (Class<?> superClass = clazz.getSuperclass(); superClass != null; superClass = superClass.getSuperclass()) { declaredFields = superClass.getDeclaredFields(); fields.addAll(Arrays.asList(declaredFields)); } List<Field> fieldList = fields.stream() .filter(field -> metaObject.hasSetter(field.getName())) .filter(field -> { TableField annotation = AnnotatedElementUtilsPlus.findDeepMergedAnnotation(field, TableField.class); return annotation != null && (annotation.fill() == option || annotation.fill() == FieldFill.INSERT_UPDATE); }) .collect(Collectors.toList()); fill(metaObject, object, clazz, fieldList); } public void fill(MetaObject metaObject, Object object, Class<?> clazz, List<Field> fieldList) { Now now = new Now(); fieldList.forEach(field -> { // 默认值 setDefaultVale(metaObject, field); // 操作人 setOptionUser(metaObject, object, clazz, field); // 操作时间 setOptionDate(metaObject, clazz, field, now); }); } private void setDefaultVale(MetaObject metaObject, Field field) { DefaultValue defaultValue = AnnotatedElementUtils.getMergedAnnotation(field, DefaultValue.class); if (defaultValue != null) { boolean canSet = this.getFieldValByName(field.getName(), metaObject) == null; if (canSet) { Object newVal = convert(field, defaultValue); this.setFieldValByName(field.getName(), newVal, metaObject); } } } private void setOptionUser(MetaObject metaObject, Object object, Class<?> clazz, Field field) { Boolean override = null; Class<? extends AutoFillHandler> value = null; FillData fillData = AnnotatedElementUtils.getMergedAnnotation(field, FillData.class); if (fillData != null) { override = fillData.override(); value = fillData.value(); } else { // 尝试获取旧版注解 OptionUser optionUser = AnnotatedElementUtils.getMergedAnnotation(field, OptionUser.class); if (optionUser != null) { override = optionUser.override(); value = optionUser.value(); } } if (override != null && value != null) { // 判断原来值为null，或者覆盖选项为true boolean canSet = this.getFieldValByName(field.getName(), metaObject) == null || override; if (canSet) { Object userInfo = null; AutoFillHandler instance = getAutoFillHandler(value); if (instance != null) { userInfo = instance.getVal(object, clazz, field); } // 如果当前未取到信息，不设置 if (userInfo != null) { try { // 赋值 this.setFieldValByName(field.getName(), userInfo, metaObject); } catch (Exception e) { String errorMsg = "数据填充赋值失败，可能是由于" + clazz.getName() + "中的字段" + field.getName() + "的类型（" + field.getType() + "）与" + instance.getClass() + "返回值的类型（" + userInfo.getClass() + "）不一致"; throw new RuntimeException(errorMsg); } } } } } /** * 缓存AutoFillHandler，同时寻找 */ private AutoFillHandler getAutoFillHandler(Class<? extends AutoFillHandler> autoFillHandler) { try { return SpringContextUtil.getBeanOfType(autoFillHandler); } catch (NoUniqueBeanDefinitionException ignore) { throw new RuntimeException("发现了多个" + autoFillHandler.getName() + "的实现，请保持spring中只有一个实例。"); } catch (NoSuchBeanDefinitionException ignore) { if (autoFillHandler.isInterface()) { log.warn("没有找到{}的实现，操作人信息无法自动填充。", autoFillHandler.getName()); } else { log.warn("{}需要注册到spring，不然操作人信息无法自动填充。", autoFillHandler.getName()); } } return null; } private void setOptionDate(MetaObject metaObject, Class<?> clazz, Field field, Now now) { Boolean override = null; String format = null; FillTime fillTime = AnnotatedElementUtils.getMergedAnnotation(field, FillTime.class); if (fillTime != null) { override = fillTime.override(); format = fillTime.format(); } else { // 尝试旧版的注解 OptionDate optionDate = AnnotatedElementUtils.getMergedAnnotation(field, OptionDate.class); if (optionDate != null) { override = optionDate.override(); format = optionDate.format(); } } if (override != null && format != null) { boolean canSet = this.getFieldValByName(field.getName(), metaObject) == null || override; if (canSet) { Class<?> type = getDateType(clazz, field); Object nowDate = Optional.ofNullabl