Add RpcClient::get_multiple_accounts_with_config()

mvines · Sep 3, 2021 · 601a2ea · 601a2ea
1 parent 2efc519
commit 601a2ea
Showing 1 changed file with 16 additions and 5 deletions.
diff --git a/client/src/rpc_client.rs b/client/src/rpc_client.rs
@@ -2966,11 +2966,22 @@ impl RpcClient {
         pubkeys: &[Pubkey],
         commitment_config: CommitmentConfig,
     ) -> RpcResult<Vec<Option<Account>>> {
-        let config = RpcAccountInfoConfig {
-            encoding: Some(UiAccountEncoding::Base64Zstd),
-            commitment: Some(self.maybe_map_commitment(commitment_config)?),
-            data_slice: None,
-        };
+        self.get_multiple_accounts_with_config(
+            pubkeys,
+            RpcAccountInfoConfig {
+                encoding: Some(UiAccountEncoding::Base64Zstd),
+                commitment: Some(self.maybe_map_commitment(commitment_config)?),
+                data_slice: None,
+            },
+        )
+    }
+
+    pub fn get_multiple_accounts_with_config(
+        &self,
+        pubkeys: &[Pubkey],
+        mut config: RpcAccountInfoConfig,
+    ) -> RpcResult<Vec<Option<Account>>> {
+        config.commitment = config.commitment.or_else(|| Some(self.commitment()));
         let pubkeys: Vec<_> = pubkeys.iter().map(|pubkey| pubkey.to_string()).collect();
         let response = self.send(RpcRequest::GetMultipleAccounts, json!([pubkeys, config]))?;
         let Response {