From 47672de7f4fb917068923176658198d00d393ec8 Mon Sep 17 00:00:00 2001
From: Jan Tuomi <jan@jantuomi.fi>
Date: Sat, 5 Oct 2024 09:46:30 +0200
Subject: Add some initial benchmarks

---
 benches/utils.rs | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 benches/utils.rs

(limited to 'benches/utils.rs')

diff --git a/benches/utils.rs b/benches/utils.rs
new file mode 100644
index 0000000..9f20e3c
--- /dev/null
+++ b/benches/utils.rs
@@ -0,0 +1,46 @@
+use log_db::*;
+use rand::distributions::Alphanumeric;
+use rand::Rng;
+use std::fmt::Debug;
+use std::io;
+
+// Function to generate a random integer
+pub fn random_int() -> i64 {
+    let mut rng = rand::thread_rng();
+    rng.gen_range(0..1000) // Random number between 0 and 1000
+}
+
+// Function to generate a random string
+pub fn random_string(len: usize) -> String {
+    let mut rng = rand::thread_rng();
+    (0..len).map(|_| rng.sample(Alphanumeric) as char).collect()
+}
+
+// Function to generate random bytes
+pub fn random_bytes(len: usize) -> Vec<u8> {
+    let mut rng = rand::thread_rng();
+    (0..len).map(|_| rng.gen()).collect()
+}
+
+// Function to generate a random record
+pub fn random_record() -> Record {
+    Record {
+        values: vec![
+            RecordValue::Int(random_int()),        // Random int value
+            RecordValue::String(random_string(5)), // Random string of length 5
+            RecordValue::Bytes(random_bytes(10)),  // Random bytes of length 10
+        ],
+    }
+}
+
+pub fn prefill_db_with_n_records<T: Eq + Clone + Debug>(
+    db: &mut DB<T>,
+    n: usize,
+) -> Result<(), io::Error> {
+    for _ in 0..n {
+        let record = random_record();
+        db.upsert(&record)?;
+    }
+
+    Ok(())
+}
-- 
cgit v1.3