Skip to main content

reifydb_core/key/
migration_event.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::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}