reifydb_core/key/
handler.rs

1// SPDX-License-Identifier: Apache-2.0
2// Copyright (c) 2026 ReifyDB
3
4use reifydb_codec::key::{
5	deserializer::KeyDeserializer,
6	encoded::{EncodedKey, EncodedKeyRange},
7	serializer::KeySerializer,
8};
9
10use crate::{
11	interface::catalog::id::HandlerId,
12	key::{EncodableKey, KeyKind},
13};
14
15#[derive(Debug, Clone, PartialEq)]
16pub struct HandlerKey {
17	pub handler: HandlerId,
18}
19
20impl HandlerKey {
21	pub fn new(handler: HandlerId) -> Self {
22		Self {
23			handler,
24		}
25	}
26
27	pub fn encoded(handler: impl Into<HandlerId>) -> EncodedKey {
28		Self::new(handler.into()).encode()
29	}
30
31	pub fn full_scan() -> EncodedKeyRange {
32		EncodedKeyRange::start_end(Some(Self::start()), Some(Self::end()))
33	}
34
35	fn start() -> EncodedKey {
36		let mut serializer = KeySerializer::with_capacity(1);
37		serializer.extend_u8(Self::KIND as u8);
38		serializer.to_encoded_key()
39	}
40
41	fn end() -> EncodedKey {
42		let mut serializer = KeySerializer::with_capacity(1);
43		serializer.extend_u8(Self::KIND as u8 - 1);
44		serializer.to_encoded_key()
45	}
46}
47
48impl EncodableKey for HandlerKey {
49	const KIND: KeyKind = KeyKind::Handler;
50
51	fn encode(&self) -> EncodedKey {
52		let mut serializer = KeySerializer::with_capacity(9);
53		serializer.extend_u8(Self::KIND as u8).extend_u64(self.handler);
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 kind: KeyKind = de.read_u8().ok()?.try_into().ok()?;
61		if kind != Self::KIND {
62			return None;
63		}
64
65		let handler = de.read_u64().ok()?;
66
67		Some(Self {
68			handler: HandlerId(handler),
69		})
70	}
71}
72
73#[cfg(test)]
74pub mod tests {
75	use super::{EncodableKey, HandlerKey};
76	use crate::interface::catalog::id::HandlerId;
77
78	#[test]
79	fn test_encode_decode() {
80		let key = HandlerKey {
81			handler: HandlerId(0xABCD),
82		};
83		let encoded = key.encode();
84		let expected = vec![0xD2, 0x3F, 0x54, 0x32];
85		assert_eq!(encoded.as_slice(), expected);
86
87		let decoded = HandlerKey::decode(&encoded).unwrap();
88		assert_eq!(decoded.handler, HandlerId(0xABCD));
89	}
90}
reifydb_core/key/handler.rs

reifydb_core/key/
handler.rs
