diff --git a/bench/src/cli.rs b/bench/src/cli.rs index df3a0a72..f7b6cd8d 100644 --- a/bench/src/cli.rs +++ b/bench/src/cli.rs @@ -15,4 +15,8 @@ pub struct Args { /// Metrics output file name #[arg(short = 'm', long, default_value_t = String::from("metrics.csv"))] pub metrics_file_name: String, + /// Lite Rpc Address + #[arg(short = 'l', long, default_value_t = String::from("http://127.0.0.1:8890"))] + pub lite_rpc_addr: String, + } diff --git a/bench/src/main.rs b/bench/src/main.rs index a98c807f..73795a7e 100644 --- a/bench/src/main.rs +++ b/bench/src/main.rs @@ -26,17 +26,18 @@ async fn main() { runs, run_interval_ms, metrics_file_name, + lite_rpc_addr, } = Args::parse(); let mut run_interval_ms = tokio::time::interval(Duration::from_millis(run_interval_ms)); + info!("Connecting to {lite_rpc_addr}"); + let rpc_client = Arc::new(RpcClient::new_with_commitment( - DEFAULT_LITE_RPC_ADDR.to_string(), + lite_rpc_addr, CommitmentConfig::confirmed(), )); - info!("Connecting to {DEFAULT_LITE_RPC_ADDR}"); - let bench_helper = BenchHelper { rpc_client }; let mut csv_writer = csv::Writer::from_path(metrics_file_name).unwrap();