Commit af7d782a authored by Tyler Neely's avatar Tyler Neely

Clean up after benchmarking tests. Enforce ordering on benchmarks.

parent 14dc3d00
...@@ -91,9 +91,7 @@ mod tests { ...@@ -91,9 +91,7 @@ mod tests {
use rocksdb::{RocksDBOptions, RocksDB, MergeOperands, new_bloom_filter, Writable}; use rocksdb::{RocksDBOptions, RocksDB, MergeOperands, new_bloom_filter, Writable};
use rocksdb::RocksDBCompactionStyle::RocksDBUniversalCompaction; use rocksdb::RocksDBCompactionStyle::RocksDBUniversalCompaction;
fn tuned_for_somebody_elses_disk() -> RocksDB { fn tuned_for_somebody_elses_disk(path: &str, opts: RocksDBOptions) -> RocksDB {
let path = "_rust_rocksdb_optimizetest";
let opts = RocksDBOptions::new();
opts.create_if_missing(true); opts.create_if_missing(true);
opts.set_block_size(524288); opts.set_block_size(524288);
opts.set_max_open_files(10000); opts.set_max_open_files(10000);
...@@ -121,10 +119,12 @@ mod tests { ...@@ -121,10 +119,12 @@ mod tests {
} }
#[bench] #[bench]
fn writes(b: &mut Bencher) { fn a_writes(b: &mut Bencher) {
// dirty hack due to parallel tests causing contention. // dirty hack due to parallel tests causing contention.
sleep_ms(1000); sleep_ms(1000);
let db = tuned_for_somebody_elses_disk(); let path = "_rust_rocksdb_optimizetest";
let opts = RocksDBOptions::new();
let db = tuned_for_somebody_elses_disk(path, opts);
let mut i = 0 as u64; let mut i = 0 as u64;
b.iter(|| { b.iter(|| {
db.put(i.to_string().as_bytes(), b"v1111"); db.put(i.to_string().as_bytes(), b"v1111");
...@@ -134,8 +134,10 @@ mod tests { ...@@ -134,8 +134,10 @@ mod tests {
} }
#[bench] #[bench]
fn reads(b: &mut Bencher) { fn b_reads(b: &mut Bencher) {
let db = tuned_for_somebody_elses_disk(); let path = "_rust_rocksdb_optimizetest";
let opts = RocksDBOptions::new();
let db = tuned_for_somebody_elses_disk(path, opts);
let mut i = 0 as u64; let mut i = 0 as u64;
b.iter(|| { b.iter(|| {
db.get(i.to_string().as_bytes()).on_error( |e| { db.get(i.to_string().as_bytes()).on_error( |e| {
...@@ -145,5 +147,6 @@ mod tests { ...@@ -145,5 +147,6 @@ mod tests {
i += 1; i += 1;
}); });
db.close(); db.close();
RocksDB::destroy(opts, path).is_ok();
} }
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment