Function plan_set_operation

nodedb-sql/src/planner/union.rs:12–52

Plan a UNION / UNION ALL / INTERSECT / EXCEPT operation.

(
    op: &SetOperator,
    left: &SetExpr,
    right: &SetExpr,
    quantifier: &SetQuantifier,
    catalog: &dyn SqlCatalog,
    functions: &FunctionRegistry,
    temporal: crate::TemporalScope,
)

/// Plan a UNION / UNION ALL / INTERSECT / EXCEPT operation.
pub fn plan_set_operation(
    op: &SetOperator,
    left: &SetExpr,
    right: &SetExpr,
    quantifier: &SetQuantifier,
    catalog: &dyn SqlCatalog,
    functions: &FunctionRegistry,
    temporal: crate::TemporalScope,
) -> Result<SqlPlan> {
    let left_plan = plan_set_expr(left, catalog, functions, temporal)?;
    let right_plan = plan_set_expr(right, catalog, functions, temporal)?;

    match op {
        SetOperator::Union => {
            let distinct = matches!(quantifier, SetQuantifier::Distinct | SetQuantifier::None);
            Ok(SqlPlan::Union {
                inputs: vec![left_plan, right_plan],
                distinct,
            })
        }
        SetOperator::Intersect => {
            let all = matches!(quantifier, SetQuantifier::All);
            Ok(SqlPlan::Intersect {
                left: Box::new(left_plan),
                right: Box::new(right_plan),
                all,
            })
        }
        SetOperator::Except => {
            let all = matches!(quantifier, SetQuantifier::All);
            Ok(SqlPlan::Except {
                left: Box::new(left_plan),
                right: Box::new(right_plan),
                all,
            })
        }
        _ => Err(SqlError::Unsupported {
            detail: format!("set operation: {op}"),
        }),
    }
}

fn plan_set_expr(
    expr: &SetExpr,