package com.xcong.excoin.configurations.interceptor;
|
|
import com.xcong.excoin.common.LoginUserUtils;
|
import com.xcong.excoin.modules.member.entity.MemberEntity;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.ibatis.executor.Executor;
|
import org.apache.ibatis.mapping.MappedStatement;
|
import org.apache.ibatis.mapping.SqlCommandType;
|
import org.apache.ibatis.plugin.*;
|
import org.springframework.stereotype.Component;
|
|
import java.lang.reflect.Field;
|
import java.util.*;
|
|
/**
|
* mybatis拦截器，自动注入创建人、创建时间、修改人、修改时间
|
*
|
* @author wzy
|
* @date 2020-05-13
|
**/
|
@Slf4j
|
//@Component
|
@Intercepts({ @Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class})})
|
public class MybatisInterceptor implements Interceptor {
|
@Override
|
public Object intercept(Invocation invocation) throws Throwable {
|
MappedStatement mappedStatement = (MappedStatement) invocation.getArgs()[0];
|
String sqlId = mappedStatement.getId();
|
log.info("----sqlId----" + sqlId);
|
SqlCommandType sqlCommandType = mappedStatement.getSqlCommandType();
|
log.info("-------------->{}", sqlCommandType);
|
|
Object parameter = invocation.getArgs()[1];
|
if (parameter == null) {
|
return invocation.proceed();
|
}
|
|
MemberEntity memberEntity = LoginUserUtils.getAppLoginUser();
|
if (SqlCommandType.INSERT == sqlCommandType) {
|
Field[] fields = this.getAllFields(parameter);
|
|
for(Field field : fields) {
|
log.info("----->{}", field.getName());
|
if ("createBy".equals(field.getName()) || "updateBy".equals(field.getName())) {
|
byField(field, parameter, memberEntity);
|
}
|
|
if ("createTime".equals(field.getName()) || "updateTime".equals(field.getName())) {
|
timeField(field, parameter, memberEntity);
|
}
|
}
|
}
|
|
if (SqlCommandType.UPDATE == sqlCommandType) {
|
Field[] fields = this.getAllFields(parameter);
|
|
for (Field field : fields) {
|
if ("updateBy".equals(field.getName())) {
|
byField(field, parameter, memberEntity);
|
}
|
|
if ("updateTime".equals(field.getName())) {
|
timeField(field, parameter, memberEntity);
|
}
|
}
|
}
|
|
return invocation.proceed();
|
}
|
|
@Override
|
public Object plugin(Object o) {
|
return Plugin.wrap(o, this);
|
}
|
|
@Override
|
public void setProperties(Properties properties) {
|
|
}
|
|
private Field[] getAllFields(Object object) {
|
Class<?> clazz = object.getClass();
|
List<Field> fieldList = new ArrayList<>();
|
while (clazz != null) {
|
fieldList.addAll(new ArrayList<>(Arrays.asList(clazz.getDeclaredFields())));
|
clazz = clazz.getSuperclass();
|
}
|
Field[] fields = new Field[fieldList.size()];
|
fieldList.toArray(fields);
|
return fields;
|
}
|
|
private void byField(Field field, Object parameter, MemberEntity memberEntity) throws IllegalAccessException {
|
field.setAccessible(true);
|
Object local = field.get(parameter);
|
field.setAccessible(false);
|
if (local != null) {
|
field.setAccessible(true);
|
field.set(parameter, memberEntity.getUsername());
|
field.setAccessible(false);
|
}
|
}
|
|
private void timeField(Field field, Object parameter, MemberEntity memberEntity) throws IllegalAccessException {
|
field.setAccessible(true);
|
Object local_createBy = field.get(parameter);
|
field.setAccessible(false);
|
if (local_createBy != null) {
|
field.setAccessible(true);
|
field.set(parameter, new Date());
|
field.setAccessible(false);
|
}
|
}
|
}
|
