#[macro_use] extern crate criterion; #[macro_use] extern crate serde_derive; use criterion::Criterion; use handlebars::{to_json, Context, Handlebars, Template}; use serde_json::json; use serde_json::value::Value as Json; use std::collections::BTreeMap; #[cfg(unix)] use criterion::profiler::Profiler; #[cfg(unix)] use pprof::protos::Message; #[cfg(unix)] use pprof::ProfilerGuard; #[cfg(unix)] use std::fs::{create_dir_all, File}; #[cfg(unix)] use std::io::Write; #[cfg(unix)] use std::path::Path; #[cfg(unix)] #[derive(Default)] struct CpuProfiler<'a> { guard: Option>, } #[cfg(unix)] impl Profiler for CpuProfiler<'_> { fn start_profiling(&mut self, _benchmark_id: &str, benchmark_dir: &Path) { create_dir_all(benchmark_dir).unwrap(); let guard = ProfilerGuard::new(100).unwrap(); self.guard = Some(guard); } fn stop_profiling(&mut self, benchmark_id: &str, benchmark_dir: &Path) { if let Ok(ref report) = self.guard.as_ref().unwrap().report().build() { let fg_file_name = benchmark_dir.join(format!("{benchmark_id}.svg")); let fg_file = File::create(fg_file_name).unwrap(); report.flamegraph(fg_file).unwrap(); let pb_file_name = benchmark_dir.join(format!("{benchmark_id}.pb")); let mut pb_file = File::create(pb_file_name).unwrap(); let profile = report.pprof().unwrap(); let mut content = Vec::new(); profile.encode(&mut content).unwrap(); pb_file.write_all(&content).unwrap(); }; self.guard = None; } } #[cfg(unix)] fn profiled() -> Criterion { Criterion::default().with_profiler(CpuProfiler::default()) } #[derive(Serialize)] struct DataWrapper { v: String, } #[derive(Serialize)] struct RowWrapper { real: Vec, dummy: Vec, } #[derive(Serialize)] struct NestedRowWrapper { parent: Vec>, } static SOURCE: &str = " {{year}}

CSL {{year}}

    {{#each teams}}
  • {{name}}: {{score}}
    • {{/each}}
"; fn make_data() -> BTreeMap { let mut data = BTreeMap::new(); data.insert("year".to_string(), to_json("2015")); let mut teams = Vec::new(); for v in [ ("Jiangsu", 43u16), ("Beijing", 27u16), ("Guangzhou", 22u16), ("Shandong", 12u16), ] .iter() { let (name, score) = *v; let mut t = BTreeMap::new(); t.insert("name".to_string(), to_json(name)); t.insert("score".to_string(), to_json(score)); teams.push(t); } data.insert("teams".to_string(), to_json(&teams)); data } fn parse_template(c: &mut Criterion) { c.bench_function("parse_template", move |b| { b.iter(|| Template::compile(SOURCE).ok().unwrap()); }); } fn render_template(c: &mut Criterion) { let mut handlebars = Handlebars::new(); handlebars .register_template_string("table", SOURCE) .expect("Invalid template format"); let ctx = Context::wraps(make_data()).unwrap(); c.bench_function("render_template", move |b| { b.iter(|| handlebars.render_with_context("table", &ctx).ok().unwrap()); }); } fn large_loop_helper(c: &mut Criterion) { let mut handlebars = Handlebars::new(); handlebars .register_template_string("test", "BEFORE\n{{#each real}}{{this.v}}{{/each}}AFTER") .expect("Invalid template format"); let real: Vec = (1..1000) .map(|i| DataWrapper { v: format!("n={i}"), }) .collect(); let dummy: Vec = (1..1000) .map(|i| DataWrapper { v: format!("n={i}"), }) .collect(); let rows = RowWrapper { real, dummy }; let ctx = Context::wraps(rows).unwrap(); c.bench_function("large_loop_helper", move |b| { b.iter(|| handlebars.render_with_context("test", &ctx).ok().unwrap()); }); } fn large_loop_helper_with_context_creation(c: &mut Criterion) { let mut handlebars = Handlebars::new(); handlebars .register_template_string("test", "BEFORE\n{{#each real}}{{this.v}}{{/each}}AFTER") .expect("Invalid template format"); let real: Vec = (1..1000) .map(|i| DataWrapper { v: format!("n={i}"), }) .collect(); let dummy: Vec = (1..1000) .map(|i| DataWrapper { v: format!("n={i}"), }) .collect(); let rows = RowWrapper { real, dummy }; c.bench_function("large_loop_helper_with_context_creation", move |b| { b.iter(|| handlebars.render("test", &rows).ok().unwrap()); }); } fn large_nested_loop(c: &mut Criterion) { let mut handlebars = Handlebars::new(); handlebars .register_template_string( "test", "BEFORE\n{{#each parent as |child|}}{{#each child}}{{this.v}}{{/each}}{{/each}}AFTER", ) .expect("Invalid template format"); let parent: Vec> = (1..100) .map(|_| { (1..10) .map(|v| DataWrapper { v: format!("v={v}"), }) .collect() }) .collect(); let rows = NestedRowWrapper { parent }; let ctx = Context::wraps(rows).unwrap(); c.bench_function("large_nested_loop", move |b| { b.iter(|| handlebars.render_with_context("test", &ctx).ok().unwrap()); }); } fn deeply_nested_partial(c: &mut Criterion) { use std::iter::repeat; let mut handlebars = Handlebars::new(); handlebars .register_partial( "nested_partial", r#"{{#each baz}}
{{this}}{{#if (not @last)}}++{{/if}}
{{/each}}"#, ) .expect("Invalid template format"); handlebars .register_partial( "partial", r#"
{{#each bar}} {{>nested_partial}} {{/each}}
"#, ) .expect("Invalid template format"); handlebars .register_template_string( "test", r#"
{{#each foo}} {{>partial}} {{/each}}
"#, ) .expect("Invalid template format"); let data = json!({ "foo": repeat(json!({ "bar": repeat(json!({ "baz": repeat("xyz").take(7).collect::>() })).take(7).collect::>() })).take(7).collect::>() }); let ctx = Context::wraps(data).unwrap(); c.bench_function("deeply_nested_partial", move |b| { b.iter(|| handlebars.render_with_context("test", &ctx).ok().unwrap()); }); } #[cfg(unix)] criterion_group!( name = benches; config = profiled(); targets = parse_template, render_template, large_loop_helper, large_loop_helper_with_context_creation, large_nested_loop, deeply_nested_partial ); #[cfg(not(unix))] criterion_group!( benches, parse_template, render_template, large_loop_helper, large_loop_helper_with_context_creation, large_nested_loop, deeply_nested_partial ); criterion_main!(benches);