uqa-execution 0.3.0

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
53
54

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

//! Resolve DROP DOMAIN targets in source order before joint domain and routine removal.

use crate::schema::namespaces::NamespaceCatalogRefresh;
use std::collections::BTreeSet;
use uqa_sql::{
    ast::DropStmt,
    schema::domains::removal::{resolve_drop_domain, BoundDomainDrop, DomainDropBinding},
    SQLError,
};

pub trait DomainRoutineRemoval {
    fn remove_domain_types_and_routines(
        &self,
        targets: &BTreeSet<u32>,
        cascade: bool,
    ) -> Result<(), SQLError>;
}
pub trait DomainDropNotices {
    fn domain_drop_notice(&self, message: &str);
}
pub struct DomainRemovalContext<'a> {
    pub refresh: &'a dyn NamespaceCatalogRefresh,
    pub binding: DomainDropBinding<'a>,
    pub removal: &'a dyn DomainRoutineRemoval,
    pub notices: &'a dyn DomainDropNotices,
}

pub fn drop_domains(
    context: &DomainRemovalContext<'_>,
    statement: &DropStmt,
) -> Result<(), SQLError> {
    context
        .refresh
        .refresh_catalog()
        .map_err(|error| SQLError::Internal(error.to_string()))?;
    let mut targets = BTreeSet::new();
    for name in &statement.names {
        match resolve_drop_domain(&context.binding, name, statement.if_exists)? {
            BoundDomainDrop::Target(oid) => {
                targets.insert(oid);
            }
            BoundDomainDrop::Skipped(message) => context.notices.domain_drop_notice(&message),
        }
    }
    context
        .removal
        .remove_domain_types_and_routines(&targets, statement.cascade)
}