Use simple first check

stijnherfst · Nov 25, 2024 · e28655c · e28655c
1 parent ab2d8e6
commit e28655c
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/crates/polars-plan/src/plans/optimizer/projection_pushdown/joins.rs b/crates/polars-plan/src/plans/optimizer/projection_pushdown/joins.rs
@@ -20,9 +20,16 @@ fn add_keys_to_accumulated_state(
     // the JOIN executor
     if add_local {
         // return the left most name as output name
-        let names = aexpr_to_leaf_names_iter(expr, expr_arena).collect::<Vec<_>>();
-        let output_name = names.first().cloned();
+        let names = aexpr_to_leaf_names_iter(expr, expr_arena);
+
+        let mut first = true;
+        let mut output_name = None;
         for name in names {
+            if first {
+                output_name = Some(name.clone());
+                first = false;
+            }
+
             let node = expr_arena.add(AExpr::Column(name));
             local_projection.push(ColumnNode(node));
         }