add get token largest accounts rpc call to client

solana-labs · Jul 29, 2022 · d310a34 · d310a34
1 parent 741076e
commit d310a34
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 0 deletions.
diff --git a/client/src/nonblocking/rpc_client.rs b/client/src/nonblocking/rpc_client.rs
@@ -5016,6 +5016,21 @@ impl RpcClient {
         .await
     }
 
+    pub async fn get_token_largest_accounts(
+        &self,
+        mint: &Pubkey,
+        commitment_config: CommitmentConfig,
+    ) -> RpcResult<Vec<RpcTokenAccountBalance>> {
+        self.send(
+            RpcRequest::GetTokenLargestAccounts,
+            json!([
+                mint.to_string(),
+                self.maybe_map_commitment(commitment_config).await?
+            ]),
+        )
+        .await
+    }
+
     pub async fn get_token_supply(&self, mint: &Pubkey) -> ClientResult<UiTokenAmount> {
         Ok(self
             .get_token_supply_with_commitment(mint, self.commitment())

diff --git a/client/src/rpc_client.rs b/client/src/rpc_client.rs
@@ -3956,6 +3956,17 @@ impl RpcClient {
         ))
     }
 
+    pub fn get_token_largest_accounts(
+        &self,
+        mint: &Pubkey,
+        commitment_config: CommitmentConfig,
+    ) -> RpcResult<Vec<RpcTokenAccountBalance>> {
+        self.invoke(
+            self.rpc_client
+                .get_token_largest_accounts(mint, commitment_config),
+        )
+    }
+
     pub fn get_token_supply(&self, mint: &Pubkey) -> ClientResult<UiTokenAmount> {
         self.invoke(self.rpc_client.get_token_supply(mint))
     }

diff --git a/client/src/rpc_request.rs b/client/src/rpc_request.rs
@@ -100,6 +100,7 @@ pub enum RpcRequest {
     GetTokenAccountBalance,
     GetTokenAccountsByDelegate,
     GetTokenAccountsByOwner,
+    GetTokenLargestAccounts,
     GetTokenSupply,
     GetTransaction,
     GetTransactionCount,
@@ -175,6 +176,7 @@ impl fmt::Display for RpcRequest {
             RpcRequest::GetTokenAccountsByDelegate => "getTokenAccountsByDelegate",
             RpcRequest::GetTokenAccountsByOwner => "getTokenAccountsByOwner",
             RpcRequest::GetTokenSupply => "getTokenSupply",
+            RpcRequest::GetTokenLargestAccounts => "getTokenLargestAccounts",
             RpcRequest::GetTransaction => "getTransaction",
             RpcRequest::GetTransactionCount => "getTransactionCount",
             RpcRequest::GetVersion => "getVersion",
@@ -322,6 +324,10 @@ mod tests {
         let test_request = RpcRequest::SendTransaction;
         let request = test_request.build_request_json(1, Value::Null);
         assert_eq!(request["method"], "sendTransaction");
+
+        let test_request = RpcRequest::GetTokenLargestAccounts;
+        let request = test_request.build_request_json(1, Value::Null);
+        assert_eq!(request["method"], "getTokenLargestAccounts");
     }
 
     #[test]