polars-plan 0.54.3

Lazy query engine for the Polars DataFrame library
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use polars_utils::pl_str::PlSmallStr;

use super::*;

pub(super) fn process_rename(
    acc_predicates: &mut PlHashMap<PlSmallStr, ExprIR>,
    expr_arena: &mut Arena<AExpr>,
    existing: &[PlSmallStr],
    new: &[PlSmallStr],
) {
    let rename_map: PlHashMap<PlSmallStr, PlSmallStr> =
        new.iter().cloned().zip(existing.iter().cloned()).collect();

    if !rename_map.is_empty() {
        for (_, expr_ir) in acc_predicates.iter_mut() {
            map_column_references(expr_ir, expr_arena, &rename_map);
        }
    }
}