[Feature-1709][core] Support use FlinkSQL CTE to preview data

aiwenmo · Nov 1, 2023 · 8187465 · 8187465
1 parent 523baf5
commit 8187465
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 1 deletion.
diff --git a/dinky-core/src/main/java/org/dinky/data/result/ResultBuilder.java b/dinky-core/src/main/java/org/dinky/data/result/ResultBuilder.java
@@ -39,6 +39,7 @@ static ResultBuilder build(
             String timeZone) {
         switch (operationType) {
             case SELECT:
+            case WITH:
                 return new SelectResultBuilder(id, maxRowNum, isChangeLog, isAutoCancel, timeZone);
             case SHOW:
             case DESC:

diff --git a/dinky-core/src/main/java/org/dinky/explainer/Explainer.java b/dinky-core/src/main/java/org/dinky/explainer/Explainer.java
@@ -126,6 +126,7 @@ public JobParam pretreatStatements(String[] statements) {
                 statementList.add(statement);
             } else if (operationType.equals(SqlType.INSERT)
                     || operationType.equals(SqlType.SELECT)
+                    || operationType.equals(SqlType.WITH)
                     || operationType.equals(SqlType.SHOW)
                     || operationType.equals(SqlType.DESCRIBE)
                     || operationType.equals(SqlType.DESC)

diff --git a/dinky-executor/src/main/java/org/dinky/parser/SqlType.java b/dinky-executor/src/main/java/org/dinky/parser/SqlType.java
@@ -63,6 +63,8 @@ public enum SqlType {
 
     CTAS("CTAS", "^CREATE\\s.*AS\\sSELECT.*$"),
 
+    WITH("WITH", "^WITH.*"),
+
     UNKNOWN("UNKNOWN", "^UNKNOWN.*");
 
     private String type;

diff --git a/dinky-web/src/pages/DataStudio/BottomContainer/Result/index.tsx b/dinky-web/src/pages/DataStudio/BottomContainer/Result/index.tsx
@@ -147,7 +147,7 @@ const Result = (props: any) => {
             id: current.id
           });
           const historyData = res.data;
-          if ('2' == historyData.status) {
+          if (historyData && '2' == historyData.status) {
             const historyId = historyData.id;
             const tableData = await handleGetOption('api/studio/getJobData', 'Get Data', {
               jobId: historyId