use super::{EncodableKey, KeyKind};
use crate::{
encoded::key::{EncodedKey, EncodedKeyRange},
interface::catalog::id::{NamespaceId, SeriesId},
util::encoding::keycode::{deserializer::KeyDeserializer, serializer::KeySerializer},
};
const VERSION: u8 = 1;
#[derive(Debug, Clone, PartialEq)]
pub struct NamespaceSeriesKey {
pub namespace: NamespaceId,
pub series: SeriesId,
}
impl NamespaceSeriesKey {
pub fn new(namespace: NamespaceId, series: SeriesId) -> Self {
Self {
namespace,
series,
}
}
pub fn encoded(namespace: impl Into<NamespaceId>, series: impl Into<SeriesId>) -> EncodedKey {
Self::new(namespace.into(), series.into()).encode()
}
pub fn full_scan(namespace: NamespaceId) -> EncodedKeyRange {
EncodedKeyRange::start_end(Some(Self::link_start(namespace)), Some(Self::link_end(namespace)))
}
fn link_start(namespace: NamespaceId) -> EncodedKey {
let mut serializer = KeySerializer::with_capacity(10);
serializer.extend_u8(VERSION).extend_u8(Self::KIND as u8).extend_u64(namespace);
serializer.to_encoded_key()
}
fn link_end(namespace: NamespaceId) -> EncodedKey {
let mut serializer = KeySerializer::with_capacity(10);
serializer.extend_u8(VERSION).extend_u8(Self::KIND as u8).extend_u64(*namespace - 1);
serializer.to_encoded_key()
}
}
impl EncodableKey for NamespaceSeriesKey {
const KIND: KeyKind = KeyKind::NamespaceSeries;
fn encode(&self) -> EncodedKey {
let mut serializer = KeySerializer::with_capacity(18);
serializer
.extend_u8(VERSION)
.extend_u8(Self::KIND as u8)
.extend_u64(self.namespace)
.extend_u64(self.series);
serializer.to_encoded_key()
}
fn decode(key: &EncodedKey) -> Option<Self> {
let mut de = KeyDeserializer::from_bytes(key.as_slice());
let version = de.read_u8().ok()?;
if version != VERSION {
return None;
}
let kind: KeyKind = de.read_u8().ok()?.try_into().ok()?;
if kind != Self::KIND {
return None;
}
let namespace = de.read_u64().ok()?;
let series = de.read_u64().ok()?;
Some(Self {
namespace: NamespaceId(namespace),
series: SeriesId(series),
})
}
}