feat: implement DatabaseCommit for DatabaseComponents (#1163)

* feat: implement DatabaseCommit for DatabaseComponents * misc: apply review suggestions
bluealloy · Mar 7, 2024 · f36f90e · f36f90e
1 parent ac99931
commit f36f90e
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/crates/primitives/src/db.rs b/crates/primitives/src/db.rs
@@ -91,6 +91,13 @@ impl<T: DatabaseRef> Database for WrapDatabaseRef<T> {
     }
 }
 
+impl<T: DatabaseRef + DatabaseCommit> DatabaseCommit for WrapDatabaseRef<T> {
+    #[inline]
+    fn commit(&mut self, changes: HashMap<Address, Account>) {
+        self.0.commit(changes)
+    }
+}
+
 /// Wraps a `dyn DatabaseRef` to provide a [`Database`] implementation.
 #[doc(hidden)]
 #[deprecated = "use `WrapDatabaseRef` instead"]

diff --git a/crates/primitives/src/db/components.rs b/crates/primitives/src/db/components.rs
@@ -7,9 +7,11 @@ pub use state::{State, StateRef};
 
 use crate::{
     db::{Database, DatabaseRef},
-    AccountInfo, Address, Bytecode, B256, U256,
+    Account, AccountInfo, Address, Bytecode, HashMap, B256, U256,
 };
 
+use super::DatabaseCommit;
+
 #[derive(Debug)]
 pub struct DatabaseComponents<S, BH> {
     pub state: S,
@@ -73,3 +75,9 @@ impl<S: StateRef, BH: BlockHashRef> DatabaseRef for DatabaseComponents<S, BH> {
             .map_err(Self::Error::BlockHash)
     }
 }
+
+impl<S: DatabaseCommit, BH: BlockHashRef> DatabaseCommit for DatabaseComponents<S, BH> {
+    fn commit(&mut self, changes: HashMap<Address, Account>) {
+        self.state.commit(changes);
+    }
+}