uqa-sql 0.5.0

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

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

//! Atomic removal of resolved routines from the routine registry.

use super::{RoutineDropTarget, RoutineRegistry};
use crate::SQLError;

/// Remove the routines `targets` names from `next`. Every target is found before any is removed, so a routine registered after the targets were resolved is retained, a routine re-created with a resolved signature under another identity is not removed, and a removal whose target has disappeared removes nothing.
pub fn remove_routine_registry_targets(
    next: &mut RoutineRegistry,
    targets: &[RoutineDropTarget],
) -> Result<(), SQLError> {
    for target in targets {
        if !next
            .get(&target.name)
            .is_some_and(|overloads| overloads.iter().any(|function| target.names(function)))
        {
            return Err(SQLError::Internal(format!(
                "resolved {} {} disappeared before DROP",
                target.kind(),
                target.label()
            )));
        }
    }
    for target in targets {
        if let Some(overloads) = next.get_mut(&target.name) {
            overloads.retain(|function| !target.names(function));
            if overloads.is_empty() {
                next.remove(&target.name);
            }
        }
    }
    Ok(())
}

#[cfg(test)]
mod tests;