之前想实现这个功能 用的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 里进行修改配置 里边的字符串需要和实体类的属性名对应