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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

// SPDX-License-Identifier: BUSL-1.1

//! KV field-level operation handlers: FieldGet, FieldSet.

use tracing::debug;

use crate::bridge::envelope::{ErrorCode, Response};
use crate::data::executor::core_loop::CoreLoop;
use crate::data::executor::response_codec;
use crate::data::executor::task::ExecutionTask;
use crate::engine::kv::current_ms;

impl CoreLoop {
    pub(in crate::data::executor) fn execute_kv_field_get(
        &self,
        task: &ExecutionTask,
        did: u64,
        tid: u64,
        collection: &str,
        key: &[u8],
        fields: &[String],
    ) -> Response {
        debug!(core = self.core_id, %collection, field_count = fields.len(), "kv field get");
        let now_ms = current_ms();

        // Read-your-own-writes: consult this transaction's staging overlay
        // before falling back to base storage (see `execute_kv_batch_get`).
        let value = match self.kv_overlay_body(task, tid, collection, key) {
            Some(overlay_result) => overlay_result,
            None => self.kv_engine.get(did, tid, collection, key, now_ms),
        };
        let Some(value) = value else {
            return self.response_error(task, ErrorCode::NotFound);
        };

        // Decode as standard msgpack map.
        let doc = match nodedb_types::json_from_msgpack(&value) {
            Ok(serde_json::Value::Object(map)) => map,
            _ => {
                return self.response_error(
                    task,
                    ErrorCode::Internal {
                        detail: "value is not a msgpack-encoded object".into(),
                    },
                );
            }
        };

        // Extract requested fields.
        let mut result = serde_json::Map::new();
        for f in fields {
            let v = doc.get(f).cloned().unwrap_or(serde_json::Value::Null);
            result.insert(f.clone(), v);
        }

        match response_codec::encode_json(&serde_json::Value::Object(result)) {
            Ok(payload) => self.response_with_payload(task, payload),
            Err(e) => self.response_error(
                task,
                ErrorCode::Internal {
                    detail: e.to_string(),
                },
            ),
        }
    }

    pub(in crate::data::executor) fn execute_kv_field_set(
        &mut self,
        ctx: super::atomic::KvAtomicCtx<'_>,
        updates: &[(String, Vec<u8>)],
    ) -> Response {
        let super::atomic::KvAtomicCtx {
            task,
            did,
            tid,
            collection,
            key,
            surrogate,
        } = ctx;
        debug!(core = self.core_id, %collection, field_count = updates.len(), "kv field set");
        let now_ms = current_ms();

        // Read current value.
        let current = self.kv_engine.get(did, tid, collection, key, now_ms);

        // Merge field updates via the pure computation shared with the
        // in-transaction staging handler (`stage_kv_transfer.rs`), so a
        // staged value and its COMMIT-time durable replay never diverge.
        let computed = match super::field_compute::merge_field_updates(current.as_deref(), updates)
        {
            Ok(c) => c,
            Err(e) => return self.response_error(task, e),
        };

        self.kv_engine.put(crate::engine::kv::KvPutParams {
            database_id: did,
            tenant_id: tid,
            collection,
            key,
            value: &computed.new_value,
            ttl_ms: 0,
            now_ms,
            surrogate,
        });
        self.note_kv_write_lsn(task, did, tid, collection, key);
        match response_codec::encode_json(
            &serde_json::json!({ "fields_added": computed.fields_added }),
        ) {
            Ok(payload) => self.response_with_payload(task, payload),
            Err(e) => self.response_error(
                task,
                ErrorCode::Internal {
                    detail: e.to_string(),
                },
            ),
        }
    }
}