修改代码

luop23 · luop23 · commit 49f6db2b0ef7 · 2019-12-06T17:20:14.000+08:00
diff --git a/README.md b/README.md
@@ -1 +1,28 @@
-基于jsqlparser 进行sql语句解析，得到字段间的血缘关系
+基于jsqlparser 进行sql语句解析，得到字段间的血缘关系
+==
+
+工具说明：
+-
+    1、本工具只解析select语句
+    2、支持子查询
+    3、支持常用数据库的常用函数，如concat、case when、nvl、isnull、cast、ifnull等等
+    4、工具可能对某些sql无法解析，有待完善
+    
+    
+注意事项
+-   
+    1、在调用方法前请确保sql的准确性，本工具只做了简单的判断
+    2、当多表查询时，请给表和字段附上相应别名，如若没有，可能会对结果产生影响
+    
+事例
+-
+    参考测试类 test()
+
+参考网址
+-
+    https://github.com/JSQLParser/JSqlParser
+    https://gitee.com/zzyijia/jsqlparser
+
+
+    
+    
diff --git a/src/main/java/com/luop/SelectParseHelper.java b/src/main/java/com/luop/SelectParseHelper.java
@@ -9,28 +9,22 @@
 import java.util.Set;
 
 /**
- * SQL解析帮助类
- * 约束：
- * 	        仅支持SELECT语句的解析
- *     查询字段不可以为*
- *     表必须要使用别名，查询字段必须使用表的别名(例：select t.id,t.address from userinfo t)
- * 说明：
- *     该方法的使用场景：控制用户不要越权访问表或者表的字段
- *     出于上述使用场景，SQL解析时只关心where之前的部分，包括子查询语句。解析结果返回所有查询涉及到的表和查询的字段。
- *     SQL解析利用开源的JSQLParser来实现，参考网址：https://github.com/JSQLParser/JSqlParser
+ * SQL解析利用开源的JSQLParser来实现，参考网址：https://github.com/JSQLParser/JSqlParser
  *
- * @author "朱云山"
+ * @Author: luoping
+ * @Date: 2019/11/5 09:50
+ * @Description:
  *
  */
 class SelectParseHelper {
     /**
-     * 获取SQL语句中用到的所有表和查询字段名称(名称均为大写)
-     *
+     * 获取血缘关系结果
      * @param sqltxt 要解析的SQL语句
      * @return
      * @throws Exception
      */
-    public static Map<String,Set<String>> getTableAndColumns(String sqltxt) throws Exception{
+    static Map<String,Set<String>> getBloodRelationResult(String sqltxt) throws Exception{
+        //第三方插件解析sql
         Statement stmt = CCJSqlParserUtil.parse(sqltxt);     //报错 说明sql语句错误
         Select selectStatement=(Select)stmt;
 
diff --git a/src/test/java/com/luop/SqlAnalysisApplicationTests.java b/src/test/java/com/luop/SqlAnalysisApplicationTests.java
@@ -3,11 +3,26 @@
 import org.junit.jupiter.api.Test;
 import org.springframework.boot.test.context.SpringBootTest;
 
+import java.util.Map;
+import java.util.Set;
+
 @SpringBootTest
 class SqlAnalysisApplicationTests {
 
     @Test
     void contextLoads() {
     }
 
+    @Test
+    void test(){
+        String sql = "select u.name userName,u.id,d.name depName,d.id depId " +
+                "from user u left join department d on u.depId = d.id where u.id = 1";
+        try {
+            Map<String, Set<String>> map = SelectParseHelper.getBloodRelationResult(sql);
+            System.out.println(map);   //{depId=[department.id], userName=[user.name], userId=[user.id], depName=[department.name]}
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+
 }
