benchmarks: add 'stringzilla'

See: https://github.com/ashvardanian/StringZilla

.vim/coc-settings.json

@@ -8,6 +8,7 @@
     "benchmarks/engines/rust-memchrold/Cargo.toml",
     "benchmarks/engines/rust-sliceslice/Cargo.toml",
     "benchmarks/engines/rust-std/Cargo.toml",
+    "benchmarks/engines/stringzilla/Cargo.toml",
     "benchmarks/shared/Cargo.toml",
     "fuzz/Cargo.toml",
     "Cargo.toml"

benchmarks/engines.toml

@@ -158,6 +158,26 @@
     bin = "cargo"
     args = ["clean"]
 
+# Engines based on the `stringzilla` library.
+#
+# See: https://github.com/ashvardanian/StringZilla
+
+[[engine]]
+  name = "stringzilla/memmem/oneshot"
+  cwd = "./engines/stringzilla"
+  [engine.version]
+    bin = "./target/release/main"
+    args = ["--version"]
+  [engine.run]
+    bin = "./target/release/main"
+    args = ["memmem-oneshot"]
+  [[engine.build]]
+    bin = "cargo"
+    args = ["build", "--release"]
+  [[engine.clean]]
+    bin = "cargo"
+    args = ["clean"]
+
 # Engines based on the Rust standard library.
 
 [[engine]]

benchmarks/engines/stringzilla/Cargo.toml

@@ -0,0 +1,23 @@
+[package]
+publish = false
+name = "main"
+version = "3.3.0"  # should match 'stringzilla' version below
+edition = "2021"
+
+[workspace]
+
+[dependencies]
+anyhow = "1.0.72"
+stringzilla = "3.3.0"
+
+[dependencies.shared]
+path = "../../shared"
+
+[[bin]]
+name = "main"
+path = "main.rs"
+
+[profile.release]
+debug = true
+codegen-units = 1
+lto = "fat"

benchmarks/engines/stringzilla/main.rs

@@ -0,0 +1,39 @@
+use std::io::Write;
+
+use shared::{Benchmark, Sample};
+
+use stringzilla::StringZilla;
+
+fn main() -> anyhow::Result<()> {
+    let Some(arg) = std::env::args_os().nth(1) else {
+        anyhow::bail!("Usage: runner (<engine-name> | --version)")
+    };
+    let Ok(arg) = arg.into_string() else {
+        anyhow::bail!("argument given is not valid UTF-8")
+    };
+    if arg == "--version" {
+        writeln!(std::io::stdout(), env!("CARGO_PKG_VERSION"))?;
+        return Ok(());
+    }
+    let engine = arg;
+    let b = Benchmark::from_stdin()?;
+    let samples = match (&*engine, &*b.model) {
+        ("memmem-oneshot", "count") => memmem_oneshot_count(&b)?,
+        (engine, model) => {
+            anyhow::bail!("unrecognized engine '{engine}' and model '{model}'")
+        }
+    };
+    let mut stdout = std::io::stdout().lock();
+    for s in samples.iter() {
+        writeln!(stdout, "{},{}", s.duration.as_nanos(), s.count)?;
+    }
+    Ok(())
+}
+
+fn memmem_oneshot_count(b: &Benchmark) -> anyhow::Result<Vec<Sample>> {
+    let haystack = &b.haystack;
+    let needle = b.one_needle()?;
+    shared::run(b, || {
+        Ok(shared::count_memmem(haystack, needle, |h, n| h.sz_find(n)))
+    })
+}