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

// SPDX-License-Identifier: BUSL-1.1

//! `REINDEX [CONCURRENTLY] collection` — rebuild indexes.
//!
//! Grammar:
//!   REINDEX [INDEX <name>] [CONCURRENTLY] <collection>
//!
//! Non-concurrent path: dispatches `MetaOp::Checkpoint` (existing semantics).
//! Concurrent path: dispatches `MetaOp::RebuildIndex { concurrent: true }` to
//! every core and awaits the cross-core ACK barrier before returning.
//!
//! The grammar is parsed once by `nodedb_sql::ddl_ast::parse` into
//! `NodedbStatement::Reindex { .. }`; this handler receives the already-parsed
//! fields and never re-tokenises the SQL string.

use nodedb_types::DatabaseId;

use crate::control::security::identity::AuthenticatedIdentity;
use crate::control::state::SharedState;
use crate::types::TraceId;
use nodedb_physical::physical_plan::MetaOp;

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

/// Execute a parsed `REINDEX [INDEX name] [CONCURRENTLY] collection` statement.
pub async fn handle_reindex(
    state: &SharedState,
    identity: &AuthenticatedIdentity,
    collection: &str,
    index_name: Option<&str>,
    concurrent: bool,
    database_id: DatabaseId,
) -> Result<Vec<DdlResult>, DdlError> {
    let collection = collection.to_lowercase();
    let index_name = index_name.map(str::to_lowercase);
    let tenant_id = identity.tenant_id;

    // Verify the collection exists.
    if state
        .credentials
        .catalog()
        .get_collection(database_id, tenant_id.as_u64(), &collection)
        .ok()
        .flatten()
        .is_none()
    {
        return Err(ddl_err(
            "42P01",
            format!("collection \"{collection}\" does not exist"),
        ));
    }

    if concurrent {
        // Concurrent path: broadcast to all cores and await per-core ACK.
        let plan = crate::bridge::envelope::PhysicalPlan::Meta(MetaOp::RebuildIndex {
            collection: collection.clone(),
            index_name,
            concurrent: true,
        });
        let trace_id = TraceId::generate();
        crate::control::server::broadcast::broadcast_register_to_all_cores(
            state,
            tenant_id,
            database_id,
            plan,
            trace_id,
        )
        .await
        .map_err(|e| ddl_err("XX000", format!("REINDEX CONCURRENTLY failed: {e}")))?;

        tracing::info!(
            %collection,
            concurrent = true,
            "REINDEX CONCURRENTLY dispatched and acknowledged by all cores"
        );
    } else {
        // Non-concurrent path: fire-and-forget (same as legacy Checkpoint).
        super::distributed::dispatch_maintenance_to_all_cores(state, tenant_id, MetaOp::Checkpoint);
        tracing::info!(%collection, concurrent = false, "REINDEX dispatched");
    }

    Ok(vec![DdlResult::Status {
        command: "REINDEX".to_string(),
        rows_affected: None,
    }])
}