reifydb_core/key/
namespace_ring_buffer.rs

1// Copyright (c) reifydb.com 2025
2// This file is licensed under the AGPL-3.0-or-later, see license.md file
3
4use 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}