From 436d2032c4aeac23434732ad374c3a46883aa5cd Mon Sep 17 00:00:00 2001
From: Kerollmops <clement@meilisearch.com>
Date: Mon, 11 Apr 2022 15:40:07 -0700
Subject: [PATCH] Add benchmarks to the flatten-serde-json subcrate

---
 flatten-serde-json/Cargo.toml            |  9 +++--
 flatten-serde-json/benches/benchmarks.rs | 42 ++++++++++++++++++++++++
 2 files changed, 49 insertions(+), 2 deletions(-)
 create mode 100644 flatten-serde-json/benches/benchmarks.rs

diff --git a/flatten-serde-json/Cargo.toml b/flatten-serde-json/Cargo.toml
index 7c18656f1..0220e8ceb 100644
--- a/flatten-serde-json/Cargo.toml
+++ b/flatten-serde-json/Cargo.toml
@@ -5,7 +5,12 @@ edition = "2021"
 description = "Flatten serde-json objects like elastic search"
 readme = "README.md"
 
-# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
-
 [dependencies]
 serde_json = "1.0"
+
+[dev-dependencies]
+criterion = { version = "0.3", features = ["html_reports"] }
+
+[[bench]]
+name = "benchmarks"
+harness = false
diff --git a/flatten-serde-json/benches/benchmarks.rs b/flatten-serde-json/benches/benchmarks.rs
new file mode 100644
index 000000000..6536bb513
--- /dev/null
+++ b/flatten-serde-json/benches/benchmarks.rs
@@ -0,0 +1,42 @@
+use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
+use flatten_serde_json::flatten;
+use serde_json::json;
+
+pub fn flatten_simple(c: &mut Criterion) {
+    let mut input = json!({
+      "a": {
+        "b": "c",
+        "d": "e",
+        "f": "g"
+      }
+    });
+    let object = input.as_object_mut().unwrap();
+
+    c.bench_with_input(BenchmarkId::new("flatten", "simple"), &object, |b, input| {
+        b.iter(|| flatten(input))
+    });
+}
+
+pub fn flatten_complex(c: &mut Criterion) {
+    let mut input = json!({
+      "a": [
+        "b",
+        ["c", "d"],
+        { "e": ["f", "g"] },
+        [
+            { "h": "i" },
+            { "e": ["j", { "z": "y" }] },
+        ],
+        ["l"],
+        "m",
+      ]
+    });
+    let object = input.as_object_mut().unwrap();
+
+    c.bench_with_input(BenchmarkId::new("flatten", "complex"), &object, |b, input| {
+        b.iter(|| flatten(input))
+    });
+}
+
+criterion_group!(benches, flatten_simple, flatten_complex);
+criterion_main!(benches);