Function restore_core_state

nodedb/src/storage/snapshot_executor.rs:136–161

Restore a single core's engine state from a `CoreSnapshot`. Opens the core's redb databases, clears existing data, and loads the snapshot data. Returns `(documents_count, vectors_count)`.

(
    data_dir: &Path,
    core_id: usize,
    snap: &CoreSnapshot,
)

/// Opens the core's redb databases, clears existing data, and loads
/// the snapshot data. Returns `(documents_count, vectors_count)`.
fn restore_core_state(
    data_dir: &Path,
    core_id: usize,
    snap: &CoreSnapshot,
) -> crate::Result<(u64, u64)> {
    let sparse_path = data_dir.join(format!("sparse/core-{core_id}.redb"));
    if let Some(parent) = sparse_path.parent() {
        // no-objectstore: redb opens a local file directly; warm-tier blob is restored, this is the engine-state landing path.
        std::fs::create_dir_all(parent).map_err(crate::Error::Io)?;
    }
    let sparse = crate::engine::sparse::btree::SparseEngine::open(&sparse_path)?;
    restore_sparse_data(&sparse, &snap.sparse_documents, &snap.sparse_indexes)?;

    let graph_path = data_dir.join(format!("graph/core-{core_id}.redb"));
    if let Some(parent) = graph_path.parent() {
        // no-objectstore: redb opens a local file directly; engine-state landing path.
        std::fs::create_dir_all(parent).map_err(crate::Error::Io)?;
    }
    let edge_store = crate::engine::graph::edge_store::store::EdgeStore::open(&graph_path)?;
    restore_edge_data(&edge_store, &snap.edges)?;

    let vectors = restore_vector_checkpoints(data_dir, core_id, &snap.hnsw_indexes)?;
    restore_crdt_checkpoints(data_dir, core_id, &snap.crdt_snapshots)?;

    Ok((snap.sparse_documents.len() as u64, vectors))
}

fn restore_sparse_data(
    sparse: &crate::engine::sparse::btree::SparseEngine,