add converter

Author: ratankaliani

Diff for: Cargo.lock

@@ -22,11 +22,17 @@ alloy-eips.workspace = true
 alloy-consensus.workspace = true
 alloy-rlp.workspace = true
 alloy-provider.workspace = true
-alloy-transport-http.workspace = true 
-alloy-rpc-client.workspace = true 
+alloy-transport-http.workspace = true
+alloy-rpc-client.workspace = true
 alloy-rpc-types = { workspace = true, features = ["eth"] }
 alloy-primitives = { workspace = true, features = ["serde"] }
-revm = { workspace = true, features = ["std", "c-kzg", "secp256k1", "portable", "blst"] }
+revm = { workspace = true, features = [
+    "std",
+    "c-kzg",
+    "secp256k1",
+    "portable",
+    "blst",
+] }
 
 # General
 anyhow.workspace = true
@@ -42,3 +48,4 @@ tracing-subscriber.workspace = true
 command-fds.workspace = true
 os_pipe.workspace = true
 rocksdb.workspace = true
+rkyv = "0.7.42"

Diff for: bin/host/Cargo.toml

@@ -1,7 +1,7 @@
 //! Contains a concrete implementation of the [KeyValueStore] trait that stores data on disk
 //! using [rocksdb].
 
-use super::KeyValueStore;
+use super::{KeyValueStore, MemoryKeyValueStore};
 use anyhow::{anyhow, Result};
 use rocksdb::{Options, DB};
 use std::path::PathBuf;
@@ -39,8 +39,24 @@ impl KeyValueStore for DiskKeyValueStore {
     fn set(&mut self, key: alloy_primitives::B256, value: Vec<u8>) -> Result<()> {
         self.db.put(*key, value).map_err(|e| anyhow!("Failed to set key-value pair: {}", e))
     }
+
+    /// Converts the [DiskKeyValueStore] to a [MemoryKeyValueStore].
+    fn to_memory_store(&self) -> MemoryKeyValueStore {
+        let mut memory_store = MemoryKeyValueStore::new();
+        let iter = self.db.iterator(rocksdb::IteratorMode::Start);
+        for item in iter {
+            if let Ok((key, value)) = item {
+                if let Ok(b256_key) = alloy_primitives::B256::try_from(key.as_ref()) {
+                    let _ = memory_store.set(b256_key, value.to_vec());
+                }
+            }
+        }
+        memory_store
+    }
 }
 
+impl DiskKeyValueStore {}
+
 impl Drop for DiskKeyValueStore {
     fn drop(&mut self) {
         let _ = DB::destroy(&Self::get_db_options(), self.data_directory.as_path());

Diff for: bin/host/src/kv/disk.rs

@@ -1,6 +1,6 @@
 //! Contains a concrete implementation of the [KeyValueStore] trait that stores data on disk.
 
-use super::KeyValueStore;
+use super::{KeyValueStore, MemoryKeyValueStore};
 use crate::cli::HostCli;
 use alloy_primitives::B256;
 use anyhow::Result;
@@ -44,4 +44,8 @@ impl KeyValueStore for LocalKeyValueStore {
     fn set(&mut self, _: B256, _: Vec<u8>) -> Result<()> {
         unreachable!("LocalKeyValueStore is read-only")
     }
+
+    fn to_memory_store(&self) -> MemoryKeyValueStore {
+        unimplemented!("LocalKeyValueStore is read-only")
+    }
 }

Diff for: bin/host/src/kv/local.rs

@@ -7,9 +7,11 @@ use std::collections::HashMap;
 
 /// A simple, synchronous key-value store that stores data in memory. This is useful for testing and
 /// development purposes.
-#[derive(Default, Clone, Debug, Eq, PartialEq)]
+#[derive(
+    Default, Clone, Debug, Eq, PartialEq, rkyv::Archive, rkyv::Serialize, rkyv::Deserialize,
+)]
 pub struct MemoryKeyValueStore {
-    store: HashMap<B256, Vec<u8>>,
+    pub store: HashMap<[u8; 32], Vec<u8>>,
 }
 
 impl MemoryKeyValueStore {
@@ -21,11 +23,15 @@ impl MemoryKeyValueStore {
 
 impl KeyValueStore for MemoryKeyValueStore {
     fn get(&self, key: B256) -> Option<Vec<u8>> {
-        self.store.get(&key).cloned()
+        self.store.get(&key.0).cloned()
     }
 
     fn set(&mut self, key: B256, value: Vec<u8>) -> Result<()> {
-        self.store.insert(key, value);
+        self.store.insert(key.0, value);
         Ok(())
     }
+
+    fn to_memory_store(&self) -> MemoryKeyValueStore {
+        self.clone()
+    }
 }

Diff for: bin/host/src/kv/mem.rs

@@ -27,4 +27,7 @@ pub trait KeyValueStore {
 
     /// Set the value associated with the given key.
     fn set(&mut self, key: B256, value: Vec<u8>) -> Result<()>;
+
+    /// Converts the [KeyValueStore] to a [MemoryKeyValueStore].
+    fn to_memory_store(&self) -> MemoryKeyValueStore;
 }

Diff for: bin/host/src/kv/mod.rs

@@ -1,7 +1,7 @@
 //! Contains a concrete implementation of the [KeyValueStore] trait that splits between two separate
 //! [KeyValueStore]s depending on [PreimageKeyType].
 
-use super::KeyValueStore;
+use super::{KeyValueStore, MemoryKeyValueStore};
 use alloy_primitives::B256;
 use anyhow::Result;
 use kona_preimage::PreimageKeyType;
@@ -44,4 +44,8 @@ where
     fn set(&mut self, key: B256, value: Vec<u8>) -> Result<()> {
         self.remote_store.set(key, value)
     }
+
+    fn to_memory_store(&self) -> MemoryKeyValueStore {
+        self.remote_store.to_memory_store()
+    }
 }