40 lines
1.3 KiB
Rust
40 lines
1.3 KiB
Rust
#![feature(test)]
|
|
|
|
extern crate test;
|
|
|
|
use rand::{thread_rng, Rng};
|
|
use solana_core::cluster_info::{ClusterInfo, Node};
|
|
use solana_core::contact_info::ContactInfo;
|
|
use solana_sdk::pubkey::Pubkey;
|
|
use solana_sdk::timing::timestamp;
|
|
use std::collections::HashMap;
|
|
use std::net::UdpSocket;
|
|
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 mut cluster_info = ClusterInfo::new_with_invalid_keypair(leader_info.info.clone());
|
|
let socket = UdpSocket::bind("0.0.0.0:0").unwrap();
|
|
|
|
const SHRED_SIZE: usize = 1024;
|
|
const NUM_SHREDS: usize = 32;
|
|
let shreds = vec![vec![0; SHRED_SIZE]; NUM_SHREDS];
|
|
let seeds = vec![[0u8; 32]; 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);
|
|
}
|
|
bencher.iter(move || {
|
|
cluster_info
|
|
.broadcast_shreds(&socket, &shreds, &seeds, Some(&stakes))
|
|
.unwrap();
|
|
});
|
|
}
|