From 8aa9b71ef78e39759c3907ab4df3ebd4f6f43cdd Mon Sep 17 00:00:00 2001 From: Roman Krasiuk Date: Mon, 18 Nov 2024 16:51:46 +0100 Subject: [PATCH] chore(witness): simplify wiped storage retrieval (#12637) --- crates/trie/trie/src/witness.rs | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/crates/trie/trie/src/witness.rs b/crates/trie/trie/src/witness.rs index b2364b385e..6f6a66a16e 100644 --- a/crates/trie/trie/src/witness.rs +++ b/crates/trie/trie/src/witness.rs @@ -213,11 +213,10 @@ where let mut storage_cursor = self.hashed_cursor_factory.hashed_storage_cursor(*hashed_address)?; // position cursor at the start - if let Some((hashed_slot, _)) = storage_cursor.seek(B256::ZERO)? { - storage_keys.insert(hashed_slot); - } - while let Some((hashed_slot, _)) = storage_cursor.next()? { + let mut current_entry = storage_cursor.seek(B256::ZERO)?; + while let Some((hashed_slot, _)) = current_entry { storage_keys.insert(hashed_slot); + current_entry = storage_cursor.next()?; } } proof_targets.insert(*hashed_address, storage_keys);