fix (#458)

linkedin · Oct 3, 2023 · a886a95 · a886a95
1 parent 221a453
commit a886a95
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/coral-visualization/src/main/java/com/linkedin/coral/vis/SqlNodeVisualizationVisitor.java b/coral-visualization/src/main/java/com/linkedin/coral/vis/SqlNodeVisualizationVisitor.java
@@ -6,6 +6,7 @@
 package com.linkedin.coral.vis;
 
 import java.util.ArrayList;
+import java.util.Collection;
 import java.util.List;
 import java.util.UUID;
 
@@ -20,6 +21,7 @@
 import org.apache.calcite.sql.SqlNode;
 import org.apache.calcite.sql.SqlNodeList;
 import org.apache.calcite.sql.SqlSelect;
+import org.apache.calcite.sql.parser.SqlParserPos;
 import org.apache.calcite.sql.util.SqlVisitor;
 
 import guru.nidi.graphviz.attribute.Label;
@@ -28,6 +30,7 @@
 import guru.nidi.graphviz.model.LinkTarget;
 import guru.nidi.graphviz.model.Node;
 
+import static com.linkedin.coral.common.calcite.CalciteUtil.*;
 import static guru.nidi.graphviz.model.Factory.*;
 
 
@@ -59,6 +62,8 @@ public Node visit(SqlCall call) {
       }
       if (sqlSelect.getSelectList() != null) {
         edges.add(edge(sqlSelect.getSelectList(), "select_list"));
+      } else {
+        edges.add(edge(createSelectStarSqlNodeList(), "select_list"));
       }
       if (sqlSelect.getOrderList() != null) {
         edges.add(edge(sqlSelect.getOrderList(), "order_by_list"));
@@ -122,4 +127,11 @@ private Link edge(SqlNode target, String label) {
   private static Node node(String label) {
     return Factory.node(UUID.randomUUID().toString()).with(Label.of(label));
   }
+
+  private static SqlNodeList createSelectStarSqlNodeList() {
+    Collection<SqlNode> sqlNodeList = new ArrayList();
+    sqlNodeList.add(createStarIdentifier(SqlParserPos.ZERO));
+    SqlNodeList selectStar = createSqlNodeList(sqlNodeList);
+    return selectStar;
+  }
 }