uqa-execution 0.4.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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

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

//! Routine registry snapshots, retained write guards, and durable publication.

pub(crate) mod encoding;

use std::ops::DerefMut;
use uqa_sql::{routines::lifecycle::RoutineRegistry, SQLError};

pub type RoutineRegistryWrite<'a> = Box<dyn DerefMut<Target = RoutineRegistry> + 'a>;
pub trait RoutineRegistryState {
    fn routine_snapshot(&self) -> RoutineRegistry;
    fn routines_write(&self) -> RoutineRegistryWrite<'_>;
}
pub trait RoutineRegistryPublication {
    fn persist_routine_definitions(&self, registry: &RoutineRegistry) -> Result<(), SQLError>;
}

#[derive(Clone, Copy)]
pub struct RoutineMutationContext<'a> {
    pub writer: &'a dyn crate::schema::namespaces::SchemaStatementWriter,
    pub names: &'a dyn uqa_sql::routines::lifecycle::names::RoutineNameCatalog,
    pub roles: &'a dyn crate::catalog::security::roles::RoleCatalogGuards,
    pub registry: &'a dyn RoutineRegistryState,
    pub publication: &'a dyn RoutineRegistryPublication,
    pub changes: &'a dyn crate::schema::namespaces::NamespaceCatalogChanges,
}

use std::{collections::BTreeMap, sync::Arc};
use uqa_sql::{ast::CreateFunction, routines::SQLUserFunction};

pub(crate) const FUNCTIONS_METADATA_KEY: &str = "sql_functions_json";

/// Select one complete metadata record so committed refresh cannot discard private routine definitions or ACL changes.
pub fn merge_private(
    catalog: Option<&dyn uqa_storage::CatalogFacade>,
    current: &Arc<RoutineRegistry>,
    committed: Arc<RoutineRegistry>,
    roles: &BTreeMap<String, uqa_sql::catalog::roles::RoleDefinition>,
) -> uqa_storage::StorageBackendResult<Arc<RoutineRegistry>> {
    let registry = if catalog.map_or(Ok(false), |catalog| {
        catalog.metadata_has_private_changes(FUNCTIONS_METADATA_KEY)
    })? {
        Arc::clone(current)
    } else {
        committed
    };
    for function in registry.values().flatten() {
        uqa_sql::routines::security::binding::validate_routine_authority(&function.def, roles)
            .map_err(|error| uqa_storage::StorageBackendError::Other(error.to_string()))?;
    }
    Ok(registry)
}

pub fn persist_sql_functions_snapshot(
    catalog: Option<&dyn uqa_storage::CatalogFacade>,
    registry: &BTreeMap<String, Vec<Arc<SQLUserFunction>>>,
) -> Result<(), SQLError> {
    for function in registry.values().flatten() {
        uqa_sql::routines::security::binding::validate_routine_authority_identities(&function.def)?;
    }
    let Some(catalog) = catalog else {
        return Ok(());
    };
    let defs: BTreeMap<String, Vec<CreateFunction>> = registry
        .iter()
        .map(|(name, overloads)| {
            (
                name.clone(),
                overloads
                    .iter()
                    .map(|function| function.def.clone())
                    .collect(),
            )
        })
        .collect();
    let json = encoding::encode(defs)
        .map_err(|err| SQLError::Internal(format!("serialize function catalog: {err}")))?;
    catalog
        .set_metadata(FUNCTIONS_METADATA_KEY, &json)
        .map_err(|err| SQLError::Internal(format!("persist function catalog: {err}")))
}