From c2614ad5c7d1e27fae68e52cd058af363464cdb4 Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期六, 21 五月 2022 23:16:38 +0800
Subject: [PATCH] 优化代码
---
java110-db/src/main/java/com/java110/db/Java110MybatisInterceptor.java | 188 +++++++++++++++++++++++++++++++++-------------
1 files changed, 134 insertions(+), 54 deletions(-)
diff --git a/java110-db/src/main/java/com/java110/db/Java110MybatisInterceptor.java b/java110-db/src/main/java/com/java110/db/Java110MybatisInterceptor.java
old mode 100644
new mode 100755
index f7da7c4..0f03572
--- a/java110-db/src/main/java/com/java110/db/Java110MybatisInterceptor.java
+++ b/java110-db/src/main/java/com/java110/db/Java110MybatisInterceptor.java
@@ -8,6 +8,7 @@
import com.java110.dto.order.OrderItemDto;
import com.java110.utils.constant.ServiceConstant;
import com.java110.utils.factory.ApplicationContextFactory;
+import com.java110.utils.util.DateUtil;
import com.java110.utils.util.StringUtil;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.mapping.BoundSql;
@@ -19,9 +20,10 @@
import org.apache.ibatis.session.Configuration;
import org.apache.ibatis.type.TypeHandlerRegistry;
import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
+import com.java110.core.log.LoggerFactory;
import org.springframework.http.*;
+import java.sql.Timestamp;
import java.text.DateFormat;
import java.util.*;
@@ -49,18 +51,19 @@
BoundSql boundSql = mappedStatement.getBoundSql(parameter);
Configuration configuration = mappedStatement.getConfiguration();
+ Map<String, Object> sqlValue = new HashMap<>();
//鑾峰彇sql璇彞
- String sql = showSql(configuration, boundSql);
+ String sql = showSql(configuration, boundSql, sqlValue, sqlCommandType);
restTemplate = ApplicationContextFactory.getBean("restTemplate", RestTemplate.class);
switch (sqlCommandType) {
case INSERT:
- dealInsertSql(mappedStatement, parameter, sql);
+ dealInsertSql(mappedStatement, parameter, sql, sqlValue);
break;
case UPDATE:
- dealUpdateSql(mappedStatement, parameter, sql);
+ dealUpdateSql(mappedStatement, parameter, sql, sqlValue);
break;
case DELETE:
- dealDeleteSql(mappedStatement, parameter, sql);
+ dealDeleteSql(mappedStatement, parameter, sql, sqlValue);
break;
}
return invocation.proceed();
@@ -72,20 +75,25 @@
* @param mappedStatement
* @param parameter
*/
- private void dealDeleteSql(MappedStatement mappedStatement, Object parameter, String sql) {
+ private void dealDeleteSql(MappedStatement mappedStatement, Object parameter, String sql, Map<String, Object> sqlValue) {
- String tmpTable = sql.substring(sql.indexOf("into") + 4, sql.indexOf("("));
+ String tmpTable = sql.substring(sql.indexOf("into") + 4, sql.indexOf("(")).trim();
+ String tmpTableHasT = tmpTable;
+ if(tmpTable.indexOf(" ") > 0){
+ tmpTable = tmpTable.substring(0,tmpTable.indexOf(" "));
+ }
String tmpWhere = sql.substring(sql.indexOf("where"));
//鎻掑叆鎿嶄綔鏃朵箣鍓嶇殑 娌℃湁鏁版嵁 鎵�浠� preValue 涓虹┖瀵硅薄
JSONArray preValues = new JSONArray();
- String execSql = "select * from " + tmpTable + " " + tmpWhere;
+ String execSql = "select * from " + tmpTableHasT + " " + tmpWhere;
queryServiceDAOImpl = ApplicationContextFactory.getBean("queryServiceDAOImpl", IQueryServiceDAO.class);
List<Map<String, Object>> deleteDatas = queryServiceDAOImpl.executeSql(execSql, null);
if (deleteDatas != null && deleteDatas.size() > 0) {
for (Map<String, Object> map : deleteDatas) {
+ dealReturnMap(map);
preValues.add(map);
}
}
@@ -120,47 +128,33 @@
* @param mappedStatement
* @param parameter
*/
- private void dealUpdateSql(MappedStatement mappedStatement, Object parameter, String sql) {
+ private void dealUpdateSql(MappedStatement mappedStatement, Object parameter, String sql, Map<String, Object> sqlValue) {
//RestTemplate restTemplate = ApplicationContextFactory.getBean("restTemplate", RestTemplate.class);
- String tmpTable = sql.substring(sql.indexOf("update") + 6, sql.indexOf("set"));
+ String tmpTable = sql.substring(sql.indexOf("update") + 6, sql.indexOf("set")).trim();
+
+ String tmpTableHasT = tmpTable;
+
+ if(tmpTable.indexOf(" ") > 0){
+ tmpTable = tmpTable.substring(0,tmpTable.indexOf(" "));
+ }
String tmpWhere = sql.substring(sql.indexOf("where"));
- String tmpKey = sql.substring(sql.indexOf("set") + 3, sql.indexOf("where"));
- String[] tmpString = tmpKey.split(",");
- Map tmpAfterMap = new HashMap();
- for (String tmp : tmpString) {
- String[] keyValues = tmp.split("=");
- String key = "";
- if (keyValues.length != 2) {
- throw new IllegalArgumentException("update 璇彞鍙兘鏈夐棶棰橈紝娌℃湁 set 涓嚭閿� " + sql);
- }
- if (keyValues[0].contains(".")) {
- key = keyValues[0].substring(keyValues[0].indexOf(".") + 1);
- } else {
- key = keyValues[0];
- }
- tmpAfterMap.put(key.trim(), keyValues[1].trim());
- }
-
- if (tmpString == null || tmpString.length < 1) {
- throw new IllegalArgumentException("update 璇彞鍙兘鏈夐棶棰橈紝娌℃湁 set 鍐呭 " + sql);
- }
//鎻掑叆鎿嶄綔鏃朵箣鍓嶇殑 娌℃湁鏁版嵁 鎵�浠� preValue 涓虹┖瀵硅薄
JSONArray preValues = new JSONArray();
JSONArray afterValues = new JSONArray();
JSONObject afterVaule = null;
- String execSql = "select * from " + tmpTable + " " + tmpWhere;
-
+ String execSql = "select * from " + tmpTableHasT + " " + tmpWhere;
queryServiceDAOImpl = ApplicationContextFactory.getBean("queryServiceDAOImpl", IQueryServiceDAO.class);
List<Map<String, Object>> deleteDatas = queryServiceDAOImpl.executeSql(execSql, null);
if (deleteDatas != null && deleteDatas.size() > 0) {
for (Map<String, Object> map : deleteDatas) {
+ dealReturnMap(map);
preValues.add(map);
afterVaule = new JSONObject();
afterVaule.putAll(map);
- afterVaule.putAll(tmpAfterMap);
+ afterVaule.putAll(sqlValue);
afterValues.add(afterVaule);
}
}
@@ -187,13 +181,37 @@
}
}
+ private void dealReturnMap(Map<String, Object> map) {
+ for (String key : map.keySet()) {
+ Object value = map.get(key);
+ if (value instanceof String) {
+ map.put(key, "'" + map.get(key) + "'");
+ } else if (value instanceof Date) {
+ String tmpValue = DateUtil.getFormatTimeString((Date) value, DateUtil.DATE_FORMATE_STRING_A);
+ map.put(key, "'" + tmpValue + "'");
+ } else if (value instanceof Timestamp) {
+ Date date = new Date(((Timestamp) value).getTime());
+ String tmpValue = DateUtil.getFormatTimeString(date, DateUtil.DATE_FORMATE_STRING_A);
+ map.put(key, "'" + tmpValue + "'");
+ } else if (value instanceof Double) {
+ map.put(key, "'" + map.get(key) + "'");
+ } else {
+ if (value != null) {
+ map.put(key, "'" + value.toString() + "'");
+ } else {
+ map.put(key, "''");
+ }
+ }
+ }
+ }
+
/**
* 澶勭悊insert 璇彞
*
* @param mappedStatement
* @param parameter
*/
- private void dealInsertSql(MappedStatement mappedStatement, Object parameter, String sql) {
+ private void dealInsertSql(MappedStatement mappedStatement, Object parameter, String sql, Map<String, Object> sqlValue) {
// RestTemplate restTemplate = ApplicationContextFactory.getBean("restTemplate", RestTemplate.class);
@@ -201,24 +219,14 @@
JSONArray preValues = new JSONArray();
JSONArray afterValues = new JSONArray();
- JSONObject afterValue = new JSONObject();
- String tmpTable = sql.substring(sql.toLowerCase().indexOf("into") + 4, sql.indexOf("("));
- String tmpKey = sql.substring(sql.indexOf("(") + 1, sql.indexOf(")"));
- String tmpValue = sql.substring(sql.lastIndexOf("(") + 1, sql.lastIndexOf(")"));
- String[] tmpKeys = tmpKey.split(",");
- String[] tmpValues = tmpValue.split(",");
- if (tmpKeys.length != tmpValues.length) {
- throw new IllegalArgumentException("sql 閿欒 key 鍜寁alue 涓暟涓嶇瓑" + sql);
+ String tmpTable = sql.substring(sql.toLowerCase().indexOf("into") + 4, sql.indexOf("(")).trim();
+
+ if(tmpTable.indexOf(" ") > 0){
+ tmpTable = tmpTable.substring(0,tmpTable.indexOf(" "));
}
- if (tmpKeys.length < 1) {
- throw new IllegalArgumentException("sql 閿欒 鏈壘鍒発ey" + sql);
- }
- for (int keyIndex = 0; keyIndex < tmpKeys.length; keyIndex++) {
- afterValue.put(tmpKeys[keyIndex], tmpValues[keyIndex]);
- }
- afterValues.add(afterValue);
+ afterValues.add(sqlValue);
JSONObject logText = new JSONObject();
logText.put("preValue", preValues);
@@ -243,6 +251,7 @@
}
+
@Override
public Object plugin(Object target) {
@@ -255,15 +264,18 @@
}
- public String showSql(Configuration configuration, BoundSql boundSql) {
+ public String showSql(Configuration configuration, BoundSql boundSql, Map<String, Object> sqlValue, SqlCommandType sqlCommandType) {
Object parameterObject = boundSql.getParameterObject();
List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();
- String sql = boundSql.getSql().replaceAll("[\\s]+", " ");
+ String sql = boundSql.getSql().replaceAll("[\\s]+", " ").toLowerCase();
+ String orgSql = sql;// 鍘熷sql
+
+ List<String> values = new ArrayList<>();
if (parameterMappings.size() > 0 && parameterObject != null) {
TypeHandlerRegistry typeHandlerRegistry = configuration.getTypeHandlerRegistry();
if (typeHandlerRegistry.hasTypeHandler(parameterObject.getClass())) {
sql = sql.replaceFirst("\\?", getParameterValue(parameterObject));
-
+ values.add(getParameterValue(parameterObject));
} else {
MetaObject metaObject = configuration.newMetaObject(parameterObject);
for (ParameterMapping parameterMapping : parameterMappings) {
@@ -271,15 +283,72 @@
if (metaObject.hasGetter(propertyName)) {
Object obj = metaObject.getValue(propertyName);
sql = sql.replaceFirst("\\?", getParameterValue(obj));
+ values.add(getParameterValue(obj));
} else if (boundSql.hasAdditionalParameter(propertyName)) {
Object obj = boundSql.getAdditionalParameter(propertyName);
sql = sql.replaceFirst("\\?", getParameterValue(obj));
+ values.add(getParameterValue(obj));
}
}
}
}
- return sql.toLowerCase();
+
+ if (sqlCommandType == SqlCommandType.INSERT) {
+ String tmpKey = orgSql.substring(orgSql.indexOf("(") + 1, orgSql.indexOf(")"));
+ String[] tmpKeys = tmpKey.split(",");
+
+// if (values.size() < tmpKeys.length) {
+// throw new IllegalArgumentException("sql 閿欒 key 鍜寁alue 涓暟涓嶇瓑" + sql);
+// }
+ for (int keyIndex = 0; keyIndex < tmpKeys.length; keyIndex++) {
+ String key = tmpKeys[keyIndex].trim();
+ String value = "";
+ if (values.size() - 1 < keyIndex) {
+ continue;
+ }
+ value = values.get(keyIndex);
+ if ("''".equals(value)) {
+ continue;
+ }
+ sqlValue.put(key, value);
+ }
+ } else if (sqlCommandType == SqlCommandType.UPDATE) {
+ String tmpKey = orgSql.substring(sql.indexOf("set") + 3, orgSql.indexOf("where"));
+ String[] tmpKeys = tmpKey.split(",");
+// if (values.size() < tmpKeys.length) {
+// throw new IllegalArgumentException("sql 閿欒 key 鍜寁alue 涓暟涓嶇瓑" + sql);
+// }
+ for (int keyIndex = 0; keyIndex < tmpKeys.length; keyIndex++) {
+ String tmpSetKey = tmpKeys[keyIndex];
+ String[] keyValues = tmpSetKey.split("=");
+ String key = "";
+ if (keyValues.length != 2) {
+ throw new IllegalArgumentException("update 璇彞鍙兘鏈夐棶棰橈紝娌℃湁 set 涓嚭閿� " + sql);
+ }
+ if (keyValues[0].contains(".")) {
+ key = keyValues[0].substring(keyValues[0].indexOf(".") + 1).trim();
+ } else {
+ key = keyValues[0].trim();
+ }
+ String value = "";
+ if (values.size() - 1 < keyIndex) {
+ continue;
+ }
+ value = values.get(keyIndex);
+ if ("''".equals(value)) {
+ continue;
+ }
+ sqlValue.put(key, value);
+ }
+
+ } else if (sqlCommandType == SqlCommandType.DELETE) {
+
+ }
+
+
+ return sql;
}
+
private String getParameterValue(Object obj) {
String value = null;
@@ -293,10 +362,21 @@
if (obj != null) {
value = obj.toString();
} else {
- value = "";
+ value = "''";
}
}
return value;
}
+
+
+ public static void main(String[] args) {
+ String tmpTable = "" +
+ " id_card,open_id,link,remark,user_id,app_type,app_user_name,nickname,headimgurl,community_name,state,app_user_id,community_id,app_type_cd,member_id\n" +
+ " ";
+ String[] a = tmpTable.split(",");
+ for(String a1:a){
+ System.out.println(a1);
+ }
+ }
}
--
Gitblit v1.8.0