Optimize benchmark performance - reduce sample counts and measurement times for CI compatibility

Co-authored-by: 0xrinegade <[email protected]>

Author: Copilot

benches/http_api_bench.rs

@@ -61,12 +61,18 @@ fn bench_mcp_initialization(c: &mut Criterion) {
         }
     });
 
-    c.bench_function("mcp_initialize", |b| {
+    let mut group = c.benchmark_group("mcp_protocol");
+    group.sample_size(15);
+    group.measurement_time(Duration::from_secs(15));
+    
+    group.bench_function("initialize", |b| {
         b.to_async(&rt).iter(|| async {
             let result = make_benchmark_request(black_box(initialize_request.clone()), port).await;
             black_box(result)
         })
     });
+    
+    group.finish();
 }
 
 /// Benchmark tools list retrieval
@@ -99,12 +105,18 @@ fn bench_tools_list(c: &mut Criterion) {
         "method": "tools/list"
     });
 
-    c.bench_function("tools_list", |b| {
+    let mut group = c.benchmark_group("mcp_tools");
+    group.sample_size(15);
+    group.measurement_time(Duration::from_secs(15));
+    
+    group.bench_function("list", |b| {
         b.to_async(&rt).iter(|| async {
             let result = make_benchmark_request(black_box(tools_request.clone()), port).await;
             black_box(result)
         })
     });
+    
+    group.finish();
 }
 
 /// Benchmark different RPC tool calls
@@ -132,6 +144,8 @@ fn bench_rpc_tool_calls(c: &mut Criterion) {
     });
 
     let mut group = c.benchmark_group("rpc_tool_calls");
+    group.sample_size(10);
+    group.measurement_time(Duration::from_secs(20));
 
     // Benchmark simple methods
     let simple_methods = vec![
@@ -214,6 +228,8 @@ fn bench_concurrent_requests(c: &mut Criterion) {
     });
 
     let mut group = c.benchmark_group("concurrent_requests");
+    group.sample_size(10);
+    group.measurement_time(Duration::from_secs(15));
 
     for concurrency in [1, 5, 10, 20].iter() {
         group.bench_with_input(BenchmarkId::new("getHealth", concurrency), concurrency, |b, &concurrency| {
@@ -254,7 +270,11 @@ fn bench_health_endpoint(c: &mut Criterion) {
         setup_benchmark_server().await.expect("Failed to setup server")
     });
 
-    c.bench_function("health_endpoint", |b| {
+    let mut group = c.benchmark_group("endpoints");
+    group.sample_size(15);
+    group.measurement_time(Duration::from_secs(10));
+    
+    group.bench_function("health", |b| {
         b.to_async(&rt).iter(|| async {
             let client = reqwest::Client::new();
             let response = client
@@ -265,6 +285,8 @@ fn bench_health_endpoint(c: &mut Criterion) {
             black_box(response.text().await)
         })
     });
+    
+    group.finish();
 }
 
 /// Benchmark metrics endpoint
@@ -275,7 +297,11 @@ fn bench_metrics_endpoint(c: &mut Criterion) {
         setup_benchmark_server().await.expect("Failed to setup server")
     });
 
-    c.bench_function("metrics_endpoint", |b| {
+    let mut group = c.benchmark_group("endpoints");
+    group.sample_size(15);
+    group.measurement_time(Duration::from_secs(10));
+    
+    group.bench_function("metrics", |b| {
         b.to_async(&rt).iter(|| async {
             let client = reqwest::Client::new();
             let response = client
@@ -286,6 +312,8 @@ fn bench_metrics_endpoint(c: &mut Criterion) {
             black_box(response.text().await)
         })
     });
+    
+    group.finish();
 }
 
 criterion_group!(

benches/rpc_methods_bench.rs

@@ -58,6 +58,8 @@ fn bench_system_methods(c: &mut Criterion) {
     });
 
     let mut group = c.benchmark_group("system_methods");
+    group.sample_size(10);
+    group.measurement_time(Duration::from_secs(20));
 
     let system_methods = vec![
         "getHealth",
@@ -119,6 +121,8 @@ fn bench_account_methods(c: &mut Criterion) {
     });
 
     let mut group = c.benchmark_group("account_methods");
+    group.sample_size(10);
+    group.measurement_time(Duration::from_secs(20));
 
     let test_pubkey = "11111111111111111111111111111112"; // System program
 
@@ -178,6 +182,8 @@ fn bench_block_transaction_methods(c: &mut Criterion) {
     });
 
     let mut group = c.benchmark_group("block_transaction_methods");
+    group.sample_size(10);
+    group.measurement_time(Duration::from_secs(20));
 
     let block_tx_methods = vec![
         ("getLatestBlockhash", json!({})),
@@ -236,6 +242,8 @@ fn bench_token_methods(c: &mut Criterion) {
     });
 
     let mut group = c.benchmark_group("token_methods");
+    group.sample_size(10);
+    group.measurement_time(Duration::from_secs(20));
 
     let token_program = "TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA"; // SPL Token program
 
@@ -293,6 +301,8 @@ fn bench_error_handling(c: &mut Criterion) {
     });
 
     let mut group = c.benchmark_group("error_handling");
+    group.sample_size(10);
+    group.measurement_time(Duration::from_secs(15));
 
     // Test invalid method names
     let invalid_method_request = json!({

benches/websocket_bench.rs

@@ -14,11 +14,7 @@ static WS_BENCHMARK_SERVER: OnceLock<u16> = OnceLock::new();
 async fn setup_shared_websocket_benchmark_server() -> Result<(tokio::task::JoinHandle<()>, u16), Box<dyn std::error::Error + Send + Sync>> {
     let port = 9003;
 
-    let mut config = Config::load().map_err(|e| format!("Failed to load config: {e}"))?;
-    // Use mock RPC endpoint for benchmarks
-    config.rpc_url = "http://localhost:8899".to_string();
-    config.timeouts.websocket_connection_seconds = 5;
-    config.timeouts.websocket_message_seconds = 5;
+    let config = Config::load().map_err(|e| format!("Failed to load config: {e}"))?;
 
     let config_arc = Arc::new(config);
 
@@ -54,12 +50,18 @@ fn bench_websocket_connection(c: &mut Criterion) {
     let rt = Runtime::new().unwrap();
     let port = get_websocket_benchmark_server_port();
 
-    c.bench_function("websocket_connection", |b| {
+    let mut group = c.benchmark_group("websocket_connection");
+    group.sample_size(15);
+    group.measurement_time(Duration::from_secs(10));
+    
+    group.bench_function("establish", |b| {
         b.to_async(&rt).iter(|| async {
             let result = connect_websocket(port).await;
             black_box(result)
         })
     });
+    
+    group.finish();
 }
 
 /// Benchmark basic WebSocket subscription (simplified)
@@ -68,6 +70,8 @@ fn bench_websocket_subscriptions(c: &mut Criterion) {
     let port = get_websocket_benchmark_server_port();
 
     let mut group = c.benchmark_group("websocket_subscriptions");
+    group.sample_size(10);
+    group.measurement_time(Duration::from_secs(15));
 
     // Only test one simple subscription method to avoid timeouts
     let subscription_methods = vec![
@@ -110,6 +114,8 @@ fn bench_websocket_error_handling(c: &mut Criterion) {
     let port = get_websocket_benchmark_server_port();
 
     let mut group = c.benchmark_group("websocket_error_handling");
+    group.sample_size(10);
+    group.measurement_time(Duration::from_secs(10));
 
     // Test invalid method
     let invalid_method_request = json!({