refactor(query): use trait to refactor physical plan

Author: zhang2014

src/query/service/src/physical_plans/format/common.rs

@@ -222,6 +222,7 @@ impl<'a> PhysicalFormat for SimplePhysicalFormat<'a> {
         self.meta
     }
 
+    #[recursive::recursive]
     fn format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         let mut children = vec![];
         for child in self.children.iter() {
@@ -234,6 +235,7 @@ impl<'a> PhysicalFormat for SimplePhysicalFormat<'a> {
         ))
     }
 
+    #[recursive::recursive]
     fn format_join(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         if self.children.len() == 1 {
             return self.children[0].format_join(ctx);
@@ -250,6 +252,7 @@ impl<'a> PhysicalFormat for SimplePhysicalFormat<'a> {
         ))
     }
 
+    #[recursive::recursive]
     fn partial_format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         if self.children.len() == 1 {
             return self.children[0].partial_format(ctx);

src/query/service/src/physical_plans/format/format_add_stream_column.rs

@@ -36,16 +36,19 @@ impl<'a> PhysicalFormat for AddStreamColumnFormatter<'a> {
         self.inner.get_meta()
     }
 
+    #[recursive::recursive]
     fn format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         // ignore self
         let input_formatter = self.inner.input.formatter()?;
         input_formatter.dispatch(ctx)
     }
 
+    #[recursive::recursive]
     fn format_join(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.format_join(ctx)
     }
 
+    #[recursive::recursive]
     fn partial_format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.partial_format(ctx)
     }

src/query/service/src/physical_plans/format/format_aggregate_expand.rs

@@ -38,6 +38,7 @@ impl<'a> PhysicalFormat for AggregateExpandFormatter<'a> {
         self.inner.get_meta()
     }
 
+    #[recursive::recursive]
     fn format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         let sets = self
             .inner
@@ -77,10 +78,12 @@ impl<'a> PhysicalFormat for AggregateExpandFormatter<'a> {
         ))
     }
 
+    #[recursive::recursive]
     fn format_join(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.format_join(ctx)
     }
 
+    #[recursive::recursive]
     fn partial_format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.partial_format(ctx)
     }

src/query/service/src/physical_plans/format/format_aggregate_final.rs

@@ -39,6 +39,7 @@ impl<'a> PhysicalFormat for AggregateFinalFormatter<'a> {
         self.inner.get_meta()
     }
 
+    #[recursive::recursive]
     fn format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         let group_by = self
             .inner
@@ -82,10 +83,12 @@ impl<'a> PhysicalFormat for AggregateFinalFormatter<'a> {
         ))
     }
 
+    #[recursive::recursive]
     fn format_join(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.format_join(ctx)
     }
 
+    #[recursive::recursive]
     fn partial_format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.partial_format(ctx)
     }

src/query/service/src/physical_plans/format/format_aggregate_partial.rs

@@ -39,6 +39,7 @@ impl<'a> PhysicalFormat for AggregatePartialFormatter<'a> {
         self.inner.get_meta()
     }
 
+    #[recursive::recursive]
     fn format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         let group_by = self
             .inner
@@ -77,10 +78,12 @@ impl<'a> PhysicalFormat for AggregatePartialFormatter<'a> {
         ))
     }
 
+    #[recursive::recursive]
     fn format_join(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.format_join(ctx)
     }
 
+    #[recursive::recursive]
     fn partial_format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.partial_format(ctx)
     }

src/query/service/src/physical_plans/format/format_async_func.rs

@@ -38,6 +38,7 @@ impl<'a> PhysicalFormat for AsyncFunctionFormatter<'a> {
         self.inner.get_meta()
     }
 
+    #[recursive::recursive]
     fn format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         let output_schema = self.inner.output_schema()?;
         let mut children = vec![FormatTreeNode::new(format!(
@@ -58,10 +59,12 @@ impl<'a> PhysicalFormat for AsyncFunctionFormatter<'a> {
         ))
     }
 
+    #[recursive::recursive]
     fn format_join(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.format_join(ctx)
     }
 
+    #[recursive::recursive]
     fn partial_format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.partial_format(ctx)
     }

src/query/service/src/physical_plans/format/format_cache_scan.rs

@@ -38,6 +38,7 @@ impl<'a> PhysicalFormat for CacheScanFormatter<'a> {
         self.inner.get_meta()
     }
 
+    #[recursive::recursive]
     fn format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         let mut children = Vec::with_capacity(2);
         children.push(FormatTreeNode::new(format!(

src/query/service/src/physical_plans/format/format_chunk_cast_schema.rs

@@ -36,15 +36,18 @@ impl<'a> PhysicalFormat for ChunkCastSchemaFormatter<'a> {
         self.inner.get_meta()
     }
 
+    #[recursive::recursive]
     fn format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         let input_formatter = self.inner.input.formatter()?;
         input_formatter.dispatch(ctx)
     }
 
+    #[recursive::recursive]
     fn format_join(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.format_join(ctx)
     }
 
+    #[recursive::recursive]
     fn partial_format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.partial_format(ctx)
     }

src/query/service/src/physical_plans/format/format_chunk_eval_scalar.rs

@@ -38,6 +38,7 @@ impl<'a> PhysicalFormat for ChunkEvalScalarFormatter<'a> {
         self.inner.get_meta()
     }
 
+    #[recursive::recursive]
     fn format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         if self.inner.eval_scalars.iter().all(|x| x.is_none()) {
             let input_formatter = self.inner.input.formatter()?;
@@ -70,10 +71,12 @@ impl<'a> PhysicalFormat for ChunkEvalScalarFormatter<'a> {
         ))
     }
 
+    #[recursive::recursive]
     fn format_join(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.format_join(ctx)
     }
 
+    #[recursive::recursive]
     fn partial_format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.partial_format(ctx)
     }

src/query/service/src/physical_plans/format/format_chunk_fill_and_reorder.rs

@@ -36,16 +36,19 @@ impl<'a> PhysicalFormat for ChunkFillAndReorderFormatter<'a> {
         self.inner.get_meta()
     }
 
+    #[recursive::recursive]
     fn format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         // ignore self
         let input_formatter = self.inner.input.formatter()?;
         input_formatter.dispatch(ctx)
     }
 
+    #[recursive::recursive]
     fn format_join(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.format_join(ctx)
     }
 
+    #[recursive::recursive]
     fn partial_format(&self, ctx: &mut FormatContext<'_>) -> Result<FormatTreeNode<String>> {
         self.inner.input.formatter()?.partial_format(ctx)
     }