ca1a282a60
Since call-sites are calling unwrap anyways, panicking seems too punitive
for our use cases.
(cherry picked from commit 7cb3b6cbe2)
Co-authored-by: behzad nouri <behzadnouri@gmail.com>
40 lines
1.0 KiB
Rust
40 lines
1.0 KiB
Rust
#![feature(test)]
|
|
|
|
extern crate test;
|
|
|
|
use {
|
|
rand::{Rng, SeedableRng},
|
|
rand_chacha::ChaChaRng,
|
|
solana_gossip::weighted_shuffle::{weighted_shuffle, WeightedShuffle},
|
|
std::iter::repeat_with,
|
|
test::Bencher,
|
|
};
|
|
|
|
fn make_weights<R: Rng>(rng: &mut R) -> Vec<u64> {
|
|
repeat_with(|| rng.gen_range(1, 100)).take(1000).collect()
|
|
}
|
|
|
|
#[bench]
|
|
fn bench_weighted_shuffle_old(bencher: &mut Bencher) {
|
|
let mut seed = [0u8; 32];
|
|
let mut rng = rand::thread_rng();
|
|
let weights = make_weights(&mut rng);
|
|
bencher.iter(|| {
|
|
rng.fill(&mut seed[..]);
|
|
weighted_shuffle::<u64, &u64, std::slice::Iter<'_, u64>>(weights.iter(), seed);
|
|
});
|
|
}
|
|
|
|
#[bench]
|
|
fn bench_weighted_shuffle_new(bencher: &mut Bencher) {
|
|
let mut seed = [0u8; 32];
|
|
let mut rng = rand::thread_rng();
|
|
let weights = make_weights(&mut rng);
|
|
bencher.iter(|| {
|
|
rng.fill(&mut seed[..]);
|
|
WeightedShuffle::new("", &weights)
|
|
.shuffle(&mut ChaChaRng::from_seed(seed))
|
|
.collect::<Vec<_>>()
|
|
});
|
|
}
|