committer: add cli for committer benchmark

Author: yoavGrs

crates/starknet_committer/src/block_committer/input.rs

@@ -101,7 +101,7 @@ impl Default for ConfigImpl {
     }
 }
 
-#[derive(Debug, Eq, PartialEq)]
+#[derive(Debug, Default, Eq, PartialEq)]
 pub struct Input<C: Config> {
     /// All relevant information for the state diff commitment.
     pub state_diff: StateDiff,

crates/starknet_committer_and_os_cli/src/committer_cli/commands.rs

@@ -1,6 +1,13 @@
+use std::collections::HashMap;
+use std::time::Instant;
+
+use rand::rngs::StdRng;
+use rand::SeedableRng;
 use starknet_committer::block_committer::commit::commit_block;
-use starknet_committer::block_committer::input::Config;
-use starknet_patricia_storage::map_storage::MapStorage;
+use starknet_committer::block_committer::input::{Config, ConfigImpl};
+use starknet_committer::block_committer::state_diff_generator::generate_random_state_diff;
+use starknet_patricia::hash::hash_trait::HashOutput;
+use starknet_patricia_storage::map_storage::{BorrowedMapStorage, MapStorage};
 use tracing::info;
 use tracing::level_filters::LevelFilter;
 use tracing_subscriber::reload::Handle;
@@ -43,3 +50,40 @@ pub async fn commit(input: InputImpl, output_path: String, storage: MapStorage)
         output.contract_storage_root_hash, output.compiled_class_root_hash,
     );
 }
+
+pub async fn run_storage_benchmark() {
+    let n_iterations = 10;
+    let seed = 42_u64; // Constant seed for reproducibility
+
+    let mut rng = StdRng::seed_from_u64(seed);
+    let mut total_time = 0;
+
+    let mut storage = MapStorage::new();
+    let mut contracts_trie_root_hash = HashOutput::default();
+    let mut classes_trie_root_hash = HashOutput::default();
+
+    for i in 0..n_iterations {
+        info!("Committer storage benchmark iteration {}/{}", i + 1, n_iterations);
+        let input = InputImpl {
+            state_diff: generate_random_state_diff(&mut rng),
+            contracts_trie_root_hash,
+            classes_trie_root_hash,
+            config: ConfigImpl::default(),
+        };
+
+        // commit block and write to storage
+        let start = Instant::now();
+        let serialized_filled_forest = SerializedForest(
+            commit_block(input, &mut storage).await.expect("Failed to commit the given block."),
+        );
+        serialized_filled_forest.0.write_to_storage(&mut storage);
+        let duration = start.elapsed();
+        total_time += duration.as_millis();
+
+        contracts_trie_root_hash = serialized_filled_forest.0.get_contract_root_hash();
+        classes_trie_root_hash = serialized_filled_forest.0.get_compiled_class_root_hash();
+    }
+
+    let avg_millis = total_time as f64 / n_iterations as f64;
+    info!("Average time: {:.2} milliseconds", avg_millis);
+}

crates/starknet_committer_and_os_cli/src/committer_cli/run_committer_cli.rs

@@ -4,7 +4,7 @@ use tracing::level_filters::LevelFilter;
 use tracing_subscriber::reload::Handle;
 use tracing_subscriber::Registry;
 
-use crate::committer_cli::commands::parse_and_commit;
+use crate::committer_cli::commands::{parse_and_commit, run_storage_benchmark};
 use crate::committer_cli::tests::python_tests::CommitterPythonTestRunner;
 use crate::shared_utils::types::{run_python_test, IoArgs, PythonTestArg};
 
@@ -22,6 +22,8 @@ enum Command {
         io_args: IoArgs,
     },
     PythonTest(PythonTestArg),
+    /// Run the committer on random data with owned storage.
+    StorageBenchmark,
 }
 
 pub async fn run_committer_cli(
@@ -37,5 +39,8 @@ pub async fn run_committer_cli(
         Command::PythonTest(python_test_arg) => {
             run_python_test::<CommitterPythonTestRunner>(python_test_arg).await;
         }
+        Command::StorageBenchmark => {
+            run_storage_benchmark().await;
+        }
     }
 }