perf: Parameterize insertion benchmark by number of keys

This commit parameterizes the insertion benchmark to evaluate performance with varying complexities of JSON documents. The benchmark now runs for documents containing 1, 10, and 100 keys.

Author: michaelmior

benches/insertion.rs

@@ -6,34 +6,51 @@ use tempfile::tempdir;
 
 fn insertion_benchmark(c: &mut Criterion) {
     let mut group = c.benchmark_group("insertion");
+
     let num_docs = 10_000;
+
     group.throughput(Throughput::Elements(num_docs as u64));
 
-    group.bench_function(BenchmarkId::new("insert", num_docs), |b| {
-        b.iter_custom(|iters| {
-            let mut total_duration = std::time::Duration::new(0, 0);
-            for _ in 0..iters {
-                // Setup for each iteration: Create a new DB in a temp directory
-                let dir = tempdir().unwrap();
-                let mut db = DB::new(dir.path().to_str().unwrap(), num_docs + 1, 10);
-                db.create_collection("test").unwrap();
-
-                let start = std::time::Instant::now();
-
-                // Benchmarked routine: Insert num_docs
-                for i in 0..num_docs {
-                    db.insert("test", hint::black_box(json!({ "a": i })))
-                        .unwrap();
-                }
+    for num_keys in [1, 10, 100].iter() {
+        group.measurement_time(std::time::Duration::from_secs(*num_keys * 5));
+        group.bench_function(BenchmarkId::new("insert", *num_keys), |b| {
+            b.iter_custom(|iters| {
+                let mut total_duration = std::time::Duration::new(0, 0);
+
+                for _ in 0..iters {
+                    // Setup for each iteration: Create a new DB in a temp directory
+                    let dir = tempdir().unwrap();
+                    let mut db = DB::new(dir.path().to_str().unwrap(), num_docs + 1, 10);
+
+                    db.create_collection("test").unwrap();
+
+                    let start = std::time::Instant::now();
 
-                total_duration += start.elapsed();
+                    // Benchmarked routine: Insert num_docs
+                    for i in 0..num_docs {
+                        let doc = {
+                            let mut doc = serde_json::Map::new();
+
+                            for j in 0..*num_keys {
+                                doc.insert(format!("key{}", j), serde_json::Value::from(i));
+                            }
+
+                            serde_json::Value::Object(doc)
+                        };
+
+                        db.insert("test", hint::black_box(doc)).unwrap();
+                    }
+
+                    total_duration += start.elapsed();
+
+                    // Teardown for each iteration: `dir` (TempDir) is dropped here,
+                    // cleaning up the temporary database files.
+                }
 
-                // Teardown for each iteration: `dir` (TempDir) is dropped here,
-                // cleaning up the temporary database files.
-            }
-            total_duration
-        })
-    });
+                total_duration
+            })
+        });
+    }
 
     group.finish();
 }