A benchmark suite for the getFirstAvailableBlock RPC call

anza-xyz · Dec 6, 2024 · 9c9de1e · 9c9de1e
1 parent b3c2274
commit 9c9de1e
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/accounts-cluster-bench/src/main.rs b/accounts-cluster-bench/src/main.rs
@@ -237,6 +237,7 @@ pub enum RpcBench {
     MultipleAccounts,
     ProgramAccounts,
     TokenAccountsByOwner,
+    FirstAvailableBlock,
 }
 
 #[derive(Debug)]
@@ -249,6 +250,7 @@ impl FromStr for RpcBench {
 
     fn from_str(s: &str) -> Result<Self, Self::Err> {
         match s {
+            "first-available-block" => Ok(RpcBench::FirstAvailableBlock),
             "slot" => Ok(RpcBench::Slot),
             "multiple-accounts" => Ok(RpcBench::MultipleAccounts),
             "token-accounts-by-owner" => Ok(RpcBench::TokenAccountsByOwner),
@@ -340,6 +342,25 @@ fn run_rpc_bench_loop(
             break;
         }
         match rpc_bench {
+            RpcBench::FirstAvailableBlock => {
+                let mut rpc_time = Measure::start("rpc-get-first-available-block");
+                match client.get_first_available_block() {
+                    Ok(_slot) => {
+                        rpc_time.stop();
+                        stats.success += 1;
+                        stats.total_success_time_us += rpc_time.as_us();
+                    }
+                    Err(e) => {
+                        rpc_time.stop();
+                        stats.total_errors_time_us += rpc_time.as_us();
+                        stats.errors += 1;
+                        if last_error.elapsed().as_secs() > 2 {
+                            info!("get_first_available_block error: {:?}", e);
+                            last_error = Instant::now();
+                        }
+                    }
+                }
+            }
             RpcBench::Slot => {
                 let mut rpc_time = Measure::start("rpc-get-slot");
                 match client.get_slot() {