Speed up getLeaderSchedule RPC call by reducing pubkey duplication (#7556)

automerge

client/src/rpc_client.rs

@@ -4,7 +4,10 @@ use crate::{
     generic_rpc_client_request::GenericRpcClientRequest,
     mock_rpc_client_request::MockRpcClientRequest,
     rpc_client_request::RpcClientRequest,
-    rpc_request::{RpcContactInfo, RpcEpochInfo, RpcRequest, RpcVersionInfo, RpcVoteAccountStatus},
+    rpc_request::{
+        RpcContactInfo, RpcEpochInfo, RpcLeaderSchedule, RpcRequest, RpcVersionInfo,
+        RpcVoteAccountStatus,
+    },
 };
 use bincode::serialize;
 use log::*;
@@ -248,7 +251,7 @@ impl RpcClient {
         })
     }
 
-    pub fn get_leader_schedule(&self, slot: Option<Slot>) -> io::Result<Option<Vec<String>>> {
+    pub fn get_leader_schedule(&self, slot: Option<Slot>) -> io::Result<Option<RpcLeaderSchedule>> {
         self.get_leader_schedule_with_commitment(slot, CommitmentConfig::default())
     }
 
@@ -256,7 +259,7 @@ impl RpcClient {
         &self,
         slot: Option<Slot>,
         commitment_config: CommitmentConfig,
-    ) -> io::Result<Option<Vec<String>>> {
+    ) -> io::Result<Option<RpcLeaderSchedule>> {
         let params = slot.map(|slot| json!(slot));
         let response = self
             .client

client/src/rpc_request.rs

@@ -6,7 +6,7 @@ use solana_sdk::{
     hash::Hash,
     transaction::{Result, Transaction},
 };
-use std::{error, fmt, io, net::SocketAddr};
+use std::{collections::HashMap, error, fmt, io, net::SocketAddr};
 
 pub type RpcResponseIn<T> = JsonResult<Response<T>>;
 pub type RpcResponse<T> = io::Result<Response<T>>;
@@ -49,6 +49,9 @@ pub struct RpcContactInfo {
     pub rpc: Option<SocketAddr>,
 }
 
+/// Map of leader base58 identity pubkeys to the slot indices relative to the first epoch slot
+pub type RpcLeaderSchedule = HashMap<String, Vec<usize>>;
+
 #[derive(Serialize, Deserialize, Clone, Debug)]
 #[serde(rename_all = "camelCase")]
 pub struct RpcEpochInfo {