gaspersic/solana
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Refactor thin_client::create_client (#24067)

Browse Source 
Refactor the thin_client::create_client to take addresses separately instead of as a tuple

Co-authored-by: Bijie Zhu <bijiezhu@Bijies-MBP.cable.rcn.com>
This commit is contained in:
BG Zhu
2022-04-06 11:03:38 -04:00
committed by GitHub
parent a38bd4acc8
commit 22224127e0
6 changed files with 44 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
local-cluster/src/cluster_tests.rs
View File

@ -60,7 +60,8 @@ pub fn spend_and_verify_all_nodes<S: ::std::hash::BuildHasher + Sync + Send>(
return;
}
let random_keypair = Keypair::new();
let client = create_client(ingress_node.client_facing_addr());
let (rpc, tpu) = ingress_node.client_facing_addr();
let client = create_client(rpc, tpu);
let bal = client
.poll_get_balance_with_commitment(
&funding_keypair.pubkey(),
@ -81,7 +82,8 @@ pub fn spend_and_verify_all_nodes<S: ::std::hash::BuildHasher + Sync + Send>(
if ignore_nodes.contains(&validator.id) {
continue;
}
let client = create_client(validator.client_facing_addr());
let (rpc, tpu) = validator.client_facing_addr();
let client = create_client(rpc, tpu);
client.poll_for_signature_confirmation(&sig, confs).unwrap();
}
});
@ -91,7 +93,8 @@ pub fn verify_balances<S: ::std::hash::BuildHasher>(
expected_balances: HashMap<Pubkey, u64, S>,
node: &ContactInfo,
) {
let client = create_client(node.client_facing_addr());
let (rpc, tpu) = node.client_facing_addr();
let client = create_client(rpc, tpu);
for (pk, b) in expected_balances {
let bal = client
.poll_get_balance_with_commitment(&pk, CommitmentConfig::processed())
@ -106,7 +109,8 @@ pub fn send_many_transactions(
max_tokens_per_transfer: u64,
num_txs: u64,
) -> HashMap<Pubkey, u64> {
let client = create_client(node.client_facing_addr());
let (rpc, tpu) = node.client_facing_addr();
let client = create_client(rpc, tpu);
let mut expected_balances = HashMap::new();
for _ in 0..num_txs {
let random_keypair = Keypair::new();
@ -197,7 +201,9 @@ pub fn kill_entry_and_spend_and_verify_rest(
let cluster_nodes =
discover_cluster(&entry_point_info.gossip, nodes, socket_addr_space).unwrap();
assert!(cluster_nodes.len() >= nodes);
let client = create_client(entry_point_info.client_facing_addr());
let (rpc, tpu) = entry_point_info.client_facing_addr();
let client = create_client(rpc, tpu);
// sleep long enough to make sure we are in epoch 3
let first_two_epoch_slots = MINIMUM_SLOTS_PER_EPOCH * (3 + 1);
@ -225,7 +231,8 @@ pub fn kill_entry_and_spend_and_verify_rest(
continue;
}
let client = create_client(ingress_node.client_facing_addr());
let (rpc, tpu) = ingress_node.client_facing_addr();
let client = create_client(rpc, tpu);
let balance = client
.poll_get_balance_with_commitment(
&funding_keypair.pubkey(),
@ -296,7 +303,8 @@ pub fn check_for_new_roots(num_new_roots: usize, contact_infos: &[ContactInfo],
assert!(loop_start.elapsed() < loop_timeout);
for (i, ingress_node) in contact_infos.iter().enumerate() {
let client = create_client(ingress_node.client_facing_addr());
let (rpc, tpu) = ingress_node.client_facing_addr();
let client = create_client(rpc, tpu);
let root_slot = client
.get_slot_with_commitment(CommitmentConfig::finalized())
.unwrap_or(0);
@ -327,7 +335,8 @@ pub fn check_no_new_roots(
.iter()
.enumerate()
.map(|(i, ingress_node)| {
let client = create_client(ingress_node.client_facing_addr());
let (rpc, tpu) = ingress_node.client_facing_addr();
let client = create_client(rpc, tpu);
let initial_root = client
.get_slot()
.unwrap_or_else(|_| panic!("get_slot for {} failed", ingress_node.id));
@ -345,7 +354,8 @@ pub fn check_no_new_roots(
let mut reached_end_slot = false;
loop {
for contact_info in contact_infos {
let client = create_client(contact_info.client_facing_addr());
let (rpc, tpu) = contact_info.client_facing_addr();
let client = create_client(rpc, tpu);
current_slot = client
.get_slot_with_commitment(CommitmentConfig::processed())
.unwrap_or_else(|_| panic!("get_slot for {} failed", contact_infos[0].id));
@ -367,7 +377,8 @@ pub fn check_no_new_roots(
}
for (i, ingress_node) in contact_infos.iter().enumerate() {
let client = create_client(ingress_node.client_facing_addr());
let (rpc, tpu) = ingress_node.client_facing_addr();
let client = create_client(rpc, tpu);
assert_eq!(
client
.get_slot()
@ -387,7 +398,8 @@ fn poll_all_nodes_for_signature(
if validator.id == entry_point_info.id {
continue;
}
let client = create_client(validator.client_facing_addr());
let (rpc, tpu) = validator.client_facing_addr();
let client = create_client(rpc, tpu);
client.poll_for_signature_confirmation(sig, confs)?;
}

13
local-cluster/src/local_cluster.rs
View File

@ -388,7 +388,8 @@ impl LocalCluster {
mut voting_keypair: Option<Arc<Keypair>>,
socket_addr_space: SocketAddrSpace,
) -> Pubkey {
let client = create_client(self.entry_point_info.client_facing_addr());
let (rpc, tpu) = self.entry_point_info.client_facing_addr();
let client = create_client(rpc, tpu);
// Must have enough tokens to fund vote account and set delegate
let should_create_vote_pubkey = voting_keypair.is_none();
@ -472,7 +473,8 @@ impl LocalCluster {
}
pub fn transfer(&self, source_keypair: &Keypair, dest_pubkey: &Pubkey, lamports: u64) -> u64 {
let client = create_client(self.entry_point_info.client_facing_addr());
let (rpc, tpu) = self.entry_point_info.client_facing_addr();
let client = create_client(rpc, tpu);
Self::transfer_with_client(&client, source_keypair, dest_pubkey, lamports)
}
@ -694,9 +696,10 @@ impl Cluster for LocalCluster {
}
fn get_validator_client(&self, pubkey: &Pubkey) -> Option<ThinClient> {
self.validators
.get(pubkey)
.map(|f| create_client(f.info.contact_info.client_facing_addr()))
self.validators.get(pubkey).map(|f| {
let (rpc, tpu) = f.info.contact_info.client_facing_addr();
create_client(rpc, tpu)
})
}
fn exit_node(&mut self, pubkey: &Pubkey) -> ClusterValidatorInfo {