feat: implement tree rendering for SortPreservingMergeExec

Signed-off-by: Alan Tang <[email protected]>

Author: Standing-Man

datafusion/physical-plan/src/sorts/sort_preserving_merge.rs

@@ -186,8 +186,15 @@ impl DisplayAs for SortPreservingMergeExec {
                 Ok(())
             }
             DisplayFormatType::TreeRender => {
-                // TODO: collect info
-                write!(f, "")
+                for (i, e) in self.expr().iter().enumerate() {
+                    let e = e.to_string();
+                    writeln!(f, "expr{i}={e}")?;
+                }
+                if let Some(fetch) = self.fetch {
+                    writeln!(f, "fetch={fetch}")?;
+                };
+
+                Ok(())
             }
         }
     }

datafusion/sqllogictest/test_files/explain_tree.slt

@@ -1353,15 +1353,15 @@ ORDER BY "date", "time";
 physical_plan
 01)┌───────────────────────────┐
 02)│  SortPreservingMergeExec  │
-03)└─────────────┬─────────────┘
-04)┌─────────────┴─────────────┐
-05)│    CoalesceBatchesExec    │
-06)└─────────────┬─────────────┘
-07)┌─────────────┴─────────────┐
-08)│         FilterExec        │
-09)│    --------------------   │
-10)│         predicate:        │
-11)│        ticker@1 = A       │
+03)│    --------------------   │
+04)│           expr0:          │
+05)│   date@0 ASC NULLS LAST   │
+06)│                           │
+07)│           expr1:          │
+08)│   time@2 ASC NULLS LAST   │
+09)└─────────────┬─────────────┘
+10)┌─────────────┴─────────────┐
+11)│    CoalesceBatchesExec    │
 12)└─────────────┬─────────────┘
 13)┌─────────────┴─────────────┐
 14)│      RepartitionExec      │
@@ -1389,9 +1389,9 @@ ORDER BY "time"
 physical_plan
 01)┌───────────────────────────┐
 02)│  SortPreservingMergeExec  │
-03)└─────────────┬─────────────┘
-04)┌─────────────┴─────────────┐
-05)│    CoalesceBatchesExec    │
+03)│    --------------------   │
+04)│           expr0:          │
+05)│   time@2 ASC NULLS LAST   │
 06)└─────────────┬─────────────┘
 07)┌─────────────┴─────────────┐
 08)│         FilterExec        │
@@ -1425,9 +1425,9 @@ ORDER BY "date"
 physical_plan
 01)┌───────────────────────────┐
 02)│  SortPreservingMergeExec  │
-03)└─────────────┬─────────────┘
-04)┌─────────────┴─────────────┐
-05)│    CoalesceBatchesExec    │
+03)│    --------------------   │
+04)│           expr0:          │
+05)│   date@0 ASC NULLS LAST   │
 06)└─────────────┬─────────────┘
 07)┌─────────────┴─────────────┐
 08)│         FilterExec        │
@@ -1537,15 +1537,15 @@ ORDER BY "ticker", "time";
 physical_plan
 01)┌───────────────────────────┐
 02)│  SortPreservingMergeExec  │
-03)└─────────────┬─────────────┘
-04)┌─────────────┴─────────────┐
-05)│    CoalesceBatchesExec    │
-06)└─────────────┬─────────────┘
-07)┌─────────────┴─────────────┐
-08)│         FilterExec        │
-09)│    --------------------   │
-10)│         predicate:        │
-11)│    date@0 = 2006-01-02    │
+03)│    --------------------   │
+04)│           expr0:          │
+05)│  ticker@1 ASC NULLS LAST  │
+06)│                           │
+07)│           expr1:          │
+08)│   time@2 ASC NULLS LAST   │
+09)└─────────────┬─────────────┘
+10)┌─────────────┴─────────────┐
+11)│    CoalesceBatchesExec    │
 12)└─────────────┬─────────────┘
 13)┌─────────────┴─────────────┐
 14)│      RepartitionExec      │