&mut self for max flexibility

ivmarkov · May 27, 2024 · 28a733f · 28a733f
1 parent 58d3d6d
commit 28a733f
Showing 1 changed file with 21 additions and 4 deletions.
diff --git a/src/persist.rs b/src/persist.rs
@@ -219,9 +219,26 @@ where
 
 /// A trait representing a key-value BLOB storage.
 pub trait KvBlobStore {
-    async fn load<'a>(&self, key: &str, buf: &'a mut [u8]) -> Result<Option<&'a [u8]>, Error>;
+    async fn load<'a>(&mut self, key: &str, buf: &'a mut [u8]) -> Result<Option<&'a [u8]>, Error>;
     async fn store(&mut self, key: &str, value: &[u8]) -> Result<(), Error>;
-    async fn remove(&self, key: &str) -> Result<(), Error>;
+    async fn remove(&mut self, key: &str) -> Result<(), Error>;
+}
+
+impl<T> KvBlobStore for &mut T
+where
+    T: KvBlobStore,
+{
+    async fn load<'a>(&mut self, key: &str, buf: &'a mut [u8]) -> Result<Option<&'a [u8]>, Error> {
+        T::load(self, key, buf).await
+    }
+
+    async fn store(&mut self, key: &str, value: &[u8]) -> Result<(), Error> {
+        T::store(self, key, value).await
+    }
+
+    async fn remove(&mut self, key: &str) -> Result<(), Error> {
+        T::remove(self, key).await
+    }
 }
 
 /// An implementation of the `KvBlobStore` trait that stores the BLOBs in a directory.
@@ -309,15 +326,15 @@ impl DirKvStore {
 
 #[cfg(feature = "std")]
 impl KvBlobStore for DirKvStore {
-    async fn load<'a>(&self, key: &str, buf: &'a mut [u8]) -> Result<Option<&'a [u8]>, Error> {
+    async fn load<'a>(&mut self, key: &str, buf: &'a mut [u8]) -> Result<Option<&'a [u8]>, Error> {
         DirKvStore::load(self, key, buf)
     }
 
     async fn store(&mut self, key: &str, value: &[u8]) -> Result<(), Error> {
         DirKvStore::store(self, key, value)
     }
 
-    async fn remove(&self, key: &str) -> Result<(), Error> {
+    async fn remove(&mut self, key: &str) -> Result<(), Error> {
         DirKvStore::remove(self, key)
     }
 }