From 8c4ff3e399e7dc40cf8c6086ef74074d438e7034 Mon Sep 17 00:00:00 2001
From: David Holroyd <dave@badgers-in-foil.co.uk>
Date: Tue, 14 Apr 2020 22:23:03 +0100
Subject: [PATCH] Add a parsing benchmark

---
 Cargo.toml       |  8 ++++++++
 benches/parse.rs | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 56 insertions(+)
 create mode 100644 benches/parse.rs

diff --git a/Cargo.toml b/Cargo.toml
index e1faeb1..88cc95c 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -32,3 +32,11 @@ strum = { version = "0.17", features = ["derive"] }
 [dev-dependencies]
 pretty_assertions = "0.6"
 version-sync = "0.9"
+criterion = "0.3.1"
+
+[[bench]]
+name = "parse"
+harness = false
+
+[profile.bench]
+debug = true
diff --git a/benches/parse.rs b/benches/parse.rs
new file mode 100644
index 0000000..f07dd0b
--- /dev/null
+++ b/benches/parse.rs
@@ -0,0 +1,48 @@
+use criterion::{criterion_group, criterion_main, Criterion, Throughput};
+use hls_m3u8;
+use hls_m3u8::tags::{ExtXDateRange, ExtXProgramDateTime};
+use hls_m3u8::types::Value;
+use hls_m3u8::{MediaPlaylist, MediaSegment};
+use std::str::FromStr;
+
+fn create_manifest_data() -> Vec<u8> {
+    let mut builder = MediaPlaylist::builder();
+    builder.media_sequence(826176645);
+    builder.has_independent_segments(true);
+    builder.target_duration(std::time::Duration::from_secs(2));
+    for i in 0..4000 {
+        let mut seg = MediaSegment::builder();
+        seg.duration(std::time::Duration::from_secs_f64(1.92))
+            .uri(format!(
+                "avc_unencrypted_global-video=3000000-{}.ts?variant=italy",
+                826176659 + i
+            ));
+        if i == 0 {
+            seg.program_date_time(ExtXProgramDateTime::new("2020-04-07T11:32:38Z"));
+        }
+        if i % 100 == 0 {
+            let mut date_range =
+                ExtXDateRange::new(format!("{}", i / 100), "2020-04-07T11:40:02.040000Z");
+            date_range.duration = Some(std::time::Duration::from_secs_f64(65.2));
+            date_range.client_attributes.insert("SCTE35-OUT".to_string(), Value::Hex(hex::decode("FC302500000000000000FFF0140500001C207FEFFE0030E3A0FE005989E000010000000070BA5ABF").unwrap()));
+            seg.date_range(date_range);
+        }
+        builder.push_segment(seg.build().unwrap());
+    }
+    builder.build().unwrap().to_string().into_bytes()
+}
+
+fn criterion_benchmark(c: &mut Criterion) {
+    let buf = create_manifest_data();
+    let mut group = c.benchmark_group("parser");
+    group.throughput(Throughput::Bytes(buf.len() as u64));
+    group.bench_function("throughput", |b| {
+        b.iter(|| {
+            let buf = String::from_utf8_lossy(&buf);
+            hls_m3u8::MediaPlaylist::from_str(&buf).unwrap()
+        });
+    });
+}
+
+criterion_group!(benches, criterion_benchmark);
+criterion_main!(benches);