reifydb_core/key/
identity.rs1use reifydb_codec::key::{
5 deserializer::KeyDeserializer,
6 encoded::{EncodedKey, EncodedKeyRange},
7 serializer::KeySerializer,
8};
9use reifydb_value::value::identity::IdentityId;
10
11use super::{EncodableKey, KeyKind};
12
13#[derive(Debug, Clone, PartialEq)]
14pub struct IdentityKey {
15 pub identity: IdentityId,
16}
17
18impl IdentityKey {
19 pub fn new(identity: IdentityId) -> Self {
20 Self {
21 identity,
22 }
23 }
24
25 pub fn encoded(identity: IdentityId) -> EncodedKey {
26 Self::new(identity).encode()
27 }
28
29 pub fn full_scan() -> EncodedKeyRange {
30 let mut start = KeySerializer::with_capacity(1);
31 start.extend_u8(Self::KIND as u8);
32 let mut end = KeySerializer::with_capacity(1);
33 end.extend_u8(Self::KIND as u8 - 1);
34 EncodedKeyRange::start_end(Some(start.to_encoded_key()), Some(end.to_encoded_key()))
35 }
36}
37
38impl EncodableKey for IdentityKey {
39 const KIND: KeyKind = KeyKind::Identity;
40
41 fn encode(&self) -> EncodedKey {
42 let mut serializer = KeySerializer::with_capacity(17);
43 serializer.extend_u8(Self::KIND as u8).extend_identity_id(&self.identity);
44 serializer.to_encoded_key()
45 }
46
47 fn decode(key: &EncodedKey) -> Option<Self> {
48 let mut de = KeyDeserializer::from_bytes(key.as_slice());
49 let kind: KeyKind = de.read_u8().ok()?.try_into().ok()?;
50 if kind != Self::KIND {
51 return None;
52 }
53 let identity = de.read_identity_id().ok()?;
54 Some(Self {
55 identity,
56 })
57 }
58}