reifydb_core/key/
namespace_binding.rs1use super::{EncodableKey, KeyKind};
5use crate::{
6 encoded::key::{EncodedKey, EncodedKeyRange},
7 interface::catalog::id::{BindingId, NamespaceId},
8 util::encoding::keycode::{deserializer::KeyDeserializer, serializer::KeySerializer},
9};
10
11#[derive(Debug, Clone, PartialEq)]
12pub struct NamespaceBindingKey {
13 pub namespace: NamespaceId,
14 pub binding: BindingId,
15}
16
17impl EncodableKey for NamespaceBindingKey {
18 const KIND: KeyKind = KeyKind::NamespaceBinding;
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.binding);
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 binding = de.read_u64().ok()?;
36
37 Some(Self {
38 namespace: NamespaceId(namespace),
39 binding: BindingId(binding),
40 })
41 }
42}
43
44impl NamespaceBindingKey {
45 pub fn encoded(namespace: impl Into<NamespaceId>, binding: impl Into<BindingId>) -> EncodedKey {
46 Self {
47 namespace: namespace.into(),
48 binding: binding.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, NamespaceBindingKey};
73 use crate::interface::catalog::id::{BindingId, NamespaceId};
74
75 #[test]
76 fn test_encode_decode() {
77 let key = NamespaceBindingKey {
78 namespace: NamespaceId(0xABCD),
79 binding: BindingId(0x123456789ABCDEF0),
80 };
81 let encoded = key.encode();
82 let decoded = NamespaceBindingKey::decode(&encoded).unwrap();
83 assert_eq!(decoded.namespace, NamespaceId(0xABCD));
84 assert_eq!(decoded.binding, BindingId(0x123456789ABCDEF0));
85 assert_eq!(key, decoded);
86 }
87}