From f69622209297111bedb6633af3d973164a4f6d93 Mon Sep 17 00:00:00 2001
From: Jan Tuomi <jan@jantuomi.fi>
Date: Sat, 5 Oct 2024 18:18:55 +0200
Subject: Improve benchmarks, apply measurements to code

---
 benches/benchmark.rs | 55 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)

(limited to 'benches')

diff --git a/benches/benchmark.rs b/benches/benchmark.rs
index c999811..43d53f7 100644
--- a/benches/benchmark.rs
+++ b/benches/benchmark.rs
@@ -2,6 +2,8 @@ mod utils;
 
 use criterion::{black_box, criterion_group, criterion_main, BenchmarkId, Criterion};
 use log_db::*;
+use std::fs::OpenOptions;
+use std::path::Path;
 use tempfile;
 use utils::*;
 
@@ -156,6 +158,58 @@ pub fn get_various_memtable_capacities(c: &mut Criterion) {
     }
 }
 
+fn reverse_read_file_with_various_buffer_sizes(c: &mut Criterion) {
+    let mut group = c.benchmark_group("reverse_read_file_with_various_buffer_sizes");
+    group.sample_size(50);
+
+    // odd powers of 2
+    let buffer_sizes = [128, 512, 2048, 8192, 32768, 131_072, 524_288];
+    const PREFILL_N: usize = 100_000;
+
+    let data_dir_obj = tempfile::tempdir().expect("Failed to get tmpdir");
+    let data_dir = &data_dir_obj
+        .path()
+        .to_str()
+        .expect("Failed to convert tmpdir path to str");
+
+    // Create a db instance for prefilling
+    let mut db = DB::configure()
+        .data_dir(&data_dir)
+        .fields(&vec![
+            (Field::Id, RecordFieldType::Int),
+            (Field::Name, RecordFieldType::String),
+            (Field::Data, RecordFieldType::Bytes),
+        ])
+        .primary_key(Field::Id)
+        .initialize()
+        .expect("Failed to initialize DB");
+
+    prefill_db(&mut db, PREFILL_N).expect("Failed to prefill DB");
+    drop(db);
+
+    for size in buffer_sizes {
+        group.bench_with_input(BenchmarkId::from_parameter(size), &size, |b, &_size| {
+            let mut file = OpenOptions::new()
+                .read(true)
+                .open(Path::new(data_dir).join("db"))
+                .expect("Failed to open log file");
+
+            b.iter(|| {
+                let mut rev_reader = ReverseLogReader::new_with_size(&mut file, size)
+                    .expect("Failed to create ReverseLogReader");
+
+                // This is to avoid optimizing out the loop
+                let mut i = 0;
+                for _ in &mut rev_reader {
+                    i += 1;
+                }
+
+                i
+            });
+        });
+    }
+}
+
 // Register the benchmark group
 criterion_group!(
     benches,
@@ -163,5 +217,6 @@ criterion_group!(
     upsert_write_durability,
     get_from_disk_various_initial_sizes,
     get_various_memtable_capacities,
+    reverse_read_file_with_various_buffer_sizes,
 );
 criterion_main!(benches);
-- 
cgit v1.3