|
| 1 | +#![feature(test)] |
| 2 | + |
| 3 | +extern crate core; |
| 4 | +extern crate lru; |
| 5 | +extern crate rand; |
| 6 | +extern crate test; |
| 7 | + |
| 8 | +use core::num::NonZeroUsize; |
| 9 | +use rand::Rng; |
| 10 | +use test::Bencher; |
| 11 | + |
| 12 | +const REPS: usize = 1 << 10; |
| 13 | +const NUM_KEYS: usize = 1 << 20; |
| 14 | +const CAPACITY: NonZeroUsize = unsafe { NonZeroUsize::new_unchecked(1 << 17) }; |
| 15 | + |
| 16 | +macro_rules! impl_put_bench { |
| 17 | + ($name: ident, $cache: ty) => { |
| 18 | + fn $name(bencher: &mut Bencher, capacity: NonZeroUsize, num_keys: usize, reps: usize) { |
| 19 | + let mut rng = rand::thread_rng(); |
| 20 | + let mut cache = <$cache>::new(capacity); |
| 21 | + for _ in 0..5 * capacity.get() { |
| 22 | + let key = rng.gen_range(0..num_keys); |
| 23 | + let _ = cache.put(key, ()); |
| 24 | + } |
| 25 | + bencher.iter(|| { |
| 26 | + for _ in 0..reps { |
| 27 | + let key = rng.gen_range(0..num_keys); |
| 28 | + let _ = cache.put(key, ()); |
| 29 | + } |
| 30 | + }); |
| 31 | + } |
| 32 | + }; |
| 33 | +} |
| 34 | + |
| 35 | +macro_rules! impl_get_bench { |
| 36 | + ($name: ident, $cache: ty) => { |
| 37 | + fn $name(bencher: &mut Bencher, capacity: NonZeroUsize, num_keys: usize, reps: usize) { |
| 38 | + let mut rng = rand::thread_rng(); |
| 39 | + let mut cache = <$cache>::new(capacity); |
| 40 | + for _ in 0..5 * capacity.get() { |
| 41 | + let key = rng.gen_range(0..num_keys); |
| 42 | + let _ = cache.put(key, ()); |
| 43 | + } |
| 44 | + bencher.iter(|| { |
| 45 | + for _ in 0..reps { |
| 46 | + let key = rng.gen_range(0..num_keys); |
| 47 | + let _ = cache.get(&key); |
| 48 | + } |
| 49 | + }); |
| 50 | + } |
| 51 | + }; |
| 52 | +} |
| 53 | + |
| 54 | +impl_put_bench!(run_put_bench, lru::LruCache<usize, ()>); |
| 55 | +// impl_put_bench!(run_put_bench_lazy, lru::lazy::LruCache<usize, ()>); |
| 56 | + |
| 57 | +impl_get_bench!(run_get_bench, lru::LruCache<usize, ()>); |
| 58 | +// impl_get_bench!(run_get_bench_lazy, lru::lazy::LruCache<usize, ()>); |
| 59 | + |
| 60 | +#[bench] |
| 61 | +fn bench_put(bencher: &mut Bencher) { |
| 62 | + run_put_bench(bencher, CAPACITY, NUM_KEYS, REPS); |
| 63 | +} |
| 64 | + |
| 65 | +// #[bench] |
| 66 | +// fn bench_put_lazy(bencher: &mut Bencher) { |
| 67 | +// run_put_bench_lazy(bencher, CAPACITY, NUM_KEYS, REPS); |
| 68 | +// } |
| 69 | + |
| 70 | +#[bench] |
| 71 | +fn bench_get(bencher: &mut Bencher) { |
| 72 | + run_get_bench(bencher, CAPACITY, NUM_KEYS, REPS); |
| 73 | +} |
| 74 | + |
| 75 | +// #[bench] |
| 76 | +// fn bench_get_lazy(bencher: &mut Bencher) { |
| 77 | +// run_get_bench_lazy(bencher, CAPACITY, NUM_KEYS, REPS); |
| 78 | +// } |
0 commit comments