Add metadata impl for chaindb mock

tari-project · Jul 6, 2022 · 1ee920b · 1ee920b
1 parent e6d4c87
commit 1ee920b
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 7 deletions.
diff --git a/dan_layer/core/src/storage/chain/metadata_key.rs b/dan_layer/core/src/storage/chain/metadata_key.rs
@@ -24,7 +24,7 @@ use std::fmt::{Display, Formatter};
 
 use crate::storage::metadata_backend_adapter::AsKeyBytes;
 
-#[derive(Debug, Clone, Copy)]
+#[derive(Debug, Clone, Copy, PartialEq, Eq)]
 pub enum ChainDbMetadataKey {
     CheckpointNumber,
 }

diff --git a/dan_layer/core/src/storage/metadata_backend_adapter.rs b/dan_layer/core/src/storage/metadata_backend_adapter.rs
@@ -23,7 +23,7 @@
 // TODO: probably want to use something like bors or consensus encoding
 use tari_utilities::message_format::MessageFormat;
 
-use crate::storage::StorageError;
+use crate::storage::AtomicDb;
 
 pub trait MetadataBackendAdapter<K: AsKeyBytes>: AtomicDb + Send + Sync + Clone {
     fn get_metadata<T: MessageFormat>(

diff --git a/dan_layer/core/src/storage/mocks/chain_db.rs b/dan_layer/core/src/storage/mocks/chain_db.rs
@@ -208,26 +208,43 @@ impl MetadataBackendAdapter<ChainDbMetadataKey> for MockChainDbBackupAdapter {
     fn get_metadata<T: MessageFormat>(
         &self,
         key: &ChainDbMetadataKey,
-        transaction: &Self::DbTransaction,
+        _transaction: &Self::DbTransaction,
     ) -> Result<Option<T>, Self::Error> {
         let lock = self.db.read()?;
-        Ok(lock.metadata.rows().find(|(k, _)| k == key))
+        let v = lock
+            .metadata
+            .rows()
+            .find(|(k, _)| k == key)
+            .map(|(_, v)| v)
+            .map(|v| T::from_binary(v).unwrap());
+        Ok(v)
     }
 
     fn set_metadata<T: MessageFormat>(
         &self,
         key: ChainDbMetadataKey,
         value: T,
-        transaction: &Self::DbTransaction,
+        _transaction: &Self::DbTransaction,
     ) -> Result<(), Self::Error> {
         let mut lock = self.db.write()?;
-        Ok(lock.metadata.rows().find(|(k, _)| k == key))
+        let value = value.to_binary().unwrap();
+        let id = lock.metadata.records().find(|(_, (k, _))| *k == key).map(|(id, _)| id);
+        match id {
+            Some(id) => {
+                lock.metadata.update(id, (key, value));
+            },
+            None => {
+                lock.metadata.insert((key, value));
+            },
+        }
+
+        Ok(())
     }
 
     fn metadata_key_exists(
         &self,
         _key: &ChainDbMetadataKey,
-        transaction: &Self::DbTransaction,
+        _transaction: &Self::DbTransaction,
     ) -> Result<bool, Self::Error> {
         todo!()
     }