04787be8b1
Broadcast stage and retransmit stage should arrange nodes on turbine broadcast tree in exactly same order. Additionally any changes to this ordering (e.g. updating how unstaked nodes are handled) requires feature gating to keep the cluster in sync. Current implementation is scattered out over several public methods and exposes too much of implementation details (e.g. usize indices into peers vector) which makes code changes and checking for feature activations more difficult. This commit encapsulates turbine peer computations into a new struct, and only exposes two public methods, get_broadcast_peer and get_retransmit_peers, for call-sites.
58 lines
1.8 KiB
Rust
58 lines
1.8 KiB
Rust
#![feature(test)]
|
|
|
|
extern crate test;
|
|
|
|
use rand::{thread_rng, Rng};
|
|
use solana_core::{
|
|
broadcast_stage::{broadcast_metrics::TransmitShredsStats, broadcast_shreds, BroadcastStage},
|
|
cluster_nodes::ClusterNodes,
|
|
};
|
|
use solana_gossip::{
|
|
cluster_info::{ClusterInfo, Node},
|
|
contact_info::ContactInfo,
|
|
};
|
|
use solana_ledger::shred::Shred;
|
|
use solana_sdk::pubkey;
|
|
use solana_sdk::timing::timestamp;
|
|
use std::{
|
|
collections::HashMap,
|
|
net::UdpSocket,
|
|
sync::{atomic::AtomicU64, Arc},
|
|
};
|
|
use test::Bencher;
|
|
|
|
#[bench]
|
|
fn broadcast_shreds_bench(bencher: &mut Bencher) {
|
|
solana_logger::setup();
|
|
let leader_pubkey = pubkey::new_rand();
|
|
let leader_info = Node::new_localhost_with_pubkey(&leader_pubkey);
|
|
let cluster_info = ClusterInfo::new_with_invalid_keypair(leader_info.info);
|
|
let socket = UdpSocket::bind("0.0.0.0:0").unwrap();
|
|
|
|
const NUM_SHREDS: usize = 32;
|
|
let shreds = vec![Shred::new_empty_data_shred(); NUM_SHREDS];
|
|
let mut stakes = HashMap::new();
|
|
const NUM_PEERS: usize = 200;
|
|
for _ in 0..NUM_PEERS {
|
|
let id = pubkey::new_rand();
|
|
let contact_info = ContactInfo::new_localhost(&id, timestamp());
|
|
cluster_info.insert_info(contact_info);
|
|
stakes.insert(id, thread_rng().gen_range(1, NUM_PEERS) as u64);
|
|
}
|
|
let cluster_info = Arc::new(cluster_info);
|
|
let cluster_nodes = ClusterNodes::<BroadcastStage>::new(&cluster_info, &stakes);
|
|
let shreds = Arc::new(shreds);
|
|
let last_datapoint = Arc::new(AtomicU64::new(0));
|
|
bencher.iter(move || {
|
|
let shreds = shreds.clone();
|
|
broadcast_shreds(
|
|
&socket,
|
|
&shreds,
|
|
&cluster_nodes,
|
|
&last_datapoint,
|
|
&mut TransmitShredsStats::default(),
|
|
)
|
|
.unwrap();
|
|
});
|
|
}
|