diff --git a/crates/database/src/states/cache.rs b/crates/database/src/states/cache.rs index 8b6870ef9b..96aa8d7f17 100644 --- a/crates/database/src/states/cache.rs +++ b/crates/database/src/states/cache.rs @@ -45,6 +45,12 @@ impl CacheState { self.has_state_clear = has_state_clear; } + /// Clear the cache state. + pub fn clear(&mut self) { + self.accounts.clear(); + self.contracts.clear(); + } + /// Helper function that returns all accounts. /// /// Used inside tests to generate merkle tree. diff --git a/crates/database/src/states/transition_state.rs b/crates/database/src/states/transition_state.rs index 06654d00af..5b0760a316 100644 --- a/crates/database/src/states/transition_state.rs +++ b/crates/database/src/states/transition_state.rs @@ -24,6 +24,11 @@ impl TransitionState { core::mem::take(self) } + /// Clear the transition state. + pub fn clear(&mut self) { + self.transitions.clear(); + } + /// Add transitions to the transition state. /// /// This will insert new [`TransitionAccount`]s, or update existing ones via