Commit @35682a525d3cc52a49c60173665376a9b3534a75

김준식 2024-02-07

240207 김준식 코드관리, 업무규칙관리 sql 검사 부분 수정

@35682a525d3cc52a49c60173665376a9b3534a75

a0e70c6

35682a5

pom.xml

--- pom.xml

+++ pom.xml


         <dependency>
             <groupId>com.github.jsqlparser</groupId>
             <artifactId>jsqlparser</artifactId>
-            <version>3.2</version>
+            <version>4.7</version>
         </dependency>
     </dependencies>
 

a0e70c6

35682a5

server/main/java/kr/co/takensoft/dataQuality/rule/service/CodeManagementService.java

--- server/main/java/kr/co/takensoft/dataQuality/rule/service/CodeManagementService.java

+++ server/main/java/kr/co/takensoft/dataQuality/rule/service/CodeManagementService.java


             Statement statement = CCJSqlParserUtil.parse(sql);
 
             // SQL 문이 SELECT 문인지 확인
-            if (statement instanceof Select) {
+            if (statement instanceof Select && sql.toUpperCase().contains("FROM")) {
                 result = codeMgDAO.insertCode(params);
             }
         } catch (JSQLParserException e) {
             // SQL 문 파싱 오류 처리
             System.out.println("SQL 문 파싱 오류: " + e.getMessage());
         }
-        System.out.println("result : " + result);
         return result;
     }
 

             Statement statement = CCJSqlParserUtil.parse(sql);
 
             // SQL 문이 SELECT 문인지 확인
-            if (statement instanceof Select) {
+            if (statement instanceof Select && sql.toUpperCase().contains("FROM")) {
                 result = codeMgDAO.updateCode(params);
             }
         } catch (JSQLParserException e) {

a0e70c6

35682a5

server/main/java/kr/co/takensoft/dataQuality/rule/service/WorkRuleManagementService.java

--- server/main/java/kr/co/takensoft/dataQuality/rule/service/WorkRuleManagementService.java

+++ server/main/java/kr/co/takensoft/dataQuality/rule/service/WorkRuleManagementService.java


 package kr.co.takensoft.dataQuality.rule.service;
 
 import kr.co.takensoft.dataQuality.rule.dao.WorkRuleManagementDAO;
+import net.sf.jsqlparser.JSQLParserException;
+import net.sf.jsqlparser.parser.CCJSqlParserUtil;
+import net.sf.jsqlparser.statement.Statement;
+import net.sf.jsqlparser.statement.select.Select;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 

     public int workRuleRegister(HashMap<String, Object> params){
         int result = 0;
         try{
-            result = wrManagementDAO.workRuleRegister(params);
-        } catch (Exception e){
+            String cnt = params.get("work_rule_cnt").toString(); // 건수 sql
+            String sql = params.get("work_rule_sql").toString(); // 분석 sql
+            // SQL 문 파싱
+            Statement cnt_statement = CCJSqlParserUtil.parse(cnt);
+            Statement sql_statement = CCJSqlParserUtil.parse(sql);
+            // SQL 문이 SELECT 문인지 확인
+            boolean compare = sql_statement instanceof Select && cnt_statement instanceof Select && sql.toUpperCase().contains("FROM") && cnt.toUpperCase().contains("FROM");
+            if(compare) {
+                result = wrManagementDAO.workRuleRegister(params);
+            }
+        } catch (JSQLParserException e){
             e.printStackTrace();
+            System.out.println("sql 파싱 오류 : " + e.getMessage());
         }
 
         return result;

     public int workRuleUpdate(HashMap<String, Object> params){
         int result = 0;
         try{
-            result = wrManagementDAO.workRuleUpdate(params);
-        } catch (Exception e){
-            e.printStackTrace();
+            String sql = params.get("work_rule_sql").toString();
+            String cnt = params.get("work_rule_cnt").toString();
+            //SQL 문 파싱
+            Statement sql_statement = CCJSqlParserUtil.parse(sql);
+            Statement cnt_statement = CCJSqlParserUtil.parse(cnt);
+            // SQL 문이 SELECT 문인지 확인
+            boolean compare = sql_statement instanceof Select && cnt_statement instanceof Select && sql.toUpperCase().contains("FROM") && cnt.toUpperCase().contains("FROM");
+            if(compare) {
+                result = wrManagementDAO.workRuleUpdate(params);
+            }
+        } catch (JSQLParserException e){
+            System.out.println("sql 파싱 오류 : " + e.getMessage());
         }
 
         return result;

Add a comment

Open 0
Closed 0

List

...	...	@@ -129,7 +129,7 @@
129	129	<dependency>
130	130	<groupId>com.github.jsqlparser</groupId>
131	131	<artifactId>jsqlparser</artifactId>
132		- <version>3.2</version>
	132	+ <version>4.7</version>
133	133	</dependency>
134	134	</dependencies>
135	135

...	...	@@ -46,14 +46,13 @@
46	46	Statement statement = CCJSqlParserUtil.parse(sql);
47	47
48	48	// SQL 문이 SELECT 문인지 확인
49		- if (statement instanceof Select) {
	49	+ if (statement instanceof Select && sql.toUpperCase().contains("FROM")) {
50	50	result = codeMgDAO.insertCode(params);
51	51	}
52	52	} catch (JSQLParserException e) {
53	53	// SQL 문 파싱 오류 처리
54	54	System.out.println("SQL 문 파싱 오류: " + e.getMessage());
55	55	}
56		- System.out.println("result : " + result);
57	56	return result;
58	57	}
59	58
...	...	@@ -71,7 +70,7 @@
71	70	Statement statement = CCJSqlParserUtil.parse(sql);
72	71
73	72	// SQL 문이 SELECT 문인지 확인
74		- if (statement instanceof Select) {
	73	+ if (statement instanceof Select && sql.toUpperCase().contains("FROM")) {
75	74	result = codeMgDAO.updateCode(params);
76	75	}
77	76	} catch (JSQLParserException e) {

...	...	@@ -1,6 +1,10 @@
1	1	package kr.co.takensoft.dataQuality.rule.service;
2	2
3	3	import kr.co.takensoft.dataQuality.rule.dao.WorkRuleManagementDAO;
	4	+import net.sf.jsqlparser.JSQLParserException;
	5	+import net.sf.jsqlparser.parser.CCJSqlParserUtil;
	6	+import net.sf.jsqlparser.statement.Statement;
	7	+import net.sf.jsqlparser.statement.select.Select;
4	8	import org.springframework.beans.factory.annotation.Autowired;
5	9	import org.springframework.stereotype.Service;
6	10
...	...	@@ -81,9 +85,19 @@
81	85	public int workRuleRegister(HashMap<String, Object> params){
82	86	int result = 0;
83	87	try{
84		- result = wrManagementDAO.workRuleRegister(params);
85		- } catch (Exception e){
	88	+ String cnt = params.get("work_rule_cnt").toString(); // 건수 sql
	89	+ String sql = params.get("work_rule_sql").toString(); // 분석 sql
	90	+ // SQL 문 파싱
	91	+ Statement cnt_statement = CCJSqlParserUtil.parse(cnt);
	92	+ Statement sql_statement = CCJSqlParserUtil.parse(sql);
	93	+ // SQL 문이 SELECT 문인지 확인
	94	+ boolean compare = sql_statement instanceof Select && cnt_statement instanceof Select && sql.toUpperCase().contains("FROM") && cnt.toUpperCase().contains("FROM");
	95	+ if(compare) {
	96	+ result = wrManagementDAO.workRuleRegister(params);
	97	+ }
	98	+ } catch (JSQLParserException e){
86	99	e.printStackTrace();
	100	+ System.out.println("sql 파싱 오류 : " + e.getMessage());
87	101	}
88	102
89	103	return result;
...	...	@@ -98,9 +112,18 @@
98	112	public int workRuleUpdate(HashMap<String, Object> params){
99	113	int result = 0;
100	114	try{
101		- result = wrManagementDAO.workRuleUpdate(params);
102		- } catch (Exception e){
103		- e.printStackTrace();
	115	+ String sql = params.get("work_rule_sql").toString();
	116	+ String cnt = params.get("work_rule_cnt").toString();
	117	+ //SQL 문 파싱
	118	+ Statement sql_statement = CCJSqlParserUtil.parse(sql);
	119	+ Statement cnt_statement = CCJSqlParserUtil.parse(cnt);
	120	+ // SQL 문이 SELECT 문인지 확인
	121	+ boolean compare = sql_statement instanceof Select && cnt_statement instanceof Select && sql.toUpperCase().contains("FROM") && cnt.toUpperCase().contains("FROM");
	122	+ if(compare) {
	123	+ result = wrManagementDAO.workRuleUpdate(params);
	124	+ }
	125	+ } catch (JSQLParserException e){
	126	+ System.out.println("sql 파싱 오류 : " + e.getMessage());
104	127	}
105	128
106	129	return result;

Delete comment