From de93953303db3a8cadcde8f09289a81e6576d7b8 Mon Sep 17 00:00:00 2001
From: 王彬 <wangbin@taover.com>
Date: Fri, 25 Dec 2020 19:18:54 +0800
Subject: [PATCH] fix a bug about core sql split

---
 src/main/java/com/taover/repository/UtilsSql.java | 113 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++--------------------------------------------
 1 file changed, 69 insertions(+), 44 deletions(-)

diff --git a/src/main/java/com/taover/repository/UtilsSql.java b/src/main/java/com/taover/repository/UtilsSql.java
index 154da98..79d4d78 100644
--- a/src/main/java/com/taover/repository/UtilsSql.java
+++ b/src/main/java/com/taover/repository/UtilsSql.java
@@ -2,6 +2,7 @@ package com.taover.repository;
 
 import java.util.HashMap;
 import java.util.Map;
+import java.util.Stack;
 
 import org.springframework.util.StringUtils;
 
@@ -9,62 +10,80 @@ public class UtilsSql {
 	
 	public static String[] splitCoreSql(String coreSql) throws Exception {		
 		//去除''内的信息
-		String coreSqlRemoveQuate = replaceSelectAndFromBetweenTopCornerMark(coreSql.toUpperCase());
-		int fromIndex = calcFromIndex(coreSqlRemoveQuate);
+		int fromIndex = calcFromIndex(coreSql);
 		if(fromIndex > -1) {
 			return new String[] {coreSql.substring(0, fromIndex), coreSql.substring(fromIndex, coreSql.length())};
 		}else {
 			throw new Exception("未找到FROM子句"); 
 		}
 	}
-	
-	private static int calcFromIndex(String coreSqlUpper) {
-		//计算位置
-		int selectSubSqlNum = 0;
-		int currWindowIndex = 0;		
-		for(currWindowIndex=0; (currWindowIndex+6)<coreSqlUpper.length(); ) {
-			String currSubStr = coreSqlUpper.substring(currWindowIndex, currWindowIndex+6);			
-			if("SELECT".equals(currSubStr)) {
-				++selectSubSqlNum;
-				currWindowIndex += 6;
-			}else if(currSubStr.startsWith(" FROM")) {
-				if(selectSubSqlNum == 1) {
-					return currWindowIndex;
-				}
-				--selectSubSqlNum;
-				currWindowIndex += 5;
+		
+	private static int calcFromIndex(String coreSql) {
+		Stack<String> operStack = new Stack<String>();
+		String coreSqlUpperCase = coreSql.toUpperCase();
+		int sqlLen = coreSqlUpperCase.length();
+		int currIndex = 0;
+		while(currIndex < sqlLen) {
+			String originContainBlank = subFirstStrEndWithBlank(coreSqlUpperCase, currIndex);
+			String itemWithoutBlank = originContainBlank.trim();
+			if("".equals(itemWithoutBlank)) {
+				currIndex += originContainBlank.length();
+				continue;
 			}
-			++currWindowIndex;
+			if(operStack.isEmpty() && "FROM".equals(itemWithoutBlank)) {
+				return currIndex;
+			}
+			dealCoupleMark(itemWithoutBlank, operStack);			
+			currIndex += originContainBlank.length();
 		}
 		return -1;
 	}
-	
-	private static String replaceSelectAndFromBetweenTopCornerMark(String sql) {
-		StringBuffer result = new StringBuffer();
-		boolean isBetweenMark = false;
-		int iMark = 0;		
-		int currIndex = 0;
-		for(currIndex=0; currIndex<sql.length(); ++currIndex) {
-			char currChar = sql.charAt(currIndex);
-			if(currChar == '\'') {
-				if(!isBetweenMark) {
-					result.append(sql.substring(iMark, currIndex));	
-					isBetweenMark = true;
-				}else {
-					result.append(sql.substring(iMark, currIndex).replaceAll("SELECT", "******").replaceAll("FROM", "****"));
-					isBetweenMark = false;
+	private static void dealCoupleMark(String itemWithoutBlank, Stack<String> operStack) {
+		char preChar = ' ';
+		for(int i=0; i<itemWithoutBlank.length(); ++i) {
+			char currChar = itemWithoutBlank.charAt(i);
+			if(preChar != '\\' && currChar == '\'') {
+				if(operStack.isEmpty() || !operStack.peek().equals("'")) {
+					operStack.push("'");
+				}else if(operStack.peek().equals("'")) {
+					operStack.pop();
+				}
+			}else if(currChar == '('){
+				if(operStack.isEmpty() || !operStack.peek().equals("'")) {
+					operStack.push("(");	
+				}
+			}else if(currChar == ')') {
+				if(!operStack.isEmpty() && operStack.peek().equals("(")) {
+					operStack.pop();
 				}
-				iMark = currIndex;
-			}else if(currChar == '\\') {
-				++currIndex;
 			}
+			preChar = currChar;
 		}
-		if(iMark > -1) {
-			result.append(sql.substring(iMark, currIndex));
-		}
-		return result.toString();
 	}
-	
+
+	private static String subFirstStrEndWithBlank(String coreSqlUpperCase, int startIndex) {		
+		boolean startWithBlank = coreSqlUpperCase.charAt(startIndex)==' ';
+		int endIndex = startIndex;
+		int sqlLen = coreSqlUpperCase.length();
+		char preChar = '0';
+		for(int i=startIndex; i<sqlLen; ++i) {
+			char currChar = coreSqlUpperCase.charAt(i);
+			if(startWithBlank && currChar!=' ') {
+				endIndex = i;
+				break;
+			}			
+			if(!startWithBlank && currChar!=' ' && preChar==' ') {
+				endIndex = i;
+				break;
+			}
+			preChar = currChar;
+			if(i == sqlLen-1) {
+				endIndex = sqlLen;
+			}
+		}		
+		return coreSqlUpperCase.substring(startIndex, endIndex);
+	}
+
 	/**
 	 * 获取排序字符串
 	 * @param sort
@@ -145,7 +164,8 @@ public class UtilsSql {
 	
 	public static void main(String[] args) {
 		String[] testSql = new String[] {
-			"select actionFrom t,(select * as 'fromCActio', dd as 'seelctsele\'ctd' from t) from www where 223",
+			"\\",
+			"select actionFrom as 'seelctsele\\'ctd',(select a as 'fromCActio' from t) from www where 223=1 ",
 			"SELECT wxorder_order.refund_way refundWay,wxorder_order.refund_delivery_sn refundDeliverySn, (SELECT sum(wxorder_compensate.`ware_refund_money`)  from `wxorder_compensate`  where wxorder_compensate.`order_id`  = wxorder_order.id) as wareRefundMoney,(SELECT sum(wxorder_compensate.`refund_money`)  from `wxorder_compensate`  where wxorder_compensate.`order_id`  = wxorder_order.id) as refundMoney,  wxorder_order.operate_refund_time applyCompensateTime,wxorder_order.refund_instructions refundInstructions, wxorder_order.channel_id channelId, wxorder_channel.name channelName, wxorder_channel.platform_code platformCode,  wxorder_order.pre_control_status as refundPreStatus, wxorder_order.ware_id wareId, wxorder_ware.name wareName,  wxorder_order.id, wxorder_order.order_sn orderSn, wxorder_order.upload_sn uploadSn, wxorder_order.money_paid moneyPaid,  wxorder_order.consignee, wxorder_order.mobile, wxorder_order.province_name provinceName, wxorder_order.city_name cityName, wxorder_order.district_name districtName, wxorder_order.address,  wxorder_order.channel_remark channelRemark, wxorder_order.customer_remark customerRemark, wxorder_order.platform_customer_remark platformCustomerRemark,  wxorder_order.sender_name senderName, wxorder_order.sender_mobile senderMobile,  date_format(wxorder_order.create_time, '%Y-%m-%d %H:%i:%s') createTime, date_format(wxorder_order.real_delivery_time, '%Y-%m-%d %H:%i:%s') realDeliveryTime,  wxorder_order.progress_distribute progressDistribute, wxorder_order.progress_delivery progressDelivery, wxorder_order.control_status controlStatus,  wxorder_order.express_name expressName, wxorder_order.express_number expressNumber,wxorder_order.shipping_price shippingPrice, wxorder_order.customer_network_name ,(select count(*) from wxorder_compensate WHERE wxorder_compensate.order_id = wxorder_order.id) as compensateCountAll,(select count(*) from wxorder_compensate WHERE wxorder_compensate.order_id = wxorder_order.id and wxorder_compensate.progress_status = 1) as compensateCountDealed ,(select count(*) from wxorder_channel_refund_payment where wxorder_channel_refund_payment.order_id = wxorder_order.id) as channelRefundPaymentStatus,(select count(*) from wxorder_ware_refund_payment where wxorder_ware_refund_payment.order_id = wxorder_order.id) as wareRefundPaymentStatus  FROM wxorder_order wxorder_order  INNER JOIN wxorder_channel wxorder_channel ON wxorder_order.channel_id=wxorder_channel.id and wxorder_channel.tenant_id=54 INNER JOIN wxorder_ware wxorder_ware ON wxorder_order.ware_id=wxorder_ware.id and wxorder_ware.tenant_id=54 INNER JOIN wxorder_order_goods wxorder_order_goods ON wxorder_order.id=wxorder_order_goods.order_id and wxorder_order_goods.tenant_id=54 WHERE 1=1  AND wxorder_order.tenant_id='54' AND wxorder_order.control_status='4' GROUP BY wxorder_order.id",
 			"select wdd, fromType from www where 223",
 		};
@@ -156,9 +176,14 @@ public class UtilsSql {
 				String[] data = splitCoreSql(item);
 				System.out.println(data[0]);
 				System.out.println(data[1]);
-			}			
+			}
 		} catch (Exception e) {
 			e.printStackTrace();
 		}
+		
+//		String testSub = " asdf asdf   asdfaf";
+//		System.out.println(subFirstStrEndWithBlank(testSub, 5));
+//		System.out.println(subFirstStrEndWithBlank(testSub, 6));
+//		System.out.println(subFirstStrEndWithBlank(testSub, 13));
 	}
 }
--
libgit2 0.21.2