nodedb 0.4.0

Local-first, real-time, edge-to-cloud hybrid database for multi-modal workloads
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
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

// SPDX-License-Identifier: BUSL-1.1

//! Catalog-local owner migration shared by DDL fallback and startup repair.

use super::auth_types::{StoredOwner, object_type};
use super::{SystemCatalog, catalog_err};

impl SystemCatalog {
    /// Rewrite both the primary record's in-band owner and its `StoredOwner`
    /// row. Unknown kinds and missing primaries fail closed. Index ownership is
    /// represented solely by `StoredOwner` and therefore has no primary row.
    pub fn rewrite_object_owner(
        &self,
        kind: &str,
        database_id: u64,
        tenant_id: u64,
        name: &str,
        new_owner: &str,
    ) -> crate::Result<()> {
        match kind {
            object_type::COLLECTION => {
                let database_id = nodedb_types::DatabaseId::new(database_id);
                let mut stored = self
                    .get_collection(database_id, tenant_id, name)?
                    .ok_or_else(|| missing(kind, tenant_id, name))?;
                stored.owner = new_owner.to_string();
                self.put_collection(database_id, &stored)?;
            }
            object_type::FUNCTION => {
                let mut stored = self
                    .get_function(tenant_id, name)?
                    .ok_or_else(|| missing(kind, tenant_id, name))?;
                stored.owner = new_owner.to_string();
                self.put_function(&stored)?;
            }
            object_type::PROCEDURE => {
                let mut stored = self
                    .get_procedure(tenant_id, name)?
                    .ok_or_else(|| missing(kind, tenant_id, name))?;
                stored.owner = new_owner.to_string();
                self.put_procedure(&stored)?;
            }
            object_type::TRIGGER => {
                let mut stored = self
                    .get_trigger(tenant_id, name)?
                    .ok_or_else(|| missing(kind, tenant_id, name))?;
                stored.owner = new_owner.to_string();
                self.put_trigger(&stored)?;
            }
            object_type::MATERIALIZED_VIEW => {
                let mut stored = self
                    .get_materialized_view(tenant_id, name)?
                    .ok_or_else(|| missing(kind, tenant_id, name))?;
                stored.owner = new_owner.to_string();
                self.put_materialized_view(&stored)?;
            }
            object_type::SEQUENCE => {
                let mut stored = self
                    .get_sequence(tenant_id, name)?
                    .ok_or_else(|| missing(kind, tenant_id, name))?;
                stored.owner = new_owner.to_string();
                self.put_sequence(&stored)?;
            }
            object_type::SCHEDULE => {
                let mut stored = self
                    .load_all_schedules()?
                    .into_iter()
                    .find(|stored| stored.tenant_id == tenant_id && stored.name == name)
                    .ok_or_else(|| missing(kind, tenant_id, name))?;
                stored.owner = new_owner.to_string();
                self.put_schedule(&stored)?;
            }
            object_type::CHANGE_STREAM => {
                let mut stored = self
                    .get_change_stream(tenant_id, name)?
                    .ok_or_else(|| missing(kind, tenant_id, name))?;
                stored.owner = new_owner.to_string();
                self.put_change_stream(&stored)?;
            }
            object_type::CONTINUOUS_AGGREGATE => {
                let mut stored = self
                    .get_continuous_aggregate(database_id, tenant_id, name)?
                    .ok_or_else(|| missing(kind, tenant_id, name))?;
                stored.owner = new_owner.to_string();
                self.put_continuous_aggregate(&stored)?;
            }
            object_type::INDEX => {}
            unknown => {
                return Err(catalog_err(
                    "rewrite object owner",
                    format!("unknown owner kind '{unknown}'"),
                ));
            }
        }

        self.put_owner(&StoredOwner {
            database_id,
            object_type: kind.to_string(),
            object_name: name.to_string(),
            tenant_id,
            owner_username: new_owner.to_string(),
        })
    }
}

fn missing(kind: &str, tenant_id: u64, name: &str) -> crate::Error {
    catalog_err(
        "rewrite object owner",
        format!("{kind} '{tenant_id}:{name}' has no primary record"),
    )
}