You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
3.2 KiB
118 lines
3.2 KiB
#![feature(test)]
|
|
|
|
extern crate libc;
|
|
extern crate lmdb;
|
|
extern crate lmdb_rkv_sys as ffi;
|
|
extern crate rand;
|
|
extern crate test;
|
|
|
|
mod utils;
|
|
|
|
use ffi::*;
|
|
use libc::size_t;
|
|
use lmdb::{Transaction, WriteFlags};
|
|
use rand::{Rng, XorShiftRng};
|
|
use std::ptr;
|
|
use test::{Bencher, black_box};
|
|
use utils::*;
|
|
|
|
#[bench]
|
|
fn bench_get_rand(b: &mut Bencher) {
|
|
let n = 100u32;
|
|
let (_dir, env) = setup_bench_db(n);
|
|
let db = env.open_db(None).unwrap();
|
|
let txn = env.begin_ro_txn().unwrap();
|
|
|
|
let mut keys: Vec<String> = (0..n).map(|n| get_key(n)).collect();
|
|
XorShiftRng::new_unseeded().shuffle(&mut keys[..]);
|
|
|
|
b.iter(|| {
|
|
let mut i = 0usize;
|
|
for key in &keys {
|
|
i = i + txn.get(db, key).unwrap().len();
|
|
}
|
|
black_box(i);
|
|
});
|
|
}
|
|
|
|
#[bench]
|
|
fn bench_get_rand_raw(b: &mut Bencher) {
|
|
let n = 100u32;
|
|
let (_dir, env) = setup_bench_db(n);
|
|
let db = env.open_db(None).unwrap();
|
|
let _txn = env.begin_ro_txn().unwrap();
|
|
|
|
let mut keys: Vec<String> = (0..n).map(|n| get_key(n)).collect();
|
|
XorShiftRng::new_unseeded().shuffle(&mut keys[..]);
|
|
|
|
let dbi = db.dbi();
|
|
let txn = _txn.txn();
|
|
|
|
let mut key_val: MDB_val = MDB_val { mv_size: 0, mv_data: ptr::null_mut() };
|
|
let mut data_val: MDB_val = MDB_val { mv_size: 0, mv_data: ptr::null_mut() };
|
|
|
|
b.iter(|| unsafe {
|
|
let mut i: size_t = 0;
|
|
for key in &keys {
|
|
key_val.mv_size = key.len() as size_t;
|
|
key_val.mv_data = key.as_bytes().as_ptr() as *mut _;
|
|
|
|
mdb_get(txn, dbi, &mut key_val, &mut data_val);
|
|
|
|
i = i + key_val.mv_size;
|
|
}
|
|
black_box(i);
|
|
});
|
|
}
|
|
|
|
#[bench]
|
|
fn bench_put_rand(b: &mut Bencher) {
|
|
let n = 100u32;
|
|
let (_dir, env) = setup_bench_db(0);
|
|
let db = env.open_db(None).unwrap();
|
|
|
|
let mut items: Vec<(String, String)> = (0..n).map(|n| (get_key(n), get_data(n))).collect();
|
|
XorShiftRng::new_unseeded().shuffle(&mut items[..]);
|
|
|
|
b.iter(|| {
|
|
let mut txn = env.begin_rw_txn().unwrap();
|
|
for &(ref key, ref data) in items.iter() {
|
|
txn.put(db, key, data, WriteFlags::empty()).unwrap();
|
|
}
|
|
txn.abort();
|
|
});
|
|
}
|
|
|
|
#[bench]
|
|
fn bench_put_rand_raw(b: &mut Bencher) {
|
|
let n = 100u32;
|
|
let (_dir, _env) = setup_bench_db(0);
|
|
let db = _env.open_db(None).unwrap();
|
|
|
|
let mut items: Vec<(String, String)> = (0..n).map(|n| (get_key(n), get_data(n))).collect();
|
|
XorShiftRng::new_unseeded().shuffle(&mut items[..]);
|
|
|
|
let dbi = db.dbi();
|
|
let env = _env.env();
|
|
|
|
let mut key_val: MDB_val = MDB_val { mv_size: 0, mv_data: ptr::null_mut() };
|
|
let mut data_val: MDB_val = MDB_val { mv_size: 0, mv_data: ptr::null_mut() };
|
|
|
|
b.iter(|| unsafe {
|
|
let mut txn: *mut MDB_txn = ptr::null_mut();
|
|
mdb_txn_begin(env, ptr::null_mut(), 0, &mut txn);
|
|
|
|
let mut i: ::libc::c_int = 0;
|
|
for &(ref key, ref data) in items.iter() {
|
|
|
|
key_val.mv_size = key.len() as size_t;
|
|
key_val.mv_data = key.as_bytes().as_ptr() as *mut _;
|
|
data_val.mv_size = data.len() as size_t;
|
|
data_val.mv_data = data.as_bytes().as_ptr() as *mut _;
|
|
|
|
i += mdb_put(txn, dbi, &mut key_val, &mut data_val, 0);
|
|
}
|
|
assert_eq!(0, i);
|
|
mdb_txn_abort(txn);
|
|
});
|
|
}
|
|
|