uqa-execution 0.5.1

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

//
// Unified Query Algebra
//
// Copyright (c) 2023-2026 Cognica, Inc.
//

//! References of stored queries, named by OID: each relation of a range table, the columns expressions name, the types of constants and coercions, and bound routines.

use super::{DependencyBuilder, References};
use uqa_sql::plan::QueryPlan;
use uqa_sql::SQLError;

impl DependencyBuilder<'_> {
    /// What `query` references, as `recordDependencyOnExpr` finds it in a rule's action.
    pub(super) fn query_references(&self, query: &QueryPlan) -> Result<References, SQLError> {
        let found =
            super::super::view_definition::query_references(self.catalog, self.resolution, query)?;
        let mut references = References::default();
        for name in &found.relations {
            if let Some(oid) = self.objects.relation_oid_by_name(name) {
                references.add_relation(oid);
            }
        }
        for (relation, column) in &found.columns {
            let Some(oid) = self.objects.relation_oid_by_name(relation) else {
                continue;
            };
            if let Some(number) = self
                .objects
                .relation(oid)
                .and_then(|relation| relation.column_number(column))
            {
                references.add_column(oid, number);
            }
        }
        let expressions = self.expressions();
        for name in &found.types {
            if let Some(oid) = expressions.type_oid(name) {
                references.add_type(oid);
            }
        }
        for binding in &found.routines {
            if let Some(oid) = expressions.routine_oid(binding) {
                references.add_routine(oid);
            }
        }
        for (ty, oid) in &found.constants {
            super::expressions::add_constant_reference(ty, *oid, &mut references);
        }
        Ok(references)
    }
}