abby/kv_bench
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update scripts

Browse Source
This commit is contained in:
abbycin 2025-08-27 15:32:15 +08:00
parent ba6af2402e
commit 82f454d7ca
Signed by: abby
GPG Key ID: B636E0F0307EF8EB
5 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
logger/src/lib.rs
View File

@ -217,7 +217,7 @@ impl Logger {
self.abort_on_error.load(Relaxed) self.abort_on_error.load(Relaxed)
} }
fn lock(&self) -> MutexGuard<()> { fn lock(&'_ self) -> MutexGuard<'_, ()> {
self.mtx_shard[get_tid() as usize & (self.mtx_shard.len() - 1)] self.mtx_shard[get_tid() as usize & (self.mtx_shard.len() - 1)]
.lock() .lock()
.unwrap() .unwrap()

2
scripts/init.sh
View File

@ -1,4 +1,4 @@
#!/usr/bin/env bash #!/usr/bin/env bash
python3 -m venv . python3 -m venv .
./bin/pip3 install pandas matplotlib ./bin/pip3 install pandas matplotlib adjustText

2
scripts/mace.sh
View File

@ -34,7 +34,7 @@ function samples() {
done done
} }
echo mode,threads,key_size,value_size,insert_ratio,ops > scripts/mace.csv echo mode,threads,key_size,value_size,insert_ratio,ops,elasped > scripts/mace.csv
samples 2>> scripts/mace.csv samples 2>> scripts/mace.csv
popd popd
./bin/python plot.py mace.csv ./bin/python plot.py mace.csv

2
scripts/rocksdb.sh
View File

@ -35,7 +35,7 @@ function samples() {
done done
} }
echo mode,threads,key_size,value_size,insert_ratio,ops > ../scripts/rocksdb.csv echo mode,threads,key_size,value_size,insert_ratio,ops,elapsed > ../scripts/rocksdb.csv
samples 1>> ../scripts/rocksdb.csv samples 1>> ../scripts/rocksdb.csv
popd popd
./bin/python plot.py rocksdb.csv ./bin/python plot.py rocksdb.csv

6
src/main.rs
View File

@ -1,4 +1,5 @@
use clap::Parser; use clap::Parser;
#[cfg(target_os = "linux")]
use logger::Logger; use logger::Logger;
use mace::{Mace, Options}; use mace::{Mace, Options};
use rand::prelude::*; use rand::prelude::*;
@ -36,8 +37,11 @@ struct Args {
} }
fn main() { fn main() {
#[cfg(target_os = "linux")]
{
Logger::init().add_file("/tmp/x.log", true); Logger::init().add_file("/tmp/x.log", true);
log::set_max_level(log::LevelFilter::Debug); log::set_max_level(log::LevelFilter::Info);
}
let args = Args::parse(); let args = Args::parse();
let path = Path::new(&args.path); let path = Path::new(&args.path);