From a8990150b50fde2e507b9df01d8e0847b2bda81d Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期日, 22 八月 2021 19:05:06 +0800
Subject: [PATCH] 优化代码
---
java110-db/src/main/java/com/java110/db/Java110MybatisInterceptor.java | 198 +++++++++++++++++++++++++++++++++++++------------
1 files changed, 148 insertions(+), 50 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..ad15598
--- 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;
@@ -22,6 +23,7 @@
import org.slf4j.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,7 +75,7 @@
* @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 tmpWhere = sql.substring(sql.indexOf("where"));
@@ -86,6 +89,7 @@
if (deleteDatas != null && deleteDatas.size() > 0) {
for (Map<String, Object> map : deleteDatas) {
+ dealReturnMap(map);
preValues.add(map);
}
}
@@ -120,47 +124,27 @@
* @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 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;
-
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 +171,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 +209,43 @@
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(",");
+// String tmpKey = sql.substring(sql.indexOf("(") + 1, sql.indexOf(")"));
+// String[] tmpKeys = tmpKey.split(",");
+// int valuePos = 0;
+// if (sql.contains("VALUES")) {
+// valuePos = sql.indexOf("VALUES") + 6;
+// } else {
+// valuePos = sql.indexOf("values") + 6;
+// }
+// String sqlValues = sql.substring(valuePos);
+// //璇存槑鎵规搷浣�
+//
+// String[] sqlVauleses = sqlValues.split("\\)");
+// JSONObject afterValue = null;
+// for (String sqlV : sqlVauleses) {
+// String tmpValue = sqlV.substring(sqlV.lastIndexOf("(") + 1);
+// String[] tmpValues = tmpValue.split(",");
+// afterValue = new JSONObject();
+//
+// if (tmpKeys.length != tmpValues.length) {
+// throw new IllegalArgumentException("sql 閿欒 key 鍜寁alue 涓暟涓嶇瓑" + sql);
+// }
+//
+// if (tmpKeys.length < 1) {
+// throw new IllegalArgumentException("sql 閿欒 鏈壘鍒発ey" + sql);
+// }
+// for (int keyIndex = 0; keyIndex < tmpKeys.length; keyIndex++) {
+// if ("''".equals(tmpValues[keyIndex])) {
+// continue;
+// }
+// afterValue.put(tmpKeys[keyIndex], tmpValues[keyIndex]);
+// }
+//
+// }
- if (tmpKeys.length != tmpValues.length) {
- throw new IllegalArgumentException("sql 閿欒 key 鍜寁alue 涓暟涓嶇瓑" + sql);
- }
-
- 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);
@@ -255,15 +282,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,14 +301,70 @@
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) {
@@ -293,10 +379,22 @@
if (obj != null) {
value = obj.toString();
} else {
- value = "";
+ value = "''";
}
}
return value;
}
+
+
+ public static void main(String[] args) {
+ String tmpKey = " prime_rate,detail_id,receivable_amount,cycles,remark,status_cd,received_amount,community_id,b_id,fee_id,state";
+ String tmpValue = "'1.00','912020080411040001','1500.0','1.0',,'0','1500.0','7020181217000001','-1',,";
+ String[] tmpKeys = tmpKey.split(",");
+ String[] tmpValues = tmpValue.split(",");
+
+ if (tmpKeys.length != tmpValues.length) {
+ throw new IllegalArgumentException("sql 閿欒 key 鍜寁alue 涓暟涓嶇瓑");
+ }
+ }
}
--
Gitblit v1.8.0