reifydb_core/key/
namespace_ring_buffer.rs1use super::{EncodableKey, KeyKind};
5use crate::{
6 EncodedKey, EncodedKeyRange,
7 interface::{NamespaceId, RingBufferId},
8 util::encoding::keycode::{KeyDeserializer, KeySerializer},
9};
10
11const VERSION: u8 = 1;
12
13#[derive(Debug, Clone, PartialEq)]
14pub struct NamespaceRingBufferKey {
15 pub namespace: NamespaceId,
16 pub ring_buffer: RingBufferId,
17}
18
19impl NamespaceRingBufferKey {
20 pub fn new(namespace: NamespaceId, ring_buffer: RingBufferId) -> Self {
21 Self {
22 namespace,
23 ring_buffer,
24 }
25 }
26
27 pub fn full_scan(namespace: NamespaceId) -> EncodedKeyRange {
28 EncodedKeyRange::start_end(Some(Self::link_start(namespace)), Some(Self::link_end(namespace)))
29 }
30
31 fn link_start(namespace: NamespaceId) -> EncodedKey {
32 let mut serializer = KeySerializer::with_capacity(10);
33 serializer.extend_u8(VERSION).extend_u8(Self::KIND as u8).extend_u64(namespace);
34 serializer.to_encoded_key()
35 }
36
37 fn link_end(namespace: NamespaceId) -> EncodedKey {
38 let mut serializer = KeySerializer::with_capacity(10);
39 serializer.extend_u8(VERSION).extend_u8(Self::KIND as u8).extend_u64(*namespace - 1);
40 serializer.to_encoded_key()
41 }
42}
43
44impl EncodableKey for NamespaceRingBufferKey {
45 const KIND: KeyKind = KeyKind::NamespaceRingBuffer;
46
47 fn encode(&self) -> EncodedKey {
48 let mut serializer = KeySerializer::with_capacity(18);
49 serializer
50 .extend_u8(VERSION)
51 .extend_u8(Self::KIND as u8)
52 .extend_u64(self.namespace)
53 .extend_u64(self.ring_buffer);
54 serializer.to_encoded_key()
55 }
56
57 fn decode(key: &EncodedKey) -> Option<Self> {
58 let mut de = KeyDeserializer::from_bytes(key.as_slice());
59
60 let version = de.read_u8().ok()?;
61 if version != VERSION {
62 return None;
63 }
64
65 let kind: KeyKind = de.read_u8().ok()?.try_into().ok()?;
66 if kind != Self::KIND {
67 return None;
68 }
69
70 let namespace = de.read_u64().ok()?;
71 let ring_buffer = de.read_u64().ok()?;
72
73 Some(Self {
74 namespace: NamespaceId(namespace),
75 ring_buffer: RingBufferId(ring_buffer),
76 })
77 }
78}