mirror of
https://github.com/qdrant/qdrant.git
synced 2026-09-29 09:27:53 -05:00
Andrey VasnetsovandIvan Pleshkov
4ae6552379
* tiny-map + benchmark * are you happy clippy * lower default capacity * fmt Co-authored-by: Ivan Pleshkov <pleshkov.ivan@gmail.com>
71 lines
2.3 KiB
Rust
71 lines
2.3 KiB
Rust
mod prof;
|
|
|
|
use std::collections::{BTreeMap, HashMap};
|
|
|
|
use criterion::{criterion_group, criterion_main, Criterion};
|
|
use rand::rngs::StdRng;
|
|
use rand::SeedableRng;
|
|
use segment::data_types::tiny_map::TinyMap;
|
|
use segment::fixtures::index_fixtures::random_vector;
|
|
|
|
const DIM: usize = 100;
|
|
|
|
fn small_map_obj(c: &mut Criterion) {
|
|
let mut group = c.benchmark_group("small-map-obj-group");
|
|
|
|
let mut rng = StdRng::seed_from_u64(42);
|
|
let default_key = "vector".to_string();
|
|
let default_key_2 = "vector1".to_string();
|
|
let default_key_3 = "vector2".to_string();
|
|
let random_vector = random_vector(&mut rng, DIM);
|
|
|
|
group.bench_function("hash-map", |b| {
|
|
b.iter(|| {
|
|
let mut map = HashMap::new();
|
|
map.insert(default_key.clone(), random_vector.clone());
|
|
map.insert(default_key_2.clone(), random_vector.clone());
|
|
map.insert(default_key_3.clone(), random_vector.clone());
|
|
let _ = map.get(&default_key_3);
|
|
});
|
|
});
|
|
|
|
group.bench_function("btree-map", |b| {
|
|
b.iter(|| {
|
|
let mut map = BTreeMap::new();
|
|
map.insert(default_key.clone(), random_vector.clone());
|
|
map.insert(default_key_2.clone(), random_vector.clone());
|
|
map.insert(default_key_3.clone(), random_vector.clone());
|
|
let _ = map.get(&default_key_3);
|
|
});
|
|
});
|
|
|
|
#[allow(clippy::vec_init_then_push)]
|
|
group.bench_function("vec-map", |b| {
|
|
b.iter(|| {
|
|
let mut map = Vec::with_capacity(3);
|
|
map.push((default_key.clone(), random_vector.clone()));
|
|
map.push((default_key_2.clone(), random_vector.clone()));
|
|
map.push((default_key_3.clone(), random_vector.clone()));
|
|
let _ = map.iter().find(|(k, _)| k == &default_key_3);
|
|
});
|
|
});
|
|
|
|
group.bench_function("tiny-map", |b| {
|
|
b.iter(|| {
|
|
let mut map = TinyMap::new();
|
|
map.insert(default_key.clone(), random_vector.clone());
|
|
map.insert(default_key_2.clone(), random_vector.clone());
|
|
map.insert(default_key_3.clone(), random_vector.clone());
|
|
let _ = map.get(&default_key_3);
|
|
});
|
|
});
|
|
}
|
|
|
|
criterion_group! {
|
|
name = benches;
|
|
config = Criterion::default().with_profiler(prof::FlamegraphProfiler::new(100));
|
|
targets = small_map_obj
|
|
}
|
|
|
|
criterion_main!(benches);
|