use crate::errors::DbResult;
use crate::logger::Logger;
use crate::table::Table;
use crate::TableId;
use serde::de::DeserializeOwned;
use serde::Serialize;
#[derive(Debug)]
#[cfg_attr(feature = "clone", derive(Clone))]
pub struct Single<T>
where
T: Serialize + DeserializeOwned,
{
table_id: TableId,
inner: Option<T>,
pub logger: Logger,
}
impl<T> Table for Single<T>
where
T: Serialize + DeserializeOwned,
{
fn init(table_id: TableId, logger: Logger) -> Self {
Self { table_id, inner: None, logger }
}
fn handle_event(&mut self, bytes: &[u8]) -> DbResult<()> {
self.inner = bincode::deserialize(bytes)?;
Ok(())
}
fn compact_repr(&self) -> DbResult<Vec<u8>> {
if let Some(v) = &self.inner {
let data = bincode::serialize(&Some(v))?;
let data = Logger::log_entry(self.table_id, data);
Ok(data)
} else {
Ok(vec![])
}
}
}
impl<T> Single<T>
where
T: Serialize + DeserializeOwned,
{
pub fn insert(&mut self, value: T) -> DbResult<Option<T>> {
let log_entry = Some(&value);
let data = bincode::serialize(&log_entry)?;
let ret = self.inner.replace(value);
self.logger.write(self.table_id, data)?;
Ok(ret)
}
pub fn get(&self) -> Option<&T> {
self.inner.as_ref()
}
pub fn clear(&mut self) -> DbResult<Option<T>> {
let log_entry = Option::<T>::None;
let data = bincode::serialize(&log_entry)?;
let ret = self.inner.take();
self.logger.write(self.table_id, data)?;
Ok(ret)
}
}