chore: Add profiler

Author: Stranger6667

.github/workflows/build.yml

@@ -73,6 +73,9 @@ jobs:
       - run: cargo fmt --all -- --check
         working-directory: ./css-inline
 
+      - run: cargo fmt --all -- --check
+        working-directory: ./profiler
+
       - run: cargo fmt --all -- --check
         working-directory: ./bindings/c
 
@@ -108,6 +111,10 @@ jobs:
         run: cargo clippy -- -D warnings
         working-directory: ./css-inline
 
+      - name: Profiler
+        run: cargo clippy -- -D warnings
+        working-directory: ./profiler
+
       - name: Python
         run: cargo clippy -- -D warnings
         working-directory: ./bindings/python

.gitignore

@@ -4,8 +4,14 @@
 css-inline/target/
 css-inline/fuzz/target/
 css-inline/fuzz/corpus/
+profiler/target/
 bindings/*/target/
 
+perf.data
+perf.data.old
+trace.svg
+dhat-heap.json
+
 # Remove Cargo.lock from gitignore if creating an executable, leave it for libraries
 # More information here https://doc.rust-lang.org/cargo/guide/cargo-toml-vs-cargo-lock.html
 Cargo.lock

profiler/Cargo.toml

@@ -0,0 +1,23 @@
+[package]
+name = "profiler"
+version = "0.1.0"
+authors = ["Dmitry Dygalo <dmitry@dygalo.dev>"]
+edition = "2021"
+
+[dependencies]
+pico-args = "0.5"
+serde = { version = "1", features = ["derive"] }
+serde_json = "1"
+dhat = "*"
+
+[dependencies.css-inline]
+path = "../css-inline"
+version = "*"
+default-features = false
+features = ["http", "file"]
+
+[features]
+dhat-heap = []
+
+[profile.release]
+debug = true

profiler/src/main.rs

@@ -0,0 +1,46 @@
+use std::fs;
+
+#[cfg(feature = "dhat-heap")]
+#[global_allocator]
+static ALLOC: dhat::Alloc = dhat::Alloc;
+
+struct Args {
+    iterations: usize,
+    benchmark: String,
+}
+
+#[derive(serde::Deserialize, Debug)]
+struct Benchmark {
+    name: String,
+    html: String,
+}
+
+fn main() -> Result<(), Box<dyn std::error::Error>> {
+    let mut args = pico_args::Arguments::from_env();
+    let args = Args {
+        iterations: args.value_from_str("--iterations")?,
+        benchmark: args.value_from_str("--benchmark")?,
+    };
+
+    let benchmarks_str =
+        fs::read_to_string("../benchmarks/benchmarks.json").expect("Failed to load benchmarks");
+    let benchmarks: Vec<Benchmark> =
+        serde_json::from_str(&benchmarks_str).expect("Failed to load benchmarks");
+    if let Some(benchmark) = benchmarks.iter().find(|x| x.name == args.benchmark) {
+        let mut output = Vec::with_capacity(
+            (benchmark.html.len() as f64 * 1.5)
+                .min(usize::MAX as f64)
+                .round() as usize,
+        );
+        for _ in 0..args.iterations {
+            #[cfg(feature = "dhat-heap")]
+            let _profiler = dhat::Profiler::new_heap();
+            css_inline::inline_to(&benchmark.html, &mut output).expect("Inlining failed");
+            output.clear();
+        }
+    } else {
+        panic!("Can not find benchmark: {}", &args.benchmark)
+    }
+
+    Ok(())
+}