mirror of
https://github.com/qdrant/qdrant.git
synced 2026-08-03 08:30:55 -05:00
3e536347e1
* Bump Rust edition to 2024 * gen is a reserved keyword now * Remove ref mut on references * Mark extern C as unsafe * Wrap unsafe function bodies in unsafe block * Geo hash implements Copy, don't reference but pass by value instead * Replace secluded self import with parent * Update execute_cluster_read_operation with new match semantics * Fix lifetime issue * Replace map_or with is_none_or * set_var is unsafe now * Reformat
72 lines
2.3 KiB
Rust
72 lines
2.3 KiB
Rust
#[cfg(not(target_os = "windows"))]
|
|
mod prof;
|
|
|
|
use std::collections::{BTreeMap, HashMap};
|
|
|
|
use criterion::{Criterion, criterion_group, criterion_main};
|
|
use rand::SeedableRng;
|
|
use rand::rngs::StdRng;
|
|
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();
|
|
targets = small_map_obj
|
|
}
|
|
|
|
criterion_main!(benches);
|