nodedb_types/json_msgpack/
writer.rs

1// SPDX-License-Identifier: Apache-2.0
2
3//! Msgpack serialization for `serde_json::Value` and `nodedb_types::Value`.
4
5use super::json_value::JsonValue;
6
7/// Serialize a `serde_json::Value` to MessagePack bytes.
8#[inline]
9pub fn json_to_msgpack(value: &serde_json::Value) -> zerompk::Result<Vec<u8>> {
10    zerompk::to_msgpack_vec(&JsonValue(value.clone()))
11}
12
13/// Serialize a `serde_json::Value` to MessagePack bytes, returning an empty
14/// msgpack map (`0x80`) on failure.
15///
16/// Suitable for filter evaluation where an empty map causes all field
17/// predicates to pass vacuously. Callers that need error propagation
18/// should use [`json_to_msgpack`] instead.
19#[inline]
20pub fn json_to_msgpack_or_empty(value: &serde_json::Value) -> Vec<u8> {
21    json_to_msgpack(value).unwrap_or_else(|_| vec![0x80])
22}
23
24/// Serialize a `nodedb_types::Value` to standard MessagePack bytes.
25///
26/// Writes standard msgpack format (fixmap 0x80-0x8F, fixstr 0xA0-0xBF, etc.)
27/// directly from `Value` — no zerompk tagged encoding.
28pub fn value_to_msgpack(value: &crate::Value) -> zerompk::Result<Vec<u8>> {
29    let mut buf = Vec::with_capacity(128);
30    write_native_value(&mut buf, value);
31    Ok(buf)
32}
33
34/// Write a `nodedb_types::Value` as standard msgpack bytes.
35fn write_native_value(buf: &mut Vec<u8>, value: &crate::Value) {
36    match value {
37        crate::Value::Null => buf.push(0xC0),
38        crate::Value::Bool(false) => buf.push(0xC2),
39        crate::Value::Bool(true) => buf.push(0xC3),
40        crate::Value::Integer(i) => write_native_int(buf, *i),
41        crate::Value::Float(f) => {
42            buf.push(0xCB);
43            buf.extend_from_slice(&f.to_be_bytes());
44        }
45        crate::Value::String(s)
46        | crate::Value::Uuid(s)
47        | crate::Value::Ulid(s)
48        | crate::Value::Regex(s) => write_native_str(buf, s),
49        crate::Value::Bytes(b) => write_native_bin(buf, b),
50        crate::Value::Array(arr) | crate::Value::Set(arr) => {
51            write_native_array_header(buf, arr.len());
52            for v in arr {
53                write_native_value(buf, v);
54            }
55        }
56        crate::Value::Object(map) => {
57            write_native_map_header(buf, map.len());
58            for (k, v) in map {
59                write_native_str(buf, k);
60                write_native_value(buf, v);
61            }
62        }
63        crate::Value::DateTime(dt) | crate::Value::NaiveDateTime(dt) => {
64            write_native_str(buf, &dt.to_string())
65        }
66        crate::Value::Duration(d) => write_native_str(buf, &d.to_string()),
67        crate::Value::Decimal(d) => write_native_str(buf, &d.to_string()),
68        crate::Value::Geometry(g) => {
69            if let Ok(s) = sonic_rs::to_string(g) {
70                write_native_str(buf, &s);
71            } else {
72                buf.push(0xC0);
73            }
74        }
75        crate::Value::Range { .. } | crate::Value::Record { .. } => buf.push(0xC0),
76        crate::Value::Vector(v) => {
77            // Encode as a standard msgpack array of float64 values so that
78            // pgwire clients receive a plain JSON number array.
79            write_native_array_header(buf, v.len());
80            for f in v.iter() {
81                buf.push(0xCB);
82                buf.extend_from_slice(&(*f as f64).to_be_bytes());
83            }
84        }
85        // ArrayCell is encoded as a 2-key map `{coords:[...], attrs:[...]}`
86        // so the pgwire `msgpack_to_json_string` transcoder produces clean
87        // JSON for clients reading slice/project rows.
88        crate::Value::ArrayCell(cell) => {
89            write_native_map_header(buf, 2);
90            write_native_str(buf, "coords");
91            write_native_array_header(buf, cell.coords.len());
92            for v in &cell.coords {
93                write_native_value(buf, v);
94            }
95            write_native_str(buf, "attrs");
96            write_native_array_header(buf, cell.attrs.len());
97            for v in &cell.attrs {
98                write_native_value(buf, v);
99            }
100        }
101    }
102}
103
104fn write_native_int(buf: &mut Vec<u8>, i: i64) {
105    if (0..=0x7F).contains(&i) {
106        buf.push(i as u8);
107    } else if (-32..0).contains(&i) {
108        buf.push(i as u8); // negative fixint
109    } else if i >= i8::MIN as i64 && i <= i8::MAX as i64 {
110        buf.push(0xD0);
111        buf.push(i as i8 as u8);
112    } else if i >= i16::MIN as i64 && i <= i16::MAX as i64 {
113        buf.push(0xD1);
114        buf.extend_from_slice(&(i as i16).to_be_bytes());
115    } else if i >= i32::MIN as i64 && i <= i32::MAX as i64 {
116        buf.push(0xD2);
117        buf.extend_from_slice(&(i as i32).to_be_bytes());
118    } else {
119        buf.push(0xD3);
120        buf.extend_from_slice(&i.to_be_bytes());
121    }
122}
123
124fn write_native_str(buf: &mut Vec<u8>, s: &str) {
125    let len = s.len();
126    if len < 32 {
127        buf.push(0xA0 | len as u8);
128    } else if len <= u8::MAX as usize {
129        buf.push(0xD9);
130        buf.push(len as u8);
131    } else if len <= u16::MAX as usize {
132        buf.push(0xDA);
133        buf.extend_from_slice(&(len as u16).to_be_bytes());
134    } else {
135        buf.push(0xDB);
136        buf.extend_from_slice(&(len as u32).to_be_bytes());
137    }
138    buf.extend_from_slice(s.as_bytes());
139}
140
141fn write_native_bin(buf: &mut Vec<u8>, b: &[u8]) {
142    let len = b.len();
143    if len <= u8::MAX as usize {
144        buf.push(0xC4);
145        buf.push(len as u8);
146    } else if len <= u16::MAX as usize {
147        buf.push(0xC5);
148        buf.extend_from_slice(&(len as u16).to_be_bytes());
149    } else {
150        buf.push(0xC6);
151        buf.extend_from_slice(&(len as u32).to_be_bytes());
152    }
153    buf.extend_from_slice(b);
154}
155
156fn write_native_array_header(buf: &mut Vec<u8>, len: usize) {
157    if len < 16 {
158        buf.push(0x90 | len as u8);
159    } else if len <= u16::MAX as usize {
160        buf.push(0xDC);
161        buf.extend_from_slice(&(len as u16).to_be_bytes());
162    } else {
163        buf.push(0xDD);
164        buf.extend_from_slice(&(len as u32).to_be_bytes());
165    }
166}
167
168fn write_native_map_header(buf: &mut Vec<u8>, len: usize) {
169    if len < 16 {
170        buf.push(0x80 | len as u8);
171    } else if len <= u16::MAX as usize {
172        buf.push(0xDE);
173        buf.extend_from_slice(&(len as u16).to_be_bytes());
174    } else {
175        buf.push(0xDF);
176        buf.extend_from_slice(&(len as u32).to_be_bytes());
177    }
178}
nodedb_types/json_msgpack/writer.rs

nodedb_types/json_msgpack/
writer.rs
