uqa-sql 0.5.2

PostgreSQL-compatible SQL compiler built on libpg_query
Documentation 
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

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

//! Base columns referenced by an executable query block, including aliases and joined output names.

use super::{
    analyze_query_block, BTreeMap, BTreeSet, CteScope, QueryBlockPlan, SQLError, SourceLineage,
};

pub fn required(
    block: &QueryBlockPlan,
    scope: &CteScope<'_>,
) -> Result<BTreeMap<String, BTreeSet<String>>, SQLError> {
    let mut scope = scope.clone();
    scope.include_authorization_only_columns = false;
    let mut universe = SourceLineage::default();
    let mut required = BTreeSet::new();
    analyze_query_block(
        block,
        block.from.as_ref(),
        &scope,
        &[],
        &mut universe,
        &mut required,
    )?;
    Ok(by_relation(required))
}

fn by_relation(required: BTreeSet<super::BaseColumn>) -> BTreeMap<String, BTreeSet<String>> {
    let mut result: BTreeMap<String, BTreeSet<String>> = BTreeMap::new();
    for column in required {
        result
            .entry(column.table)
            .or_default()
            .insert(column.column);
    }
    result
}

/// Columns read through a mutation's additional sources, using the same aliases as SELECT privilege analysis.
pub fn for_source_expressions(
    source: &crate::plan::SourcePlan,
    expressions: &[&crate::ScalarExpr],
    scope: &CteScope<'_>,
) -> Result<BTreeMap<String, BTreeSet<String>>, SQLError> {
    let mut scope = scope.clone();
    scope.include_authorization_only_columns = false;
    let (_, required) = super::source_expression_dependencies(source, expressions, &scope)?;
    Ok(by_relation(required))
}

/// Columns read from a mutation target, including RETURNING's OLD/NEW aliases and correlated subqueries.
pub fn for_target_expressions(
    table: &str,
    qualifiers: &BTreeSet<String>,
    expressions: &[&crate::ScalarExpr],
    scope: &CteScope<'_>,
) -> Result<BTreeMap<String, BTreeSet<String>>, SQLError> {
    let mut scope = scope.clone();
    scope.include_authorization_only_columns = false;
    let (_, required) = super::table_expression_dependencies(
        table,
        qualifiers,
        expressions,
        &scope.scalar_subqueries,
        &[],
        &scope,
    )?;
    Ok(by_relation(required))
}