53 lines
1.2 KiB
Rust
53 lines
1.2 KiB
Rust
use criterion::{black_box, criterion_group, criterion_main, Criterion};
|
|
use rand;
|
|
|
|
use ip2region2::searcher::{
|
|
get_block_by_size, get_full_cache, get_vector_index_cache, search_by_ip, searcher_init,
|
|
};
|
|
|
|
fn search_by_ip_bench(c: &mut Criterion) {
|
|
c.bench_function("search_by_ip_bench", |b| {
|
|
searcher_init(None);
|
|
b.iter(|| {
|
|
search_by_ip(rand::random::<u32>()).unwrap();
|
|
})
|
|
});
|
|
}
|
|
|
|
fn get_block_by_size_bench(c: &mut Criterion) {
|
|
c.bench_function("get_block_by_size_bench", |b| {
|
|
b.iter(|| {
|
|
black_box(get_block_by_size(
|
|
get_full_cache(),
|
|
rand::random::<u16>() as usize,
|
|
4,
|
|
));
|
|
})
|
|
});
|
|
}
|
|
|
|
fn get_full_cache_bench(c: &mut Criterion) {
|
|
c.bench_function("get_full_cache_bench", |b| {
|
|
b.iter(|| {
|
|
black_box(get_full_cache());
|
|
})
|
|
});
|
|
}
|
|
|
|
fn get_vec_index_cache_bench(c: &mut Criterion) {
|
|
c.bench_function("get_vec_index_cache_bench", |b| {
|
|
b.iter(|| {
|
|
black_box(get_vector_index_cache());
|
|
})
|
|
});
|
|
}
|
|
|
|
criterion_group!(
|
|
benches,
|
|
search_by_ip_bench,
|
|
get_block_by_size_bench,
|
|
get_full_cache_bench,
|
|
get_vec_index_cache_bench,
|
|
);
|
|
criterion_main!(benches);
|