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

// SPDX-License-Identifier: BUSL-1.1

//! Handler for `SHOW DATABASES`.
//!
//! Ported from the pgwire `ddl::database::show` handler. The tenant-admin gate,
//! catalog list, per-database collection count, status mapping, and parent
//! clone rendering are preserved verbatim; only the result construction changed
//! from pgwire `QueryResponse` to the protocol-neutral [`DdlResult`] over
//! `ShapedRows`. Every column is a `text_field` in the original, so all columns
//! stay `Text`.

use serde_json::{Map, Value as JsonValue};

use crate::control::security::catalog::database_types::DatabaseStatus;
use crate::control::security::identity::AuthenticatedIdentity;
use crate::control::state::SharedState;

use super::super::super::result::{DdlError, DdlResult};
use super::gate::require_tenant_admin;
use super::support::{ddl_err, text_rows};

/// Handle `SHOW DATABASES`.
pub fn show_databases(
    state: &SharedState,
    identity: &AuthenticatedIdentity,
) -> Result<Vec<DdlResult>, DdlError> {
    require_tenant_admin(identity, "show databases")?;

    let catalog = state.credentials.catalog();

    let databases = catalog
        .list_databases()
        .map_err(|e| ddl_err("XX000", format!("catalog list failed: {e}")))?;

    let columns = vec![
        "name".to_string(),
        "status".to_string(),
        "created_at_lsn".to_string(),
        "quota_id".to_string(),
        "collection_count".to_string(),
        "tenant_count".to_string(),
        "parent_clone".to_string(),
    ];

    let mut rows: Vec<Map<String, JsonValue>> = Vec::with_capacity(databases.len());
    for db in &databases {
        let status_str = match db.status {
            DatabaseStatus::Active => "active",
            DatabaseStatus::Deactivated => "deactivated",
            DatabaseStatus::Cloning => "cloning",
            DatabaseStatus::Mirroring => "mirroring",
        };

        let collection_count = catalog
            .load_all_collections(db.id)
            .map(|c| c.len())
            .unwrap_or(0);

        let parent_clone = db
            .parent_clone
            .as_ref()
            .map(|p| format!("db:{}", p.source_db_id.as_u64()))
            .unwrap_or_default();

        let mut row = Map::new();
        row.insert("name".to_string(), JsonValue::String(db.name.clone()));
        row.insert(
            "status".to_string(),
            JsonValue::String(status_str.to_string()),
        );
        row.insert(
            "created_at_lsn".to_string(),
            JsonValue::String(db.created_at_lsn.to_string()),
        );
        row.insert(
            "quota_id".to_string(),
            JsonValue::String(db.quota_ref.to_string()),
        );
        row.insert(
            "collection_count".to_string(),
            JsonValue::String(collection_count.to_string()),
        );
        // tenant_count: per-database tenant index not yet wired
        row.insert(
            "tenant_count".to_string(),
            JsonValue::String("0".to_string()),
        );
        row.insert("parent_clone".to_string(), JsonValue::String(parent_clone));
        rows.push(row);
    }

    Ok(text_rows(columns, rows))
}