Implement tree explain for AggregateExec

Author: zebsme

datafusion/physical-plan/src/aggregates/mod.rs

@@ -809,8 +809,60 @@ impl DisplayAs for AggregateExec {
                 }
             }
             DisplayFormatType::TreeRender => {
-                // TODO: collect info
-                write!(f, "")?;
+                let g: Vec<String> = if self.group_by.is_single() {
+                    self.group_by
+                        .expr
+                        .iter()
+                        .map(|(e, alias)| {
+                            let e = e.to_string();
+                            if &e != alias {
+                                format!("{e} as {alias}")
+                            } else {
+                                e
+                            }
+                        })
+                        .collect()
+                } else {
+                    self.group_by
+                        .groups
+                        .iter()
+                        .map(|group| {
+                            let terms = group
+                                .iter()
+                                .enumerate()
+                                .map(|(idx, is_null)| {
+                                    if *is_null {
+                                        let (e, alias) = &self.group_by.null_expr[idx];
+                                        let e = e.to_string();
+                                        if &e != alias {
+                                            format!("{e} as {alias}")
+                                        } else {
+                                            e
+                                        }
+                                    } else {
+                                        let (e, alias) = &self.group_by.expr[idx];
+                                        let e = e.to_string();
+                                        if &e != alias {
+                                            format!("{e} as {alias}")
+                                        } else {
+                                            e
+                                        }
+                                    }
+                                })
+                                .collect::<Vec<String>>()
+                                .join(", ");
+                            format!("({terms})")
+                        })
+                        .collect()
+                };
+                let a: Vec<String> = self
+                    .aggr_expr
+                    .iter()
+                    .map(|agg| agg.name().to_string())
+                    .collect();
+                writeln!(f, "mode={:?}", self.mode)?;
+                writeln!(f, "group_by={}", g.join(", "))?;
+                writeln!(f, "aggr={}", a.join(", "))?;
             }
         }
         Ok(())

datafusion/sqllogictest/test_files/explain_tree.slt

@@ -166,25 +166,42 @@ explain SELECT string_col, SUM(bigint_col) FROM table1 GROUP BY string_col;
 physical_plan
 01)┌───────────────────────────┐
 02)│       AggregateExec       │
-03)└─────────────┬─────────────┘
-04)┌─────────────┴─────────────┐
-05)│    CoalesceBatchesExec    │
-06)└─────────────┬─────────────┘
-07)┌─────────────┴─────────────┐
-08)│      RepartitionExec      │
-09)└─────────────┬─────────────┘
-10)┌─────────────┴─────────────┐
-11)│       AggregateExec       │
+03)│    --------------------   │
+04)│           aggr:           │
+05)│   sum(table1.bigint_col)  │
+06)│                           │
+07)│         group_by:         │
+08)│ string_col@0 as string_col│
+09)│                           │
+10)│           mode:           │
+11)│      FinalPartitioned     │
 12)└─────────────┬─────────────┘
 13)┌─────────────┴─────────────┐
-14)│      RepartitionExec      │
+14)│    CoalesceBatchesExec    │
 15)└─────────────┬─────────────┘
 16)┌─────────────┴─────────────┐
-17)│       DataSourceExec      │
-18)│    --------------------   │
-19)│          files: 1         │
-20)│        format: csv        │
-21)└───────────────────────────┘
+17)│      RepartitionExec      │
+18)└─────────────┬─────────────┘
+19)┌─────────────┴─────────────┐
+20)│       AggregateExec       │
+21)│    --------------------   │
+22)│           aggr:           │
+23)│   sum(table1.bigint_col)  │
+24)│                           │
+25)│         group_by:         │
+26)│ string_col@0 as string_col│
+27)│                           │
+28)│       mode: Partial       │
+29)└─────────────┬─────────────┘
+30)┌─────────────┴─────────────┐
+31)│      RepartitionExec      │
+32)└─────────────┬─────────────┘
+33)┌─────────────┴─────────────┐
+34)│       DataSourceExec      │
+35)│    --------------------   │
+36)│          files: 1         │
+37)│        format: csv        │
+38)└───────────────────────────┘
 
 # Limit
 query TT