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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

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

//! Bind every owned key index to its independent constraint and physical identities.

use super::{
    index_definition, invalid, metadata_error, same_row, BTreeMap, CatalogIndexRow,
    CatalogObjectAllocator, CatalogOidClass, ColumnDef, IndexCatalogIdentity, IndexDefinition,
    IndexKey, IndexRegistryChange, RelationIdentity, StorageBackendResult, TableConstraintSet,
};

pub(in crate::schema::indexes) fn prepare(
    catalog: &crate::catalog::CatalogReadView,
    table: &str,
    table_object_id: [u8; 16],
    columns: &[ColumnDef],
    constraints: &TableConstraintSet,
    allocator: &mut dyn CatalogObjectAllocator,
) -> StorageBackendResult<IndexRegistryChange> {
    let relation = RelationIdentity::from_legacy_name(table).map_err(invalid)?;
    let previous = &catalog.snapshot().definitions.catalog_indexes;
    let mut owned = BTreeMap::new();
    for row in previous.values().filter(|row| row.table_name == table) {
        let definition = index_definition(row)?;
        if let Some(owner) = definition.relationships.owning_constraint {
            if owned.insert(owner, (row, definition)).is_some() {
                return Err(invalid("constraint owns more than one index"));
            }
        }
    }
    let mut change = IndexRegistryChange::default();
    for key in &constraints.key_constraints {
        let owner = key
            .catalog_identity
            .ok_or_else(|| invalid("index owner has no constraint identity"))?;
        let name = key
            .name
            .as_ref()
            .ok_or_else(|| invalid("index owner has no constraint name"))?;
        let name = RelationIdentity::new(&relation.schema, name);
        let mut definition = IndexDefinition::for_constraint(key, columns).map_err(invalid)?;
        let previous_owned = owned.remove(&owner.object_id);
        let retained = previous_owned
            .as_ref()
            .map(|(_, definition)| {
                definition
                    .catalog
                    .clone()
                    .ok_or_else(|| invalid("owned index has no catalog identity"))
            })
            .transpose()?;
        if let Some(existing) = previous.get(&name) {
            if previous_owned
                .as_ref()
                .is_none_or(|(row, _)| row.relation != existing.relation)
            {
                return Err(invalid(format!(
                    "constraint index `{}` conflicts with another relation",
                    name.qualified_name()
                )));
            }
        }
        let identity = if let Some(identity) = retained {
            identity.validate(table_object_id).map_err(invalid)?;
            allocator
                .include_catalog_identity(
                    &previous_owned.as_ref().expect("retained index").0.relation,
                    CatalogOidClass::Relation,
                    identity.identity,
                )
                .map_err(metadata_error)?;
            definition.relationships.parent_index = previous_owned
                .as_ref()
                .and_then(|(_, definition)| definition.relationships.parent_index);
            identity
        } else {
            IndexCatalogIdentity::allocate(table_object_id, allocator).map_err(metadata_error)?
        };
        definition
            .relationships
            .validate(identity.identity.object_id)
            .map_err(invalid)?;
        definition.catalog = Some(identity);
        let row = CatalogIndexRow {
            relation: name,
            table_name: table.to_string(),
            index_type: if key.without_overlaps {
                "gist"
            } else {
                "btree"
            }
            .into(),
            columns_json: serde_json::to_string(
                &key.columns
                    .iter()
                    .cloned()
                    .map(IndexKey::Column)
                    .collect::<Vec<_>>(),
            )?,
            parameters_json: "{}".into(),
            definition_json: Some(serde_json::to_string(&definition)?),
        };
        if let Some((old, _)) = previous_owned {
            if old.relation != row.relation {
                change.removals.push(old.clone());
            }
            if same_row(old, &row) {
                continue;
            }
        }
        change.upserts.push(row);
    }
    change
        .removals
        .extend(owned.into_values().map(|(row, _)| row.clone()));
    Ok(change)
}