From 60bad33e7eafbb245a3c71d1fdbc1e2c0d58f108 Mon Sep 17 00:00:00 2001
From: minghong <englefly@gmail.com>
Date: Thu, 6 Apr 2023 08:55:05 +0800
Subject: [PATCH] [fix](nereids) explain shape refactor #18399

previous pr 18296 has a bug when parse SHAPE_PLAN.
---
 .../src/main/java/org/apache/doris/nereids/NereidsPlanner.java | 2 --
 .../org/apache/doris/nereids/parser/LogicalPlanBuilder.java    | 3 +++
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/fe/fe-core/src/main/java/org/apache/doris/nereids/NereidsPlanner.java b/fe/fe-core/src/main/java/org/apache/doris/nereids/NereidsPlanner.java
index 1604bc46a3..2cd44ccb38 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/nereids/NereidsPlanner.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/nereids/NereidsPlanner.java
@@ -218,8 +218,6 @@ public class NereidsPlanner extends Planner {
                 String tree = physicalPlan.treeString();
                 LOG.info(tree);
             }
-
-            System.out.println(physicalPlan.shape("  "));
             if (explainLevel == ExplainLevel.OPTIMIZED_PLAN
                     || explainLevel == ExplainLevel.ALL_PLAN
                     || explainLevel == ExplainLevel.SHAPE_PLAN) {
diff --git a/fe/fe-core/src/main/java/org/apache/doris/nereids/parser/LogicalPlanBuilder.java b/fe/fe-core/src/main/java/org/apache/doris/nereids/parser/LogicalPlanBuilder.java
index 65cdf62ad5..f22ff8d63c 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/nereids/parser/LogicalPlanBuilder.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/nereids/parser/LogicalPlanBuilder.java
@@ -1772,6 +1772,9 @@ public class LogicalPlanBuilder extends DorisParserBaseVisitor<Object> {
         if (planTypeContext.PARSED() != null) {
             return ExplainLevel.PARSED_PLAN;
         }
+        if (planTypeContext.SHAPE() != null) {
+            return ExplainLevel.SHAPE_PLAN;
+        }
         return ExplainLevel.ALL_PLAN;
     }