uqa-execution 0.3.0

Volcano physical operators with row-batch pipelines
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

//
// Unified Query Algebra
//
// Copyright (c) 2023-2026 Cognica, Inc.
//
//! Recursive physical operator assembly for FROM sources.

/// Build a complete FROM source as a pull-based physical operator. Unlike the compatibility `build_join_rows_*` entry points below, this is the query executor's primary path and never collects a join, view, CTE, derived table, or table-function result into a cardinality-sized `Vec`.
use super::{
    attach_qualifier_filter, build_function_group_source_operator, build_function_source_operator,
    build_join_source_operator, build_subquery_source_operator, build_table_source_operator,
    build_values_source_operator, qualifier_filter, ColumnPrune, CteScope, QualifierFilters,
    SQLError, SQLParam, SourceContext, SourcePlan,
};

pub fn build_join_operator_with_ctes<'a, S: Clone + Send + Sync + 'static>(
    context: &SourceContext<'a, S>,
    from: &SourcePlan,
    params: &'a [SQLParam],
    ctes: &mut CteScope<S>,
    prune: Option<&ColumnPrune>,
    filters: Option<&QualifierFilters>,
) -> Result<Box<dyn crate::PhysicalOperator + 'a>, SQLError> {
    build_join_operator_with_ctes_at_path(context, from, params, ctes, prune, filters, None)
}

pub fn build_join_operator_with_recheck_pins<'a, S: Clone + Send + Sync + 'static>(
    context: &SourceContext<'a, S>,
    from: &SourcePlan,
    params: &'a [SQLParam],
    ctes: &mut CteScope<S>,
    prune: Option<&ColumnPrune>,
    filters: Option<&QualifierFilters>,
) -> Result<Box<dyn crate::PhysicalOperator + 'a>, SQLError> {
    build_join_operator_with_ctes_at_path(
        context,
        from,
        params,
        ctes,
        prune,
        filters,
        Some(Vec::new()),
    )
}

/// Recursively assemble a source tree while tracking tuple-recheck paths.
pub(super) fn build_join_operator_with_ctes_at_path<'a, S: Clone + Send + Sync + 'static>(
    context: &SourceContext<'a, S>,
    from: &SourcePlan,
    params: &'a [SQLParam],
    ctes: &mut CteScope<S>,
    prune: Option<&ColumnPrune>,
    filters: Option<&QualifierFilters>,
    recheck_path: Option<Vec<u8>>,
) -> Result<Box<dyn crate::PhysicalOperator + 'a>, SQLError> {
    use crate::PhysicalOperator;

    if let Some(source) = recheck_path
        .as_deref()
        .and_then(|path| ctes.recheck_source_row(path))
    {
        let scan: Box<dyn PhysicalOperator + 'a> = Box::new(crate::TableScan::from_physical_rows(
            source.schema,
            vec![source.row],
        ));
        if matches!(from, SourcePlan::Values { .. })
            || qualifier_filter(filters, &source.qualifier)
                .is_some_and(|predicate| uqa_sql::semantics::contains_retrieval(&predicate))
        {
            return Ok(scan);
        }
        return Ok(attach_qualifier_filter(
            scan,
            &source.qualifier,
            filters,
            context,
            params,
            ctes,
        ));
    }

    match from {
        SourcePlan::Table { .. } => {
            build_table_source_operator(context, from, params, ctes, prune, filters)
        }
        SourcePlan::Join { .. } => {
            build_join_source_operator(context, from, params, ctes, prune, filters, recheck_path)
        }
        SourcePlan::Values { .. } => {
            build_values_source_operator(context, from, params, ctes, prune)
        }
        SourcePlan::Function { .. } => {
            build_function_source_operator(context, from, params, ctes, prune, filters)
        }
        SourcePlan::FunctionGroup { .. } => {
            build_function_group_source_operator(context, from, params, ctes, prune, filters)
        }
        SourcePlan::Subquery { .. } => {
            build_subquery_source_operator(context, from, params, ctes, prune, filters)
        }
    }
}