44 lines
1.3 KiB
Rust
44 lines
1.3 KiB
Rust
|
use client::database::Database;
|
||
|
|
use criterion::{criterion_group, criterion_main, Criterion};
|
||
|
|
use helios::prelude::FileDB;
|
||
|
|
use tempfile::tempdir;
|
||
|
|
|
||
|
|
mod harness;
|
||
|
|
|
||
|
|
criterion_main!(file_db);
|
||
|
|
criterion_group! {
|
||
|
|
name = file_db;
|
||
|
|
config = Criterion::default();
|
||
|
|
targets = save_checkpoint, load_checkpoint
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Benchmark saving/writing a checkpoint to the file db.
|
||
|
|
pub fn save_checkpoint(c: &mut Criterion) {
|
||
|
|
c.bench_function("save_checkpoint", |b| {
|
||
|
|
let checkpoint = vec![1, 2, 3];
|
||
|
|
b.iter(|| {
|
||
|
|
let temp_dir = tempdir().unwrap().into_path();
|
||
|
|
let db = FileDB::new(temp_dir);
|
||
|
|
db.save_checkpoint(checkpoint.clone()).unwrap();
|
||
|
|
})
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Benchmark loading a checkpoint from the file db.
|
||
|
|
pub fn load_checkpoint(c: &mut Criterion) {
|
||
|
|
c.bench_function("load_checkpoint", |b| {
|
||
|
|
// First write to the db
|
||
|
|
let temp_dir = tempdir().unwrap().into_path();
|
||
|
|
let db = FileDB::new(temp_dir.clone());
|
||
|
|
let written_checkpoint = vec![1, 2, 3];
|
||
|
|
db.save_checkpoint(written_checkpoint.clone()).unwrap();
|
||
|
|
|
||
|
|
// Then read from the db
|
||
|
|
b.iter(|| {
|
||
|
|
let db = FileDB::new(temp_dir.clone());
|
||
|
|
let checkpoint = db.load_checkpoint().unwrap();
|
||
|
|
assert_eq!(checkpoint, written_checkpoint.clone());
|
||
|
|
})
|
||
|
|
});
|
||
|
|
}
|