separate methods for storage changes and storage overlay

matter-labs · May 9, 2024 · e9fcb3e · e9fcb3e
1 parent 9dee00b
commit e9fcb3e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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,