From 5c2b246d5d5687633de64259776dcab780c5e8f5 Mon Sep 17 00:00:00 2001 From: Matthias Seitz Date: Sun, 1 Feb 2026 14:00:00 +0100 Subject: [PATCH] feat(database): add clear() to CacheState and TransitionState Amp-Thread-ID: https://ampcode.com/threads/T-019c1944-8542-72b6-b9f7-b7337cce5213 Co-authored-by: Amp --- crates/database/src/states/cache.rs | 6 ++++++ crates/database/src/states/transition_state.rs | 5 +++++ 2 files changed, 11 insertions(+) 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