Add some benchmarks ⏱️

Cargo.toml

@@ -3,6 +3,7 @@ members = [
   "crates/ditto-cst",
   "crates/ditto-ast",
   "crates/ditto-checker",
+  "crates/ditto-benchmarks",
   "crates/ditto-config",
   "crates/ditto-codegen-js",
   "crates/ditto-fmt",

crates/ditto-benchmarks/Cargo.toml

@@ -0,0 +1,23 @@
+[package]
+name = "ditto-benchmarks"
+version = "0.0.1"
+edition = "2021"
+license = "BSD-3-Clause"
+
+[lib]
+doctest = false
+
+[dependencies]
+ditto-cst = { path = "../ditto-cst" }
+ditto-ast = { path = "../ditto-ast" }
+ditto-checker = { path = "../ditto-checker" }
+ditto-codegen-js = { path = "../ditto-codegen-js" }
+serde_json = "1.0"
+
+[dev-dependencies]
+criterion = "0.3"
+
+[[bench]]
+name = "vdom-html"
+# path = "benches/vdom-Html/criterion.rs"
+harness = false

crates/ditto-benchmarks/benches/common/mod.rs

@@ -0,0 +1,41 @@
+pub fn preamble(
+    source: &'static str,
+    module_exports: Vec<&'static str>,
+) -> (
+    ditto_cst::Module,
+    ditto_ast::Module,
+    ditto_checker::Everything,
+    ditto_codegen_js::Config,
+) {
+    let cst_module = ditto_cst::Module::parse(source).unwrap();
+    let mut everything = ditto_checker::Everything::default();
+    for module_exports_json in module_exports {
+        let (mn, exports): (ditto_ast::ModuleName, ditto_ast::ModuleExports) =
+            serde_json::from_str(module_exports_json).unwrap();
+        everything.modules.insert(mn, exports);
+    }
+    let (ast_module, warnings) =
+        ditto_checker::check_module(&everything, cst_module.clone()).unwrap();
+
+    assert!(warnings.is_empty());
+    return (cst_module, ast_module, everything, codegen_js_config());
+}
+
+pub fn codegen_js_config() -> ditto_codegen_js::Config {
+    ditto_codegen_js::Config {
+        module_name_to_path: Box::new(|(package_name, module_name)| {
+            let module_path = module_name
+                .0
+                .into_iter()
+                .map(|proper_name| proper_name.0)
+                .collect::<Vec<_>>()
+                .join(".");
+
+            match package_name {
+                None => module_path,
+                Some(ditto_ast::PackageName(pkg)) => format!("{}/{}", pkg, module_path),
+            }
+        }),
+        foreign_module_path: "./foreign.js".into(),
+    }
+}

crates/ditto-benchmarks/benches/vdom-html.rs

@@ -0,0 +1,37 @@
+mod common;
+
+use criterion::{black_box, criterion_group, criterion_main, BatchSize, Criterion};
+
+pub fn criterion_benchmark(c: &mut Criterion) {
+    let source = include_str!("vdom-html/source.ditto");
+    let (cst_module, ast_module, everything, codegen_js_config) = common::preamble(
+        source,
+        vec![include_str!("vdom-html/Attributes.ast-exports")],
+    );
+    let mut group = c.benchmark_group("vdom:Html");
+    group.bench_function("parse", |b| {
+        b.iter(|| ditto_cst::Module::parse(black_box(source)).unwrap())
+    });
+    group.bench_function("check", |b| {
+        b.iter_batched(
+            || cst_module.clone(),
+            |cst_module| ditto_checker::check_module(black_box(&everything), black_box(cst_module)),
+            BatchSize::LargeInput,
+        )
+    });
+    group.bench_function("codegen-js", |b| {
+        b.iter_batched(
+            || ast_module.clone(),
+            |ast_module| ditto_codegen_js::codegen(&codegen_js_config, ast_module),
+            BatchSize::LargeInput,
+        )
+    });
+    group.finish()
+}
+
+criterion_group! {
+    name = benches;
+    config = Criterion::default().sample_size(30);
+    targets = criterion_benchmark
+}
+criterion_main!(benches);