From 8a2a463d46b3dce73015462c9f6a2a258dd25f5b Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期五, 25 二月 2022 11:40:11 +0800
Subject: [PATCH] 优化diamante
---
java110-db/src/main/java/com/java110/db/Java110MybatisInterceptor.java | 176 +++++++++++++++++++++++++++++++++-------------------------
1 files changed, 99 insertions(+), 77 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 cdd5fd1..0f03572
--- a/java110-db/src/main/java/com/java110/db/Java110MybatisInterceptor.java
+++ b/java110-db/src/main/java/com/java110/db/Java110MybatisInterceptor.java
@@ -20,7 +20,7 @@
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;
@@ -51,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();
@@ -74,14 +75,18 @@
* @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);
@@ -123,38 +128,23 @@
* @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);
@@ -164,7 +154,7 @@
preValues.add(map);
afterVaule = new JSONObject();
afterVaule.putAll(map);
- afterVaule.putAll(tmpAfterMap);
+ afterVaule.putAll(sqlValue);
afterValues.add(afterVaule);
}
}
@@ -221,7 +211,7 @@
* @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);
@@ -230,41 +220,13 @@
JSONArray afterValues = new JSONArray();
- String tmpTable = sql.substring(sql.toLowerCase().indexOf("into") + 4, sql.indexOf("("));
- 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 tmpTable = sql.substring(sql.toLowerCase().indexOf("into") + 4, sql.indexOf("(")).trim();
- 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]);
- }
- afterValues.add(afterValue);
+ if(tmpTable.indexOf(" ") > 0){
+ tmpTable = tmpTable.substring(0,tmpTable.indexOf(" "));
}
+ afterValues.add(sqlValue);
JSONObject logText = new JSONObject();
logText.put("preValue", preValues);
@@ -289,6 +251,7 @@
}
+
@Override
public Object plugin(Object target) {
@@ -301,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) {
@@ -317,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;
@@ -348,13 +371,12 @@
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 涓暟涓嶇瓑");
+ 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