reifydb_core/key/
namespace_view.rs1use super::{EncodableKey, KeyKind};
5use crate::{
6 encoded::key::{EncodedKey, EncodedKeyRange},
7 interface::catalog::id::{NamespaceId, ViewId},
8 util::encoding::keycode::{deserializer::KeyDeserializer, serializer::KeySerializer},
9};
10
11#[derive(Debug, Clone, PartialEq)]
12pub struct NamespaceViewKey {
13 pub namespace: NamespaceId,
14 pub view: ViewId,
15}
16
17impl EncodableKey for NamespaceViewKey {
18 const KIND: KeyKind = KeyKind::NamespaceView;
19
20 fn encode(&self) -> EncodedKey {
21 let mut serializer = KeySerializer::with_capacity(17);
22 serializer.extend_u8(Self::KIND as u8).extend_u64(self.namespace).extend_u64(self.view);
23 serializer.to_encoded_key()
24 }
25
26 fn decode(key: &EncodedKey) -> Option<Self> {
27 let mut de = KeyDeserializer::from_bytes(key.as_slice());
28
29 let kind: KeyKind = de.read_u8().ok()?.try_into().ok()?;
30 if kind != Self::KIND {
31 return None;
32 }
33
34 let namespace = de.read_u64().ok()?;
35 let view = de.read_u64().ok()?;
36
37 Some(Self {
38 namespace: NamespaceId(namespace),
39 view: ViewId(view),
40 })
41 }
42}
43
44impl NamespaceViewKey {
45 pub fn encoded(namespace: impl Into<NamespaceId>, view: impl Into<ViewId>) -> EncodedKey {
46 Self {
47 namespace: namespace.into(),
48 view: view.into(),
49 }
50 .encode()
51 }
52
53 pub fn full_scan(namespace_id: NamespaceId) -> EncodedKeyRange {
54 EncodedKeyRange::start_end(Some(Self::link_start(namespace_id)), Some(Self::link_end(namespace_id)))
55 }
56
57 fn link_start(namespace_id: NamespaceId) -> EncodedKey {
58 let mut serializer = KeySerializer::with_capacity(9);
59 serializer.extend_u8(Self::KIND as u8).extend_u64(namespace_id);
60 serializer.to_encoded_key()
61 }
62
63 fn link_end(namespace_id: NamespaceId) -> EncodedKey {
64 let mut serializer = KeySerializer::with_capacity(9);
65 serializer.extend_u8(Self::KIND as u8).extend_u64(*namespace_id - 1);
66 serializer.to_encoded_key()
67 }
68}
69
70#[cfg(test)]
71pub mod tests {
72 use super::{EncodableKey, NamespaceViewKey};
73 use crate::interface::catalog::id::{NamespaceId, ViewId};
74
75 #[test]
76 fn test_encode_decode() {
77 let key = NamespaceViewKey {
78 namespace: NamespaceId(0xABCD),
79 view: ViewId(0x123456789ABCDEF0),
80 };
81 let encoded = key.encode();
82
83 let expected: Vec<u8> =
84 vec![0xEE, 0x3F, 0x54, 0x32, 0x00, 0xED, 0xCB, 0xA9, 0x87, 0x65, 0x43, 0x21, 0x0F];
85
86 assert_eq!(encoded.as_slice(), expected);
87
88 let key = NamespaceViewKey::decode(&encoded).unwrap();
89 assert_eq!(key.namespace, 0xABCD);
90 assert_eq!(key.view, 0x123456789ABCDEF0);
91 }
92
93 #[test]
94 fn test_order_preserving() {
95 let key1 = NamespaceViewKey {
96 namespace: NamespaceId::SYSTEM,
97 view: ViewId(100),
98 };
99 let key2 = NamespaceViewKey {
100 namespace: NamespaceId::SYSTEM,
101 view: ViewId(200),
102 };
103 let key3 = NamespaceViewKey {
104 namespace: NamespaceId::DEFAULT,
105 view: ViewId(1),
106 };
107
108 let encoded1 = key1.encode();
109 let encoded2 = key2.encode();
110 let encoded3 = key3.encode();
111
112 assert!(encoded3 < encoded2, "ordering not preserved");
113 assert!(encoded2 < encoded1, "ordering not preserved");
114 }
115}