uqa_sql/routines/lifecycle/
relations.rs

1//
2// Unified Query Algebra
3//
4// Copyright (c) 2023-2026 Cognica, Inc.
5//
6
7//! Stored relation, sequence OID, column, and namespace references in routine definitions.
8
9use crate::{
10    ast::{CreateFunction, Expr, FunctionBody},
11    SQLError,
12};
13use std::collections::BTreeSet;
14use uqa_core::Value;
15
16pub trait RoutineRelationOids {
17    fn bound_regclass_oid(&self, name: &str) -> Result<Option<i64>, SQLError>;
18}
19
20pub fn is_regclass(name: &str) -> bool {
21    name.eq_ignore_ascii_case("regclass") || name.eq_ignore_ascii_case("pg_catalog.regclass")
22}
23
24pub fn regclass_oid(expression: &Expr) -> Option<i64> {
25    match expression {
26        Expr::TypedLiteral {
27            value: Value::Int(oid),
28            ty,
29        } if is_regclass(ty) => Some(*oid),
30        Expr::Cast { expr, ty, .. } if is_regclass(ty) => regclass_oid(expr),
31        _ => None,
32    }
33}
34
35pub fn stored_routine_references_relations(
36    catalog: &dyn RoutineRelationOids,
37    definition: &CreateFunction,
38    relations: &BTreeSet<String>,
39) -> Result<bool, SQLError> {
40    if relations.is_empty() {
41        return Ok(false);
42    }
43    let mut oids = BTreeSet::new();
44    for relation in relations {
45        if let Some(oid) = catalog.bound_regclass_oid(relation)? {
46            oids.insert(oid);
47        }
48    }
49    let mut depends = false;
50    let mut visit = |expression: &mut crate::ast::Expr| {
51        depends |= regclass_oid(expression).is_some_and(|oid| oids.contains(&oid));
52        Ok(())
53    };
54    for default in definition
55        .params
56        .iter()
57        .filter_map(|parameter| parameter.default.as_ref())
58    {
59        crate::catalog::stored_ast::visit_stored_expression(&mut default.clone(), &mut visit)?;
60    }
61    if let FunctionBody::Statements(statements) = &definition.body {
62        for statement in statements {
63            if crate::catalog::stored_ast::stored_statement_relation_names(statement)?
64                .iter()
65                .any(|name| relations.contains(name))
66            {
67                return Ok(true);
68            }
69            crate::catalog::stored_ast::visit_stored_statement_expressions(
70                &mut statement.clone(),
71                &mut visit,
72            )?;
73        }
74    }
75    Ok(depends)
76}
uqa_sql/routines/lifecycle/relations.rs

uqa_sql/routines/lifecycle/
relations.rs
