uqa-sql 0.3.6

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

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

//! Durable table input shapes, independent of the columns a statement actually reads.

use super::{
    ColumnBindingContext, ColumnBindingMode, ColumnScope, FromClause, RelationIdentity, SQLError,
    StoredColumnBinder,
};

impl StoredColumnBinder<'_> {
    pub(super) fn bind_table_source(
        &mut self,
        source: &mut FromClause,
        context: &ColumnBindingContext,
    ) -> Result<ColumnScope, SQLError> {
        let FromClause::Table {
            name,
            qualifier,
            alias,
            column_aliases,
            bound_columns,
            ..
        } = source
        else {
            unreachable!("table column binding requires a table source");
        };
        self.bind_table_source_columns(name, bound_columns, context)?;
        let scope = self.table_scope(
            name,
            qualifier,
            alias.as_deref(),
            column_aliases,
            bound_columns.as_deref(),
            context,
        )?;
        self.remove_dropped_column_aliases(column_aliases, &scope);
        if let Some(columns) = bound_columns {
            self.remove_dropped_column_aliases(columns, &scope);
        }
        Ok(scope)
    }

    pub(super) fn bind_table_source_columns(
        &mut self,
        name: &str,
        bound_columns: &mut Option<Vec<String>>,
        context: &ColumnBindingContext,
    ) -> Result<(), SQLError> {
        if context.ctes.contains_key(&name.to_ascii_lowercase()) {
            return Ok(());
        }
        if let (Some(columns), ColumnBindingMode::Rename { relation, from, to }) =
            (bound_columns, self.mode)
        {
            let identity = RelationIdentity::from_legacy_name(name).map_err(SQLError::Internal)?;
            if &identity == relation {
                for column in columns {
                    if column == from {
                        *column = to.to_string();
                        self.alias_shape_changed = true;
                    }
                }
            }
        }
        Ok(())
    }
}