refactor(blockifier): move logic to function

starkware-libs · Nov 17, 2024 · 7a27e86 · 7a27e86
1 parent e97d2d4
commit 7a27e86
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 9 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/blockifier/src/execution/call_info.rs b/crates/blockifier/src/execution/call_info.rs
@@ -136,6 +136,16 @@ impl CallInfo {
         CallInfoIter { call_infos }
     }
 
+    fn specific_event_summary(&self) -> EventSummary {
+        let mut event_summary = EventSummary::default();
+        event_summary.n_events = self.execution.events.len();
+        for OrderedEvent { event, .. } in self.execution.events.iter() {
+            event_summary.total_event_data_size += u64_from_usize(event.data.0.len());
+            event_summary.total_event_keys += u64_from_usize(event.keys.len());
+        }
+        event_summary
+    }
+
     pub fn summarize(&self) -> ExecutionSummary {
         let mut executed_class_hashes: HashSet<ClassHash> = HashSet::new();
         let mut visited_storage_entries: HashSet<StorageEntry> = HashSet::new();
@@ -165,14 +175,7 @@ impl CallInfo {
             );
 
             // Events.
-            event_summary.n_events += call_info.execution.events.len();
-            for OrderedEvent { event, .. } in call_info.execution.events.iter() {
-                // TODO(barak: 18/03/2024): Once we start charging per byte
-                // change to num_bytes_keys
-                // and num_bytes_data.
-                event_summary.total_event_data_size += u64_from_usize(event.data.0.len());
-                event_summary.total_event_keys += u64_from_usize(event.keys.len());
-            }
+            event_summary += call_info.specific_event_summary();
         }
 
         ExecutionSummary {