feat: rocks temporary storage (mimicking sled's implementation) (#562)

* feat: rocks temporary storage (mimicking sled's implementation) * fmt
cloudwalk · Apr 10, 2024 · f75ece6 · f75ece6
1 parent 9c12bd1
commit f75ece6
Show file tree

Hide file tree

Showing 8 changed files with 789 additions and 606 deletions.
diff --git a/src/config.rs b/src/config.rs
@@ -33,6 +33,7 @@ use crate::eth::storage::PostgresExternalRpcStorageConfig;
 use crate::eth::storage::PostgresPermanentStorage;
 use crate::eth::storage::PostgresPermanentStorageConfig;
 use crate::eth::storage::RocksPermanentStorage;
+use crate::eth::storage::RocksTemporary;
 use crate::eth::storage::SledTemporary;
 use crate::eth::storage::StratusStorage;
 use crate::eth::storage::TemporaryStorage;
@@ -546,6 +547,7 @@ pub struct TemporaryStorageConfig {
 pub enum TemporaryStorageKind {
     InMemory,
     Sled,
+    Rocks,
 }
 
 impl TemporaryStorageConfig {
@@ -554,6 +556,7 @@ impl TemporaryStorageConfig {
         match self.temp_storage_kind {
             TemporaryStorageKind::InMemory => Ok(Arc::new(InMemoryTemporaryStorage::default())),
             TemporaryStorageKind::Sled => Ok(Arc::new(SledTemporary::new()?)),
+            TemporaryStorageKind::Rocks => Ok(Arc::new(RocksTemporary::new()?)),
         }
     }
 }
@@ -565,6 +568,7 @@ impl FromStr for TemporaryStorageKind {
         match s {
             "inmemory" => Ok(Self::InMemory),
             "sled" => Ok(Self::Sled),
+            "rocks" => Ok(Self::Rocks),
             s => Err(anyhow!("unknown temporary storage: {}", s)),
         }
     }

diff --git a/src/eth/storage/inmemory/inmemory_temporary.rs b/src/eth/storage/inmemory/inmemory_temporary.rs
@@ -152,7 +152,7 @@ impl TemporaryStorage for InMemoryTemporaryStorage {
     }
 }
 
-#[derive(Debug)]
+#[derive(Debug, Clone)]
 pub struct InMemoryTemporaryAccount {
     pub info: Account,
     pub slots: HashMap<SlotIndex, Slot>,

diff --git a/src/eth/storage/mod.rs b/src/eth/storage/mod.rs
@@ -23,7 +23,8 @@ pub use postgres_external_rpc::PostgresExternalRpcStorage;
 pub use postgres_external_rpc::PostgresExternalRpcStorageConfig;
 pub use postgres_permanent::PostgresPermanentStorage;
 pub use postgres_permanent::PostgresPermanentStorageConfig;
-pub use rocks::RocksPermanentStorage;
+pub use rocks::rocks_permanent::RocksPermanentStorage;
+pub use rocks::rocks_temporary::RocksTemporary;
 pub use sled::SledTemporary;
 pub use storage_error::StorageError;
 pub use stratus_storage::StratusStorage;