diff --git a/bench-tps/src/bench_tps_client.rs b/bench-tps/src/bench_tps_client.rs index 329eced8e9..3ab15bec11 100644 --- a/bench-tps/src/bench_tps_client.rs +++ b/bench-tps/src/bench_tps_client.rs @@ -91,6 +91,8 @@ pub trait BenchTpsClient { pubkey: &Pubkey, commitment_config: CommitmentConfig, ) -> Result; + + fn get_multiple_accounts(&self, pubkeys: &[Pubkey]) -> Result>>; } mod bank_client; diff --git a/bench-tps/src/bench_tps_client/bank_client.rs b/bench-tps/src/bench_tps_client/bank_client.rs index 20323656a3..a61cc1b7d7 100644 --- a/bench-tps/src/bench_tps_client/bank_client.rs +++ b/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>> { + unimplemented!("BankClient doesn't support get_multiple_accounts"); + } } diff --git a/bench-tps/src/bench_tps_client/rpc_client.rs b/bench-tps/src/bench_tps_client/rpc_client.rs index e3fdcf211c..4f5007ad97 100644 --- a/bench-tps/src/bench_tps_client/rpc_client.rs +++ b/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>> { + RpcClient::get_multiple_accounts(self, pubkeys).map_err(|err| err.into()) + } } diff --git a/bench-tps/src/bench_tps_client/thin_client.rs b/bench-tps/src/bench_tps_client/thin_client.rs index 08681c291b..7f8c287635 100644 --- a/bench-tps/src/bench_tps_client/thin_client.rs +++ b/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>> { + self.rpc_client() + .get_multiple_accounts(pubkeys) + .map_err(|err| err.into()) + } } diff --git a/bench-tps/src/bench_tps_client/tpu_client.rs b/bench-tps/src/bench_tps_client/tpu_client.rs index 52b1e74356..0b027303c3 100644 --- a/bench-tps/src/bench_tps_client/tpu_client.rs +++ b/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>> { + self.rpc_client() + .get_multiple_accounts(pubkeys) + .map_err(|err| err.into()) + } }