brk/_src/structs/serialized_btreemap.rs

use std::{collections::BTreeMap, fmt::Debug, path::Path};

use allocative::Allocative;
use bincode::{Decode, Encode};
use serde::{de::DeserializeOwned, Deserialize, Serialize};

use crate::io::Serialization;

use super::{Date, DateMap, MapChunkId, MapKey, MapSerialized, MapValue, Timestamp};

pub type SerializedDateMap<T> = SerializedBTreeMap<Date, T>;
pub type SerializedTimeMap<T> = SerializedBTreeMap<Timestamp, T>;

#[derive(Debug, Default, Serialize, Deserialize, Encode, Decode, Allocative)]
pub struct SerializedBTreeMap<Key, Value>
where
    Key: Ord,
{
    version: u32,
    pub map: BTreeMap<Key, Value>,
}

impl<Key, Value, ChunkId> MapSerialized<Key, Value, ChunkId> for SerializedBTreeMap<Key, Value>
where
    Self: Debug + Serialize + DeserializeOwned + Encode + Decode,
    ChunkId: MapChunkId,
    Key: MapKey<ChunkId> + Serialize,
    Value: MapValue,
{
    fn new(version: u32) -> Self {
        Self {
            version,
            map: BTreeMap::default(),
        }
    }

    fn get_last_key(&self, _: &ChunkId) -> Option<Key> {
        self.map.last_key_value().map(|(k, _)| k.to_owned())
    }

    fn version(&self) -> u32 {
        self.version
    }

    fn get(&self, key: &Key) -> Option<&Value> {
        self.map.get(key)
    }

    fn last(&self) -> Option<&Value> {
        self.map.last_key_value().map(|(_, v)| v)
    }

    fn extend(&mut self, map: BTreeMap<Key, Value>) {
        self.map.extend(map)
    }

    fn import_all(path: &Path, serialization: &Serialization) -> Self {
        let mut s = None;

        DateMap::<usize>::_read_dir(path, serialization)
            .iter()
            .for_each(|(_, path)| {
                let map = serialization.import::<Self>(path).unwrap();

                if s.is_none() {
                    s.replace(map);
                } else {
                    #[allow(clippy::unnecessary_unwrap)]
                    s.as_mut().unwrap().map.extend(map.map);
                }
            });

        s.unwrap()
    }

    fn to_csv(self, id: &str) -> String {
        let mut csv = format!("{},{}\n", Key::map_name(), id);
        self.map.iter().for_each(|(k, v)| {
            csv += &format!("{},{}\n", k, v);
        });
        csv
    }

    fn map(&self) -> &impl Serialize {
        &self.map
    }
}