add get_multiple_accounts to BenchTpsClient (#27379)

bench-tps/src/bench_tps_client.rs

@@ -91,6 +91,8 @@ pub trait BenchTpsClient {
         pubkey: &Pubkey,
         commitment_config: CommitmentConfig,
     ) -> Result<Account>;
+
+    fn get_multiple_accounts(&self, pubkeys: &[Pubkey]) -> Result<Vec<Option<Account>>>;
 }
 
 mod bank_client;

bench-tps/src/bench_tps_client/bank_client.rs

@@ -107,4 +107,8 @@ impl BenchTpsClient for BankClient {
                 })
             })
     }
+
+    fn get_multiple_accounts(&self, _pubkeys: &[Pubkey]) -> Result<Vec<Option<Account>>> {
+        unimplemented!("BankClient doesn't support get_multiple_accounts");
+    }
 }

bench-tps/src/bench_tps_client/rpc_client.rs

@@ -99,4 +99,8 @@ impl BenchTpsClient for RpcClient {
                 })
             })
     }
+
+    fn get_multiple_accounts(&self, pubkeys: &[Pubkey]) -> Result<Vec<Option<Account>>> {
+        RpcClient::get_multiple_accounts(self, pubkeys).map_err(|err| err.into())
+    }
 }

bench-tps/src/bench_tps_client/thin_client.rs

@@ -104,4 +104,10 @@ impl BenchTpsClient for ThinClient {
                 })
             })
     }
+
+    fn get_multiple_accounts(&self, pubkeys: &[Pubkey]) -> Result<Vec<Option<Account>>> {
+        self.rpc_client()
+            .get_multiple_accounts(pubkeys)
+            .map_err(|err| err.into())
+    }
 }

bench-tps/src/bench_tps_client/tpu_client.rs

@@ -118,4 +118,10 @@ impl BenchTpsClient for TpuClient {
                 })
             })
     }
+
+    fn get_multiple_accounts(&self, pubkeys: &[Pubkey]) -> Result<Vec<Option<Account>>> {
+        self.rpc_client()
+            .get_multiple_accounts(pubkeys)
+            .map_err(|err| err.into())
+    }
 }