fixbug 字段血缘的relation

dlink-core/src/main/java/com/dlink/explainer/lineage/LineageColumnGenerator.java

@@ -68,17 +68,17 @@ public class LineageColumnGenerator {
             Trans trans = entry.getValue();
             if (trans instanceof SinkTrans) {
                 TableCA tableCA = new TableCA((SinkTrans) trans);
-                matchSinkField(tableCA,trans);
+                matchSinkField(tableCA, trans);
                 searchColumnCAId(tableCA);
             }
         }
         chainRelation();
     }
 
-    private void matchSinkField(TableCA tableCA,Trans trans){
-        for(ColumnCA columnCA: columnCAS){
-            for(String fieldName: tableCA.getFields()){
-                if(columnCA.getName().equals(fieldName)){
+    private void matchSinkField(TableCA tableCA, Trans trans) {
+        for (ColumnCA columnCA : columnCAS) {
+            for (String fieldName : tableCA.getFields()) {
+                if (columnCA.getName().equals(fieldName)) {
                     int cid = index++;
                     ColumnCA sinkColumnCA = new ColumnCA(cid, fieldName, fieldName, fieldName, fieldName, fieldName, tableCA, trans);
                     columnCASMaps.put(cid, sinkColumnCA);
@@ -124,7 +124,7 @@ public class LineageColumnGenerator {
                     break;
                 }
             }
-            if(columnCA.getId()==cid){
+            if (columnCA.getId() == cid) {
                 return;
             }
             if (!isHad) {
@@ -176,20 +176,22 @@ public class LineageColumnGenerator {
     }
 
     private void chainRelation() {
-        Set<NodeRel> nodeRelsChain = new HashSet<>();
         for (Integer item : sourceColumns) {
-            nodeRelsChain.add(new NodeRel(item, getNextSuf(item)));
+            buildSinkSuf(item, item);
         }
-        columnCASRelChain = nodeRelsChain;
     }
 
-    private Integer getNextSuf(Integer sufId) {
+    private void buildSinkSuf(Integer preId, Integer sourcePreId) {
         for (NodeRel nodeRel : columnCASRel) {
-            if (nodeRel.getPreId() == sufId) {
-                return getNextSuf(nodeRel.getSufId());
+            if (nodeRel.getPreId() == preId) {
+                Integer nextSufId = nodeRel.getSufId();
+                if (sinkColumns.contains(nextSufId)) {
+                    columnCASRelChain.add(new NodeRel(sourcePreId, nextSufId));
+                    continue;
+                }
+                buildSinkSuf(nextSufId, sourcePreId);
             }
         }
-        return sufId;
     }
 
     public Map<Integer, Trans> getTransMaps() {

dlink-core/src/main/java/com/dlink/utils/MapParseUtils.java

@@ -115,6 +115,11 @@ public class MapParseUtils {
         }
         Deque<Integer> stack = new LinkedList<>();
         for (int i = 0; i < inStr.length(); i++) {
+            if (inStr.charAt(i) == ']' && stack.size() == 0) {
+                selects.add(inStr.substring(startIndex, i));
+                startIndex = i + 1;
+                return selects;
+            }
             if (inStr.charAt(i) == ',' && stack.size() == 0) {
                 selects.add(inStr.substring(startIndex, i));
                 startIndex = i + 1;

dlink-core/src/test/java/com/dlink/core/LineageTest.java

@@ -28,7 +28,7 @@ public class LineageTest {
                 ") WITH (\n" +
                 " 'connector' = 'print'\n" +
                 ");\n" +
-                "insert into TT select a||c A ,b B from ST";
+                "insert into TT select a||c A ,b||c B from ST";
         LineageResult result = LineageBuilder.getLineage(sql);
         System.out.println("end");
     }