Add target-node flag to bench-tps (#10876)

This commit is contained in:
sakridge
2020-07-01 21:10:12 -07:00
committed by GitHub
parent 0d6ddf578d
commit 3884323cb6
2 changed files with 31 additions and 1 deletions

View File

@@ -32,6 +32,7 @@ fn main() {
use_move,
multi_client,
num_lamports_per_account,
target_node,
..
} = &cli_config;
@@ -82,6 +83,19 @@ fn main() {
exit(1);
}
Arc::new(client)
} else if let Some(target_node) = target_node {
info!("Searching for target_node: {:?}", target_node);
let mut target_client = None;
for node in nodes {
if node.id == *target_node {
target_client = Some(Arc::new(get_client(&[node])));
break;
}
}
target_client.unwrap_or_else(|| {
eprintln!("Target node {} not found", target_node);
exit(1);
})
} else {
Arc::new(get_client(&nodes))
};