From 49e4cc3daf85bd4a86325988e23f68de2f64b700 Mon Sep 17 00:00:00 2001
From: tamo <tamo@meilisearch.com>
Date: Tue, 6 Apr 2021 19:17:24 +0200
Subject: [PATCH] add the words criterion to the bench

---
 milli/Cargo.toml       |  4 ++++
 milli/benches/words.rs | 35 +++++++++++++++++++++++++++++++++++
 2 files changed, 39 insertions(+)
 create mode 100644 milli/benches/words.rs

diff --git a/milli/Cargo.toml b/milli/Cargo.toml
index 175c15679..5184d028b 100644
--- a/milli/Cargo.toml
+++ b/milli/Cargo.toml
@@ -63,3 +63,7 @@ default = []
 [[bench]]
 name = "typo"
 harness = false
+
+[[bench]]
+name = "words"
+harness = false
diff --git a/milli/benches/words.rs b/milli/benches/words.rs
new file mode 100644
index 000000000..92ca0a784
--- /dev/null
+++ b/milli/benches/words.rs
@@ -0,0 +1,35 @@
+mod utils;
+
+use std::time::Duration;
+use criterion::{criterion_group, criterion_main, BenchmarkId};
+
+fn bench_words(c: &mut criterion::Criterion) {
+    let index = utils::base_setup(Some(vec!["words".to_string()]));
+
+    let queries = [
+        "the black saint and the sinner lady and the good doggo ", // four words to pop
+        "les liaisons dangeureuses 1793 ", // one word to pop
+        "The Disneyland Children's Sing-Alone song ", // two words to pop
+        "seven nation mummy ", // one word to pop
+        "7000 Danses / Le Baiser / je me trompe de mots ", // four words to pop
+        "Bring Your Daughter To The Slaughter but now this is not part of the title ", // nine words to pop
+        "whathavenotnsuchforth and then a good amount of words tot pop in order to match the first one ", // 16
+    ];
+
+    let mut group = c.benchmark_group("words");
+    group.measurement_time(Duration::from_secs(10));
+
+    for query in &queries {
+        group.bench_with_input(BenchmarkId::from_parameter(query), &query, |b, &query| {
+            b.iter(|| {
+                let rtxn = index.read_txn().unwrap();
+                let _documents_ids = index.search(&rtxn).query(*query).execute().unwrap();
+            });
+        });
+    }
+
+    group.finish();
+}
+
+criterion_group!(benches, bench_words);
+criterion_main!(benches);