reifydb_core/key/
migration_event.rs1use reifydb_codec::key::{
5 deserializer::KeyDeserializer,
6 encoded::{EncodedKey, EncodedKeyRange},
7 serializer::KeySerializer,
8};
9
10use crate::{
11 interface::catalog::id::MigrationEventId,
12 key::{EncodableKey, KeyKind},
13};
14
15#[derive(Debug, Clone, PartialEq)]
16pub struct MigrationEventKey {
17 pub event: MigrationEventId,
18}
19
20impl MigrationEventKey {
21 pub fn new(event: MigrationEventId) -> Self {
22 Self {
23 event,
24 }
25 }
26
27 pub fn encoded(event: impl Into<MigrationEventId>) -> EncodedKey {
28 Self::new(event.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 MigrationEventKey {
49 const KIND: KeyKind = KeyKind::MigrationEvent;
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.event);
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 event = de.read_u64().ok()?;
66
67 Some(Self {
68 event: MigrationEventId(event),
69 })
70 }
71}