nodedb 0.3.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

// SPDX-License-Identifier: BUSL-1.1

use pgwire::api::results::Response;
use pgwire::error::PgWireResult;

use crate::control::security::identity::AuthenticatedIdentity;
use crate::control::state::SharedState;

pub(super) async fn dispatch(
    state: &SharedState,
    identity: &AuthenticatedIdentity,
    _sql: &str,
    upper: &str,
    parts: &[&str],
) -> Option<PgWireResult<Vec<Response>>> {
    // User management.
    // CREATE USER and ALTER USER are fully dispatched via typed AST (ast.rs).
    if upper.starts_with("DROP USER ") {
        return Some(super::super::user::drop_user(state, identity, parts));
    }

    // Service accounts.
    if upper.starts_with("CREATE SERVICE ACCOUNT ") {
        return Some(super::super::service_account::create_service_account(
            state, identity, parts,
        ));
    }
    if upper.starts_with("DROP SERVICE ACCOUNT ") {
        return Some(super::super::service_account::drop_service_account(
            state, identity, parts,
        ));
    }
    if upper.starts_with("ALTER SERVICE ACCOUNT ") {
        return Some(
            super::super::service_account_alter::alter_service_account_set_databases(
                state, identity, parts,
            ),
        );
    }

    // System-level settings (ALTER SYSTEM SET ...).
    if upper.starts_with("ALTER SYSTEM ") {
        return Some(super::super::system_ddl::alter_system(
            state, identity, parts,
        ));
    }

    // Tenant management.
    if upper.starts_with("CREATE TENANT ") {
        return Some(super::super::tenant::create_tenant(state, identity, parts));
    }
    if upper.starts_with("ALTER TENANT ") {
        return Some(super::super::tenant::alter_tenant(state, identity, parts));
    }
    if upper.starts_with("DROP TENANT ") {
        return Some(super::super::tenant::drop_tenant(state, identity, parts));
    }
    if upper.starts_with("PURGE TENANT ") {
        return Some(super::super::tenant::purge_tenant(state, identity, parts).await);
    }
    if upper.starts_with("SHOW TENANT USAGE") {
        return Some(super::super::tenant::show_tenant_usage(
            state, identity, parts,
        ));
    }
    if upper.starts_with("SHOW TENANT QUOTA") {
        return Some(super::super::tenant::show_tenant_quota(
            state, identity, parts,
        ));
    }

    // GRANT / REVOKE (role-membership and object-permission) are parsed
    // into typed `AuthStmt` variants and dispatched via the AST router —
    // no string-prefix branch is needed here.

    // Role management.
    if upper.starts_with("CREATE ROLE ") {
        return Some(super::super::role::create_role(state, identity, parts));
    }
    if upper.starts_with("ALTER ROLE ") {
        return Some(super::super::role::alter_role(state, identity, parts));
    }
    if upper.starts_with("DROP ROLE ") {
        return Some(super::super::role::drop_role(state, identity, parts));
    }

    None
}