ruoyi-vue 后端添加公共字段填充功能

之前想实现这个功能 用的MetaObjectHandler 全部配置完成 不知道为什么不生效  所以改用Interceptor来实现这个功能 

上代码

/**
 * 公共字段自动填充拦截器 mybatis插件
 *
 * updateTime 更新时间
 * updateId 更新人ID
 * createTime 创建时间
 * createId 创建人ID
 *
 */
@Slf4j
@Component
@Intercepts({ @Signature(type = Executor.class, method = "update", args = { MappedStatement.class, Object.class }) })
public class FieldFillInterceptor implements Interceptor {

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        Object[] args = invocation.getArgs();
        MappedStatement ms = (MappedStatement) args[0];
        SqlCommandType sqlCommandType = ms.getSqlCommandType();

        if (SqlCommandType.INSERT.equals(sqlCommandType)) {
            Object entity = args[1];
            Class<?> entityClass = entity.getClass();
            // 使用反射检查实体类是否包含指定的属性
            /**
             * createTime 创建时间
             * createId 创建人ID
             */
            if (hasProperty(entityClass, "createTime")) {
                entityClass.getMethod("setCreateTime", LocalDateTime.class).invoke(entity, LocalDateTime.now());
            }
            if (hasProperty(entityClass, "createId")) {
                entityClass.getMethod("setCreateId", Long.class).invoke(entity, SecurityUtils.getUserId());
            }
        } else if (SqlCommandType.UPDATE.equals(sqlCommandType)) {
            Object entity = args[1];
            Class<?> entityClass = entity.getClass();
            // 使用反射检查实体类是否包含指定的属性
            /**
             * updateTime 更新时间
             * updateId 更新人ID
             */
            if (hasProperty(entityClass, "updateTime")) {
                entityClass.getMethod("setUpdateTime", LocalDateTime.class).invoke(entity, LocalDateTime.now());
            }
            if (hasProperty(entityClass, "updateId")) {
                entityClass.getMethod("setUpdateId", Long.class).invoke(entity, SecurityUtils.getUserId());
            }
        }

        return invocation.proceed();
    }

    private boolean hasProperty(Class<?> entityClass, String propertyName) {
        try {
            entityClass.getDeclaredField(propertyName);
            return true;
        } catch (NoSuchFieldException e) {
            return false;
        }
    }
}

还要再mybatis-config.xml文件里添加这个插件

<configuration>

    <plugins>
        <!-- 公共字段自动填充插件 -->
        <plugin interceptor="com.hoshine.framework.interceptor.FieldFillInterceptor"/>
    </plugins>

</configuration>

到这里就完成了 这里的公共字段填充的属性可以根据需要在FieldFillInterceptor 里进行修改配置 里边的字符串需要和实体类的属性名对应