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

// SPDX-License-Identifier: BUSL-1.1

//! `pg_attribute` row producer — one row per collection field.

use std::collections::HashMap;

use nodedb_types::Value;

use crate::control::security::identity::AuthenticatedIdentity;
use crate::control::server::pgwire::catalog::oid::stable_collection_oid;
use crate::control::state::SharedState;

use super::collections::{encode_row, field_type_to_oid, load_collections, type_oid_is_collatable};

pub fn pg_attribute(
    state: &SharedState,
    identity: &AuthenticatedIdentity,
) -> crate::Result<Vec<Vec<u8>>> {
    let mut rows = Vec::new();
    for coll in load_collections(state, identity) {
        let rel_oid = stable_collection_oid(coll.tenant_id, &coll.name);
        for (col_num, (field_name, field_type)) in coll.fields.iter().enumerate() {
            let (_, is_primary_key, is_not_null, default_expr) =
                nodedb_sql::ddl_ast::collection_type::parse_column_type_str_full(field_type);
            let type_oid = field_type_to_oid(field_type);
            let mut r: HashMap<String, Value> = HashMap::with_capacity(20);
            r.insert("attrelid".into(), Value::Integer(rel_oid));
            r.insert("attname".into(), Value::String(field_name.clone()));
            r.insert("atttypid".into(), Value::Integer(type_oid));
            r.insert("attstattarget".into(), Value::Integer(-1));
            r.insert("attlen".into(), Value::Integer(-1));
            r.insert("attnum".into(), Value::Integer((col_num + 1) as i64));
            r.insert("attndims".into(), Value::Integer(0));
            r.insert("attcacheoff".into(), Value::Integer(-1));
            r.insert("atttypmod".into(), Value::Integer(-1));
            r.insert("attbyval".into(), Value::Bool(false));
            r.insert("attstorage".into(), Value::String("p".into()));
            r.insert("attalign".into(), Value::String("i".into()));
            r.insert(
                "attnotnull".into(),
                Value::Bool(is_primary_key || is_not_null),
            );
            r.insert("atthasdef".into(), Value::Bool(default_expr.is_some()));
            r.insert("attidentity".into(), Value::String(String::new()));
            r.insert("attgenerated".into(), Value::String(String::new()));
            r.insert("attisdropped".into(), Value::Bool(false));
            r.insert("attislocal".into(), Value::Bool(true));
            r.insert("attinhcount".into(), Value::Integer(0));
            r.insert(
                "attcollation".into(),
                Value::Integer(if type_oid_is_collatable(type_oid) {
                    100
                } else {
                    0
                }),
            );
            rows.push(encode_row(r)?);
        }
    }
    Ok(rows)
}