Merge pull request #7401 from leiysky/explain-physical-plan

chore(planner): Add statistics info in explain result

src/query/ast/src/ast/format/mod.rs

@@ -26,8 +26,8 @@ pub use syntax::pretty_statement;
 
 #[derive(Clone)]
 pub struct FormatTreeNode<T: Display + Clone> {
-    payload: T,
-    children: Vec<Self>,
+    pub payload: T,
+    pub children: Vec<Self>,
 }
 
 impl<T> FormatTreeNode<T>

src/query/service/src/interpreters/fragments/v2/plan_fragment.rs

@@ -205,6 +205,7 @@ impl PhysicalPlanReplacer for ReplaceReadSource {
         Ok(PhysicalPlan::TableScan(TableScan {
             source: Box::new(self.source.clone()),
             name_mapping: plan.name_mapping.clone(),
+            table_index: plan.table_index,
         }))
     }
 }

src/query/service/src/interpreters/interpreter_explain_v2.rs

@@ -25,6 +25,7 @@ use super::fragments::Fragmenter;
 use super::QueryFragmentsActions;
 use crate::interpreters::Interpreter;
 use crate::sessions::QueryContext;
+use crate::sql::executor::PhysicalPlan;
 use crate::sql::executor::PhysicalPlanBuilder;
 use crate::sql::executor::PipelineBuilder;
 use crate::sql::optimizer::SExpr;
@@ -53,7 +54,17 @@ impl Interpreter for ExplainInterpreterV2 {
             ExplainKind::Ast(stmt) | ExplainKind::Syntax(stmt) => {
                 self.explain_ast_or_syntax(stmt.clone())?
             }
-            ExplainKind::Raw | ExplainKind::Plan => self.explain_raw_or_plan(&self.plan)?,
+            ExplainKind::Raw => self.explain_plan(&self.plan)?,
+            ExplainKind::Plan => match &self.plan {
+                Plan::Query {
+                    s_expr, metadata, ..
+                } => {
+                    let builder = PhysicalPlanBuilder::new(metadata.clone(), self.ctx.clone());
+                    let plan = builder.build(s_expr).await?;
+                    self.explain_physical_plan(&plan, metadata)?
+                }
+                _ => self.explain_plan(&self.plan)?,
+            },
             ExplainKind::Pipeline => match &self.plan {
                 Plan::Query {
                     s_expr, metadata, ..
@@ -116,7 +127,7 @@ impl ExplainInterpreterV2 {
         ])])
     }
 
-    pub fn explain_raw_or_plan(&self, plan: &Plan) -> Result<Vec<DataBlock>> {
+    pub fn explain_plan(&self, plan: &Plan) -> Result<Vec<DataBlock>> {
         let result = plan.format_indent()?;
         let line_splitted_result: Vec<&str> = result.lines().collect();
         let formatted_plan = Series::from_data(line_splitted_result);
@@ -125,6 +136,19 @@ impl ExplainInterpreterV2 {
         ])])
     }
 
+    pub fn explain_physical_plan(
+        &self,
+        plan: &PhysicalPlan,
+        metadata: &MetadataRef,
+    ) -> Result<Vec<DataBlock>> {
+        let result = plan.format(metadata.clone())?;
+        let line_splitted_result: Vec<&str> = result.lines().collect();
+        let formatted_plan = Series::from_data(line_splitted_result);
+        Ok(vec![DataBlock::create(self.schema.clone(), vec![
+            formatted_plan,
+        ])])
+    }
+
     pub async fn explain_pipeline(
         &self,
         s_expr: SExpr,