feat: read arguments from input

Cargo.toml

@@ -27,8 +27,8 @@ name = "mini-tree-server"
 path = "bin/mini_tree_server.rs"
 
 [[bin]]
-name = "orchestrator-bench"
-path = "benches/build_mst_with_workers.rs"
+name = "aggregation-bench"
+path = "benches/build_aggregation_mst.rs"
 
 [features]
 docker = []

benches/build_mst_with_workers.rs → benches/build_aggregation_mst.rs

@@ -1,8 +1,13 @@
 #![feature(generic_const_exprs)]
 use const_env::from_env;
-use std::{error::Error, fs};
+use std::{error::Error, fs, env};
 use summa_aggregation::{executor::CloudSpawner, orchestrator::Orchestrator};
 use tokio::time::Instant;
+use summa_backend::{
+    apis::round::{Round, Snapshot},
+    contracts::signer::{AddressInput, SummaSigner},
+    tests::initialize_test_env,
+};
 
 #[from_env]
 const LEVELS: usize = 20;
@@ -14,13 +19,17 @@ async fn main() -> Result<(), Box<dyn Error>> {
     // We assume that custodians, when setting up their distributed environment, will obtain the URLs of worker nodes.
     // In this example, we use two worker URLs corresponding to the workers spawned earlier.
     // It is important to ensure that the number of URLs matches the number of executors.
-    let worker_node_urls = vec!["127.0.0.1:4000".to_string()];
-    let total_workers = worker_node_urls.len();
+    let worker_node_urls: Vec<String> = env::args().skip(1).collect();
+
+    // Ensure that at least one worker node URL is provided
+    if worker_node_urls.is_empty() {
+        return Err("No worker node URLs provided. Usage: cargo run <URL1> <URL2> ...".into());
+    }
 
     const N_CURRENCIES: usize = 1;
     const N_BYTES: usize = 14;
-        
-    // Read the directory and collect CSV file paths
+
+    // // Read the directory and collect CSV file paths
     let csv_directory = format!("benches/csv/level_{}/{}_chunks", LEVELS, CHUNK);
     let csv_file_paths: Vec<String> = fs::read_dir(csv_directory)?
         .filter_map(|entry| {
@@ -35,7 +44,7 @@ async fn main() -> Result<(), Box<dyn Error>> {
         .collect();
 
     println!(
-        "LEVELS: {}, N_CURRENCIES: {}, number_of_csv: {}, number_of_workers: {}, ",
+        "LEVELS: {}, N_CURRENCIES: {}, chunk: {}, number_of_workers: {}, ",
         LEVELS,
         N_CURRENCIES,
         csv_file_paths.len(),
@@ -46,17 +55,19 @@ async fn main() -> Result<(), Box<dyn Error>> {
     let start = Instant::now();
 
     let spawner = CloudSpawner::new(None, worker_node_urls.clone(), 4000);
-    
+
     let orchestrator =
         Orchestrator::<N_CURRENCIES, N_BYTES>::new(Box::new(spawner), csv_file_paths);
 
-    let _aggregation_merkle_sum_tree = orchestrator
+    let aggregation_merkle_sum_tree = orchestrator
         .create_aggregation_mst(worker_node_urls.len())
         .await
         .unwrap();
+
     println!(
         "Time to create aggregation merkle sum tree: {:?} s",
         start.elapsed()
     );
+    println!("aggregation_mst root: {:?}", aggregation_merkle_sum_tree.root());
     Ok(())
 }