uqa-execution 0.4.5

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

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

//! Fresh namespace authority retains only schemas changed by the current transaction.

use std::{collections::BTreeMap, sync::Arc};
use uqa_sql::catalog::{roles::RoleDefinition, security::BoundSchemaSecurity};
use uqa_storage::{CatalogFacade, StorageBackendError, StorageBackendResult};

pub fn merge_private(
    catalog: Option<&dyn CatalogFacade>,
    current: &BTreeMap<String, BoundSchemaSecurity>,
    committed: Arc<BTreeMap<String, BoundSchemaSecurity>>,
    roles: &BTreeMap<String, RoleDefinition>,
) -> StorageBackendResult<Arc<BTreeMap<String, BoundSchemaSecurity>>> {
    merge_private_records(current, committed, roles, |name| {
        catalog.map_or(Ok(false), |catalog| {
            catalog.schema_has_private_changes(name)
        })
    })
}

fn merge_private_records(
    current: &BTreeMap<String, BoundSchemaSecurity>,
    mut committed: Arc<BTreeMap<String, BoundSchemaSecurity>>,
    roles: &BTreeMap<String, RoleDefinition>,
    mut is_private: impl FnMut(&str) -> StorageBackendResult<bool>,
) -> StorageBackendResult<Arc<BTreeMap<String, BoundSchemaSecurity>>> {
    {
        let names = current
            .keys()
            .chain(committed.keys())
            .collect::<std::collections::BTreeSet<_>>();
        let private = names
            .into_iter()
            .filter_map(|name| match is_private(name) {
                Ok(true) => Some(Ok(name.clone())),
                Ok(false) => None,
                Err(error) => Some(Err(error)),
            })
            .collect::<StorageBackendResult<Vec<_>>>()?;
        if !private.is_empty() {
            let merged = Arc::make_mut(&mut committed);
            for name in private {
                if let Some(security) = current.get(&name) {
                    merged.insert(name, security.clone());
                } else {
                    merged.remove(&name);
                }
            }
        }
    }
    for security in committed.values() {
        security
            .validate(roles)
            .map_err(StorageBackendError::Other)?;
    }
    Ok(committed)
}

#[cfg(test)]
mod tests;