Move benchmarking to Criterion

Author: shepmaster

Cargo.toml

@@ -14,10 +14,19 @@ documentation = "https://docs.rs/sxd-document"
 
 license = "MIT"
 
+[[bench]]
+name = "string_pool"
+harness = false
+required-features = ["__internal_expose_string_pool"]
+
 [features]
 compile_failure = []
-unstable = []
+
+__internal_expose_string_pool = []
 
 [dependencies]
 peresil = "0.3.0"
 typed-arena = "2.0"
+
+[dev-dependencies]
+criterion = "0.5.1"

benches/string_pool.rs

@@ -0,0 +1,64 @@
+use criterion::{criterion_group, criterion_main, Criterion, Throughput};
+use sxd_document::__internal::StringPool;
+
+fn single_string(c: &mut Criterion) {
+    let mut group = c.benchmark_group("single_string");
+
+    let original = "hello";
+
+    group.throughput(Throughput::Bytes(original.len() as u64));
+    group.bench_function("single_string", |b| {
+        let pool = StringPool::new();
+        b.iter(|| pool.intern(original));
+    });
+
+    group.finish();
+}
+
+fn many_unique_string(c: &mut Criterion) {
+    let mut group = c.benchmark_group("many_unique_string");
+
+    let strings: Vec<_> = (0..1000).map(|i| format!("str{i}str")).collect();
+    let total_len = strings.iter().map(|s| s.len()).sum::<usize>();
+
+    group.throughput(Throughput::Bytes(total_len as u64));
+    group.bench_function("many_unique_string", |b| {
+        let pool = StringPool::new();
+
+        b.iter(|| {
+            for s in &strings {
+                pool.intern(s);
+            }
+        });
+    });
+
+    group.finish();
+}
+
+fn many_repeated_string(c: &mut Criterion) {
+    let mut group = c.benchmark_group("many_repeated_string");
+
+    let strings: Vec<_> = (0..1000).map(|i| format!("str{}str", i % 100)).collect();
+    let total_len = strings.iter().map(|s| s.len()).sum::<usize>();
+
+    group.throughput(Throughput::Bytes(total_len as u64));
+    group.bench_function("many_unique_string", |b| {
+        let pool = StringPool::new();
+
+        b.iter(|| {
+            for s in &strings {
+                pool.intern(s);
+            }
+        });
+    });
+
+    group.finish();
+}
+
+criterion_group!(
+    benches,
+    single_string,
+    many_unique_string,
+    many_repeated_string,
+);
+criterion_main!(benches);

src/lib.rs

@@ -49,7 +49,6 @@
 //! Try to leverage the type system as much as possible.
 
 #![deny(rust_2018_idioms)]
-#![cfg_attr(feature = "unstable", feature(test))]
 
 #[macro_use]
 extern crate peresil;
@@ -68,6 +67,11 @@ pub mod parser;
 pub mod thindom;
 pub mod writer;
 
+#[cfg(feature = "__internal_expose_string_pool")]
+pub mod __internal {
+    pub use super::string_pool::StringPool;
+}
+
 pub use crate::str::XmlChar;
 
 static XML_NS_PREFIX: &str = "xml";

src/string_pool.rs

@@ -275,45 +275,3 @@ mod test {
         assert_eq!(interned.as_ptr(), ptr_val);
     }
 }
-
-#[cfg(feature = "unstable")]
-mod bench {
-    extern crate test;
-
-    use self::test::Bencher;
-
-    use super::StringPool;
-
-    #[bench]
-    fn single_string(b: &mut Bencher) {
-        let s = StringPool::new();
-        b.iter(|| s.intern("hello"));
-        b.bytes = "hello".len() as u64;
-    }
-
-    #[bench]
-    fn many_unique_string(b: &mut Bencher) {
-        let s = StringPool::new();
-
-        let strings: Vec<String> = (0..1000).map(|i| format!("str{}str", i)).collect();
-        b.iter(|| {
-            for ss in strings.iter() {
-                s.intern(ss);
-            }
-        });
-        b.bytes = strings.iter().fold(0, |a, s| a + s.len()) as u64;
-    }
-
-    #[bench]
-    fn many_repeated_string(b: &mut Bencher) {
-        let s = StringPool::new();
-
-        let strings: Vec<String> = (0..1000).map(|i| format!("str{}str", i % 100)).collect();
-        b.iter(|| {
-            for ss in strings.iter() {
-                s.intern(ss);
-            }
-        });
-        b.bytes = strings.iter().fold(0, |a, s| a + s.len()) as u64;
-    }
-}