124 lines
4.3 KiB
Rust
124 lines
4.3 KiB
Rust
#![feature(test)]
|
|
extern crate test;
|
|
|
|
use rand::{thread_rng, Rng};
|
|
use solana_runtime::append_vec::test_utils::{create_test_account, get_append_vec_path};
|
|
use solana_runtime::append_vec::AppendVec;
|
|
use std::sync::{Arc, Mutex};
|
|
use std::thread::sleep;
|
|
use std::thread::spawn;
|
|
use std::time::Duration;
|
|
use test::Bencher;
|
|
|
|
#[bench]
|
|
fn append_vec_append(bencher: &mut Bencher) {
|
|
let path = get_append_vec_path("bench_append");
|
|
let vec = AppendVec::new(&path.path, true, 64 * 1024);
|
|
bencher.iter(|| {
|
|
let (meta, account) = create_test_account(0);
|
|
if vec.append_account(meta, &account).is_none() {
|
|
vec.reset();
|
|
}
|
|
});
|
|
}
|
|
|
|
fn add_test_accounts(vec: &AppendVec, size: usize) -> Vec<(usize, usize)> {
|
|
(0..size)
|
|
.into_iter()
|
|
.filter_map(|sample| {
|
|
let (meta, account) = create_test_account(sample);
|
|
vec.append_account(meta, &account).map(|pos| (sample, pos))
|
|
})
|
|
.collect()
|
|
}
|
|
|
|
#[bench]
|
|
fn append_vec_sequential_read(bencher: &mut Bencher) {
|
|
let path = get_append_vec_path("seq_read");
|
|
let vec = AppendVec::new(&path.path, true, 64 * 1024);
|
|
let size = 1_000;
|
|
let mut indexes = add_test_accounts(&vec, size);
|
|
bencher.iter(|| {
|
|
let (sample, pos) = indexes.pop().unwrap();
|
|
let (account, _next) = vec.get_account(pos).unwrap();
|
|
let (_meta, test) = create_test_account(sample);
|
|
assert_eq!(account.data, test.data.as_slice());
|
|
indexes.push((sample, pos));
|
|
});
|
|
}
|
|
#[bench]
|
|
fn append_vec_random_read(bencher: &mut Bencher) {
|
|
let path = get_append_vec_path("random_read");
|
|
let vec = AppendVec::new(&path.path, true, 64 * 1024);
|
|
let size = 1_000;
|
|
let indexes = add_test_accounts(&vec, size);
|
|
bencher.iter(|| {
|
|
let random_index: usize = thread_rng().gen_range(0, indexes.len());
|
|
let (sample, pos) = &indexes[random_index];
|
|
let (account, _next) = vec.get_account(*pos).unwrap();
|
|
let (_meta, test) = create_test_account(*sample);
|
|
assert_eq!(account.data, test.data.as_slice());
|
|
});
|
|
}
|
|
|
|
#[bench]
|
|
fn append_vec_concurrent_append_read(bencher: &mut Bencher) {
|
|
let path = get_append_vec_path("concurrent_read");
|
|
let vec = Arc::new(AppendVec::new(&path.path, true, 1024 * 1024));
|
|
let vec1 = vec.clone();
|
|
let indexes: Arc<Mutex<Vec<(usize, usize)>>> = Arc::new(Mutex::new(vec![]));
|
|
let indexes1 = indexes.clone();
|
|
spawn(move || loop {
|
|
let sample = indexes1.lock().unwrap().len();
|
|
let (meta, account) = create_test_account(sample);
|
|
if let Some(pos) = vec1.append_account(meta, &account) {
|
|
indexes1.lock().unwrap().push((sample, pos))
|
|
} else {
|
|
break;
|
|
}
|
|
});
|
|
while indexes.lock().unwrap().is_empty() {
|
|
sleep(Duration::from_millis(100));
|
|
}
|
|
bencher.iter(|| {
|
|
let len = indexes.lock().unwrap().len();
|
|
let random_index: usize = thread_rng().gen_range(0, len);
|
|
let (sample, pos) = indexes.lock().unwrap().get(random_index).unwrap().clone();
|
|
let (account, _next) = vec.get_account(pos).unwrap();
|
|
let (_meta, test) = create_test_account(sample);
|
|
assert_eq!(account.data, test.data.as_slice());
|
|
});
|
|
}
|
|
|
|
#[bench]
|
|
fn append_vec_concurrent_read_append(bencher: &mut Bencher) {
|
|
let path = get_append_vec_path("concurrent_read");
|
|
let vec = Arc::new(AppendVec::new(&path.path, true, 1024 * 1024));
|
|
let vec1 = vec.clone();
|
|
let indexes: Arc<Mutex<Vec<(usize, usize)>>> = Arc::new(Mutex::new(vec![]));
|
|
let indexes1 = indexes.clone();
|
|
spawn(move || loop {
|
|
let len = indexes1.lock().unwrap().len();
|
|
if len == 0 {
|
|
continue;
|
|
}
|
|
let random_index: usize = thread_rng().gen_range(0, len + 1);
|
|
let (sample, pos) = indexes1
|
|
.lock()
|
|
.unwrap()
|
|
.get(random_index % len)
|
|
.unwrap()
|
|
.clone();
|
|
let (account, _next) = vec1.get_account(pos).unwrap();
|
|
let (_meta, test) = create_test_account(sample);
|
|
assert_eq!(account.data, test.data.as_slice());
|
|
});
|
|
bencher.iter(|| {
|
|
let sample: usize = thread_rng().gen_range(0, 256);
|
|
let (meta, account) = create_test_account(sample);
|
|
if let Some(pos) = vec.append_account(meta, &account) {
|
|
indexes.lock().unwrap().push((sample, pos))
|
|
}
|
|
});
|
|
}
|