db

1 files changed, 11 insertions, 2 deletions

diff --git a/database/src/backends/memory.rs b/database/src/backends/memory.rs
index 97c8c2c..2f19ce6 100644
--- a/database/src/backends/memory.rs
+++ b/database/src/backends/memory.rs
@@ -4,13 +4,18 @@
     Copyright (C) 2025 metamuffin <metamuffin.org>
 */
 
-use crate::backends::DatabaseStorage;
+use crate::backends::KV;
 use anyhow::Result;
 use std::{collections::BTreeMap, sync::RwLock};
 
 pub struct Memory(RwLock<BTreeMap<Vec<u8>, Vec<u8>>>);
 
-impl DatabaseStorage for Memory {
+impl Memory {
+    pub fn new() -> Self {
+        Self(RwLock::new(BTreeMap::new()))
+    }
+}
+impl KV for Memory {
     fn set(&self, key: &[u8], value: &[u8]) -> Result<()> {
         self.0.write().unwrap().insert(key.to_vec(), value.to_vec());
         Ok(())
@@ -18,6 +23,10 @@ impl DatabaseStorage for Memory {
     fn get<'a>(&'a self, key: &[u8]) -> Result<Option<Vec<u8>>> {
         Ok(self.0.read().unwrap().get(key).cloned())
     }
+    fn del(&self, key: &[u8]) -> Result<()> {
+        self.0.write().unwrap().remove(key);
+        Ok(())
+    }
     fn next(&self, key: &[u8]) -> Result<Option<Vec<u8>>> {
         let r = self.0.read().unwrap();
         Ok(r.range(key.to_vec()..).next().map(|(k, _)| k.to_owned()))