Display projection

Dandandan · Nov 4, 2023 · 2dcc875 · 2dcc875
1 parent 53b0cea
commit 2dcc875
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/datafusion/expr/src/logical_plan/plan.rs b/datafusion/expr/src/logical_plan/plan.rs
@@ -162,7 +162,9 @@ impl LogicalPlan {
                 projected_schema, ..
             }) => projected_schema,
             LogicalPlan::Projection(Projection { schema, .. }) => schema,
-            LogicalPlan::Filter(Filter { input, .. }) => input.schema(),
+            LogicalPlan::Filter(Filter {
+                projected_schema, ..
+            }) => projected_schema,
             LogicalPlan::Distinct(Distinct { input }) => input.schema(),
             LogicalPlan::Window(Window { schema, .. }) => schema,
             LogicalPlan::Aggregate(Aggregate { schema, .. }) => schema,
@@ -1547,8 +1549,21 @@ impl LogicalPlan {
                     }
                     LogicalPlan::Filter(Filter {
                         predicate: ref expr,
+                        projection,
+                        projected_schema,
                         ..
-                    }) => write!(f, "Filter: {expr}"),
+                    }) => {
+                        if let Some(indices) = projection {
+                            let names: Vec<&str> = indices
+                                .iter()
+                                .map(|i| projected_schema.field(*i).name().as_str())
+                                .collect();
+
+                            write!(f, "Filter: {expr}, projection=[{}]", names.join(", "))
+                        } else {
+                            write!(f, "Filter: {expr}")
+                        }
+                    }
                     LogicalPlan::Window(Window {
                         ref window_expr, ..
                     }) => {