feat: postgres temporary storage

src/config.rs

@@ -33,6 +33,8 @@ use crate::eth::storage::PostgresExternalRpcStorage;
 use crate::eth::storage::PostgresExternalRpcStorageConfig;
 use crate::eth::storage::PostgresPermanentStorage;
 use crate::eth::storage::PostgresPermanentStorageConfig;
+use crate::eth::storage::PostgresTemporaryStorage;
+use crate::eth::storage::PostgresTemporaryStorageConfig;
 #[cfg(feature = "rocks")]
 use crate::eth::storage::RocksPermanentStorage;
 #[cfg(feature = "rocks")]
@@ -562,22 +564,40 @@ pub struct TemporaryStorageConfig {
     /// Temporary storage implementation.
     #[arg(long = "temp-storage", env = "TEMP_STORAGE")]
     pub temp_storage_kind: TemporaryStorageKind,
+
+    #[arg(long = "temp-storage-connections", env = "TEMP_STORAGE_CONNECTIONS")]
+    pub temp_storage_connections: u32,
+
+    /// Permamenent storage timeout when opening a connection (in millis).
+    #[arg(long = "temp-storage-timeout", env = "TEMP_STORAGE_TIMEOUT")]
+    pub temp_storage_timeout_millis: u64,
 }
 
 #[derive(Clone, Debug)]
 pub enum TemporaryStorageKind {
     InMemory,
     #[cfg(feature = "rocks")]
     Rocks,
+    Postgres {
+        url: String,
+    },
 }
 
 impl TemporaryStorageConfig {
     /// Initializes temporary storage implementation.
     pub async fn init(&self) -> anyhow::Result<Arc<dyn TemporaryStorage>> {
-        match self.temp_storage_kind {
+        match self.temp_storage_kind.clone() {
             TemporaryStorageKind::InMemory => Ok(Arc::new(InMemoryTemporaryStorage::default())),
             #[cfg(feature = "rocks")]
             TemporaryStorageKind::Rocks => Ok(Arc::new(RocksTemporary::new().await?)),
+            TemporaryStorageKind::Postgres { url } => Ok(Arc::new(
+                PostgresTemporaryStorage::new(PostgresTemporaryStorageConfig {
+                    url,
+                    connections: self.temp_storage_connections,
+                    acquire_timeout: Duration::from_millis(self.temp_storage_timeout_millis),
+                })
+                .await?,
+            )),
         }
     }
 }
@@ -590,6 +610,7 @@ impl FromStr for TemporaryStorageKind {
             "inmemory" => Ok(Self::InMemory),
             #[cfg(feature = "rocks")]
             "rocks" => Ok(Self::Rocks),
+            s if s.starts_with("postgres://") => Ok(Self::Postgres { url: s.to_string() }),
             s => Err(anyhow!("unknown temporary storage: {}", s)),
         }
     }

src/eth/primitives/execution_account_changes.rs

@@ -1,5 +1,7 @@
 use std::collections::HashMap;
 
+use anyhow::Context;
+
 use super::CodeHash;
 use crate::eth::primitives::Account;
 use crate::eth::primitives::Address;
@@ -114,3 +116,18 @@ impl ExecutionAccountChanges {
         not(self.new_account)
     }
 }
+
+impl TryFrom<ExecutionAccountChanges> for Account {
+    type Error = anyhow::Error;
+    fn try_from(value: ExecutionAccountChanges) -> Result<Self, Self::Error> {
+        Ok(Account {
+            address: value.address,
+            nonce: value.nonce.take().context("No nonce in changes")?,
+            balance: value.balance.take().context("No balance in changes")?,
+            bytecode: value.bytecode.take().context("No bytecode in changes")?,
+            code_hash: value.code_hash,
+            static_slot_indexes: value.static_slot_indexes.take().context("No static_slot_indexes in changes")?,
+            mapping_slot_indexes: value.mapping_slot_indexes.take().context("No mapping_slot_indexes in changes")?,
+        })
+    }
+}

src/eth/storage/mod.rs

@@ -5,7 +5,7 @@ mod external_rpc_storage;
 mod inmemory;
 mod permanent_storage;
 mod postgres_external_rpc;
-mod postgres_permanent;
+mod postgres;
 #[cfg(feature = "rocks")]
 pub mod rocks;
 
@@ -21,8 +21,10 @@ pub use inmemory::InMemoryTemporaryStorage;
 pub use permanent_storage::PermanentStorage;
 pub use postgres_external_rpc::PostgresExternalRpcStorage;
 pub use postgres_external_rpc::PostgresExternalRpcStorageConfig;
-pub use postgres_permanent::PostgresPermanentStorage;
-pub use postgres_permanent::PostgresPermanentStorageConfig;
+pub use postgres::PostgresPermanentStorage;
+pub use postgres::PostgresPermanentStorageConfig;
+pub use postgres::PostgresTemporaryStorage;
+pub use postgres::PostgresTemporaryStorageConfig;
 #[cfg(feature = "rocks")]
 pub use rocks::rocks_permanent::RocksPermanentStorage;
 #[cfg(feature = "rocks")]

src/eth/storage/postgres_permanent/mod.rs → src/eth/storage/postgres/mod.rs

@@ -1,6 +1,9 @@
 #[allow(clippy::module_inception)]
 mod postgres_permanent;
 mod types;
+mod postgres_temporary;
 
 pub use postgres_permanent::PostgresPermanentStorage;
 pub use postgres_permanent::PostgresPermanentStorageConfig;
+pub use postgres_temporary::PostgresTemporaryStorage;
+pub use postgres_temporary::PostgresTemporaryStorageConfig;