From e9fcb3ecc8295096650f472b464ed089e32e5b6c Mon Sep 17 00:00:00 2001
From: Joonatan Saarhelo <joon.saar@gmail.com>
Date: Thu, 9 May 2024 02:10:28 +0200
Subject: [PATCH] separate methods for storage changes and storage overlay

---
 src/modified_world.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/modified_world.rs b/src/modified_world.rs
index f413a2b8..978912d5 100644
--- a/src/modified_world.rs
+++ b/src/modified_world.rs
@@ -123,10 +123,14 @@ impl ModifiedWorld {
         (refund, (update_cost as i32) - (prepaid as i32))
     }
 
-    pub fn get_storage_changes(&self) -> &BTreeMap<(H160, U256), U256> {
+    pub fn get_storage_state(&self) -> &BTreeMap<(H160, U256), U256> {
         self.storage_changes.as_ref()
     }
 
+    pub fn get_storage_changes(&self) -> BTreeMap<(H160, U256), (Option<U256>, U256)> {
+        self.storage_changes.changes_after(0)
+    }
+
     pub fn get_storage_changes_after(
         &self,
         snapshot: &Snapshot,