brk: first commit

_src/structs/serialized_btreemap.rs

@@ -0,0 +1,87 @@
+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
+    }
+}