From 0928200f9ca991957f129cc9c1a77c025f4186d0 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Thu, 14 May 2020 17:31:14 +0800
Subject: [PATCH] add sql log and mybatis interceptor
---
src/main/java/com/xcong/excoin/configurations/interceptor/MybatisInterceptor.java | 80 +++++++++++++++------------------------
1 files changed, 31 insertions(+), 49 deletions(-)
diff --git a/src/main/java/com/xcong/excoin/configurations/interceptor/MybatisInterceptor.java b/src/main/java/com/xcong/excoin/configurations/interceptor/MybatisInterceptor.java
index 9203353..144a8c0 100644
--- a/src/main/java/com/xcong/excoin/configurations/interceptor/MybatisInterceptor.java
+++ b/src/main/java/com/xcong/excoin/configurations/interceptor/MybatisInterceptor.java
@@ -1,12 +1,14 @@
package com.xcong.excoin.configurations.interceptor;
import com.xcong.excoin.common.LoginUserUtils;
+import com.xcong.excoin.common.system.base.BaseEntity;
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.apache.ibatis.session.defaults.DefaultSqlSession;
import org.springframework.stereotype.Component;
import java.lang.reflect.Field;
@@ -19,7 +21,7 @@
* @date 2020-05-13
**/
@Slf4j
-//@Component
+@Component
@Intercepts({ @Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class})})
public class MybatisInterceptor implements Interceptor {
@Override
@@ -35,33 +37,27 @@
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 (parameter instanceof DefaultSqlSession.StrictMap) {
+ Map map = (Map) parameter;
+ List list = (List) map.get("list");
+ for (Object o : list) {
+ injectForInsert(o);
}
-
- if ("createTime".equals(field.getName()) || "updateTime".equals(field.getName())) {
- timeField(field, parameter, memberEntity);
- }
+ } else {
+ injectForInsert(parameter);
}
}
if (SqlCommandType.UPDATE == sqlCommandType) {
- Field[] fields = this.getAllFields(parameter);
-
- for (Field field : fields) {
- if ("updateBy".equals(field.getName())) {
- byField(field, parameter, memberEntity);
+ if (parameter instanceof DefaultSqlSession.StrictMap) {
+ Map map = (Map) parameter;
+ List list = (List) map.get("list");
+ for (Object o : list) {
+ injectForUpdate(o);
}
-
- if ("updateTime".equals(field.getName())) {
- timeField(field, parameter, memberEntity);
- }
+ } else {
+ injectForUpdate(parameter);
}
}
@@ -78,37 +74,23 @@
}
- 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);
+ public void injectForInsert(Object o) {
+ MemberEntity member = LoginUserUtils.getAppLoginUser();
+ if (o instanceof BaseEntity) {
+ BaseEntity baseEntity = (BaseEntity) o;
+ baseEntity.setCreateBy(member.getUsername());
+ baseEntity.setUpdateBy(member.getUsername());
+ baseEntity.setCreateTime(new Date());
+ baseEntity.setUpdateTime(new Date());
}
}
- 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);
+ public void injectForUpdate(Object o) {
+ MemberEntity member = LoginUserUtils.getAppLoginUser();
+ if (o instanceof BaseEntity) {
+ BaseEntity baseEntity = (BaseEntity) o;
+ baseEntity.setUpdateBy(member.getUsername());
+ baseEntity.setUpdateTime(new Date());
}
}
}
--
Gitblit v1.9.1