Function rewrite_extension_inputs

datafusion/expr/src/logical_plan/tree_node.rs:364–380

Rewrites all inputs for an Extension node "in place" (it currently has to copy values because there are no APIs for in place modification) Should be removed when we have an API for in place modifications of the extension to avoid these copies

(
    extension: Extension,
    f: F,
)

/// Should be removed when we have an API for in place modifications of the
/// extension to avoid these copies
fn rewrite_extension_inputs<F: FnMut(LogicalPlan) -> Result<Transformed<LogicalPlan>>>(
    extension: Extension,
    f: F,
) -> Result<Transformed<Extension>> {
    let Extension { node } = extension;

    node.inputs()
        .into_iter()
        .cloned()
        .map_until_stop_and_collect(f)?
        .map_data(|new_inputs| {
            let exprs = node.expressions();
            Ok(Extension {
                node: node.with_exprs_and_inputs(exprs, new_inputs)?,
            })
        })
}

/// This macro is used to determine continuation during combined transforming
/// traversals.