reifydb_core/key/
column.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::ColumnId, object::ObjectId},
12	key::{
13		EncodableKey, KeyKind,
14		catalog::{KeyDeserializerCatalogExt, KeySerializerCatalogExt},
15	},
16};
17
18#[derive(Debug, Clone, PartialEq)]
19pub struct ColumnKey {
20	pub object: ObjectId,
21	pub column: ColumnId,
22}
23
24impl EncodableKey for ColumnKey {
25	const KIND: KeyKind = KeyKind::Column;
26
27	fn encode(&self) -> EncodedKey {
28		let mut serializer = KeySerializer::with_capacity(18);
29		serializer.extend_u8(Self::KIND as u8).extend_object_id(self.object).extend_u64(self.column);
30		serializer.to_encoded_key()
31	}
32
33	fn decode(key: &EncodedKey) -> Option<Self> {
34		let mut de = KeyDeserializer::from_bytes(key.as_slice());
35
36		let kind: KeyKind = de.read_u8().ok()?.try_into().ok()?;
37		if kind != Self::KIND {
38			return None;
39		}
40
41		let object = de.read_object_id().ok()?;
42		let column = de.read_u64().ok()?;
43
44		Some(Self {
45			object,
46			column: ColumnId(column),
47		})
48	}
49}
50
51impl ColumnKey {
52	pub fn encoded(object: impl Into<ObjectId>, column: impl Into<ColumnId>) -> EncodedKey {
53		Self {
54			object: object.into(),
55			column: column.into(),
56		}
57		.encode()
58	}
59
60	pub fn full_scan(object: impl Into<ObjectId>) -> EncodedKeyRange {
61		let object = object.into();
62		EncodedKeyRange::start_end(Some(Self::start(object)), Some(Self::end(object)))
63	}
64
65	fn start(object: ObjectId) -> EncodedKey {
66		let mut serializer = KeySerializer::with_capacity(10);
67		serializer.extend_u8(Self::KIND as u8).extend_object_id(object);
68		serializer.to_encoded_key()
69	}
70
71	fn end(object: ObjectId) -> EncodedKey {
72		let mut serializer = KeySerializer::with_capacity(10);
73		serializer.extend_u8(Self::KIND as u8).extend_object_id(object.prev());
74		serializer.to_encoded_key()
75	}
76}
77
78#[cfg(test)]
79pub mod tests {
80	use super::EncodableKey;
81	use crate::{
82		interface::catalog::{id::ColumnId, object::ObjectId},
83		key::ColumnKey,
84	};
85
86	#[test]
87	fn test_encode_decode() {
88		let key = ColumnKey {
89			object: ObjectId::table(0xABCD),
90			column: ColumnId(0x123456789ABCDEF0),
91		};
92		let encoded = key.encode();
93
94		let expected: Vec<u8> = vec![
95			0xF8, 0x01, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0x54, 0x32, 0xED, 0xCB, 0xA9, 0x87, 0x65, 0x43,
96			0x21, 0x0F,
97		];
98
99		assert_eq!(encoded.as_slice(), expected);
100
101		let key = ColumnKey::decode(&encoded).unwrap();
102		assert_eq!(key.object, 0xABCD);
103		assert_eq!(key.column, 0x123456789ABCDEF0);
104	}
105
106	#[test]
107	fn test_order_preserving() {
108		let key1 = ColumnKey {
109			object: ObjectId::table(1),
110			column: ColumnId(100),
111		};
112		let key2 = ColumnKey {
113			object: ObjectId::table(1),
114			column: ColumnId(200),
115		};
116		let key3 = ColumnKey {
117			object: ObjectId::table(2),
118			column: ColumnId(0),
119		};
120
121		let encoded1 = key1.encode();
122		let encoded2 = key2.encode();
123		let encoded3 = key3.encode();
124
125		assert!(encoded3 < encoded2, "ordering not preserved");
126		assert!(encoded2 < encoded1, "ordering not preserved");
127	}
128}
reifydb_core/key/column.rs

reifydb_core/key/
column.rs
