Move gossip modules into solana-gossip crate (#17352)
* Move gossip modules to solana-gossip * Update Protocol abi digest due to move * Move gossip benches and hook up CI * Remove unneeded Result entries * Single use statements
This commit is contained in:
75
gossip/benches/crds_shards.rs
Normal file
75
gossip/benches/crds_shards.rs
Normal file
@@ -0,0 +1,75 @@
|
||||
#![feature(test)]
|
||||
|
||||
extern crate test;
|
||||
|
||||
use {
|
||||
rand::{thread_rng, Rng},
|
||||
solana_gossip::{
|
||||
crds::{Crds, VersionedCrdsValue},
|
||||
crds_shards::CrdsShards,
|
||||
crds_value::CrdsValue,
|
||||
},
|
||||
solana_sdk::timing::timestamp,
|
||||
std::iter::repeat_with,
|
||||
test::Bencher,
|
||||
};
|
||||
|
||||
const CRDS_SHARDS_BITS: u32 = 8;
|
||||
|
||||
fn new_test_crds_value<R: Rng>(rng: &mut R) -> VersionedCrdsValue {
|
||||
let value = CrdsValue::new_rand(rng, None);
|
||||
let label = value.label();
|
||||
let mut crds = Crds::default();
|
||||
crds.insert(value, timestamp()).unwrap();
|
||||
crds.get(&label).cloned().unwrap()
|
||||
}
|
||||
|
||||
fn bench_crds_shards_find(bencher: &mut Bencher, num_values: usize, mask_bits: u32) {
|
||||
let mut rng = thread_rng();
|
||||
let values: Vec<_> = repeat_with(|| new_test_crds_value(&mut rng))
|
||||
.take(num_values)
|
||||
.collect();
|
||||
let mut shards = CrdsShards::new(CRDS_SHARDS_BITS);
|
||||
for (index, value) in values.iter().enumerate() {
|
||||
assert!(shards.insert(index, value));
|
||||
}
|
||||
bencher.iter(|| {
|
||||
let mask = rng.gen();
|
||||
let _hits = shards.find(mask, mask_bits).count();
|
||||
});
|
||||
}
|
||||
|
||||
#[bench]
|
||||
fn bench_crds_shards_find_0(bencher: &mut Bencher) {
|
||||
bench_crds_shards_find(bencher, 100_000, 0);
|
||||
}
|
||||
|
||||
#[bench]
|
||||
fn bench_crds_shards_find_1(bencher: &mut Bencher) {
|
||||
bench_crds_shards_find(bencher, 100_000, 1);
|
||||
}
|
||||
|
||||
#[bench]
|
||||
fn bench_crds_shards_find_3(bencher: &mut Bencher) {
|
||||
bench_crds_shards_find(bencher, 100_000, 3);
|
||||
}
|
||||
|
||||
#[bench]
|
||||
fn bench_crds_shards_find_5(bencher: &mut Bencher) {
|
||||
bench_crds_shards_find(bencher, 100_000, 5);
|
||||
}
|
||||
|
||||
#[bench]
|
||||
fn bench_crds_shards_find_7(bencher: &mut Bencher) {
|
||||
bench_crds_shards_find(bencher, 100_000, 7);
|
||||
}
|
||||
|
||||
#[bench]
|
||||
fn bench_crds_shards_find_8(bencher: &mut Bencher) {
|
||||
bench_crds_shards_find(bencher, 100_000, 8);
|
||||
}
|
||||
|
||||
#[bench]
|
||||
fn bench_crds_shards_find_9(bencher: &mut Bencher) {
|
||||
bench_crds_shards_find(bencher, 100_000, 9);
|
||||
}
|
Reference in New Issue
Block a user