Add benchmark test for mbe

Author: edwin0cheng

Cargo.lock

@@ -19,3 +19,6 @@ parser = { path = "../parser", version = "0.0.0" }
 tt = { path = "../tt", version = "0.0.0" }
 test_utils = { path = "../test_utils", version = "0.0.0" }
 
+[dev-dependencies]
+profile = { path = "../profile", version = "0.0.0" }
+

crates/mbe/Cargo.toml

@@ -0,0 +1,93 @@
+use std::fs;
+
+use profile::StopWatch;
+use rustc_hash::FxHashMap;
+use syntax::{
+    ast::{self, NameOwner},
+    AstNode,
+};
+use test_utils::{project_dir, skip_slow_tests};
+
+use crate::{ast_to_token_tree, MacroRules};
+
+#[test]
+fn parse_a_lot_macro_rules() {
+    if skip_slow_tests() {
+        return;
+    }
+    let rules = macro_rules_fixtures_tt();
+    let mut time = StopWatch::start();
+    for (_, tt) in rules {
+        MacroRules::parse(&tt).unwrap();
+    }
+    println!("elapsed time : {}", time.elapsed());
+}
+
+#[test]
+fn expand_a_lot_macro_rules() {
+    if skip_slow_tests() {
+        return;
+    }
+    let rules = macro_rules_fixtures();
+    let invocations = invocation_fixtures();
+
+    let mut time = StopWatch::start();
+    for (idx, (id, tt)) in invocations.into_iter().enumerate() {
+        let res = rules[&id].expand(&tt);
+        if let Some(err) = res.err {
+            println!("Faile to expand [{}] {}: error = {}, content = {}", idx, id, err, tt);
+        }
+    }
+    println!("elapsed time : {}", time.elapsed());
+}
+
+fn macro_rules_fixtures() -> FxHashMap<String, MacroRules> {
+    macro_rules_fixtures_tt()
+        .into_iter()
+        .map(|(id, tt)| (id, MacroRules::parse(&tt).unwrap()))
+        .collect()
+}
+
+fn macro_rules_fixtures_tt() -> FxHashMap<String, tt::Subtree> {
+    let rules = macro_rules_fixtures_ast();
+    rules
+        .into_iter()
+        .map(|(id, rule)| {
+            let (def_tt, _) = ast_to_token_tree(&rule.token_tree().unwrap()).unwrap();
+            (id, def_tt)
+        })
+        .collect()
+}
+
+fn macro_rules_fixtures_ast() -> FxHashMap<String, ast::MacroRules> {
+    let file = project_dir().join("crates/mbe/test_data/fixture-rules.rs");
+    let big_file = fs::read_to_string(file).unwrap();
+    let source_file = ast::SourceFile::parse(&big_file).ok().unwrap();
+
+    source_file
+        .syntax()
+        .descendants()
+        .filter_map(ast::MacroRules::cast)
+        .map(|it| (it.name().unwrap().to_string(), it))
+        .collect()
+}
+
+fn invocation_fixtures() -> FxHashMap<String, tt::Subtree> {
+    let file = project_dir().join("crates/mbe/test_data/fixture-invocations.rs");
+    let big_file = fs::read_to_string(file).unwrap();
+    // Replace $crate to crate, since $crate may transform to crate name in realilty
+    let big_file = big_file.replace("$crate", "crate");
+
+    let source_file = ast::SourceFile::parse(&big_file).ok().unwrap();
+
+    source_file
+        .syntax()
+        .descendants()
+        .filter_map(ast::MacroCall::cast)
+        .map(|it| {
+            let id = it.path().unwrap().to_string();
+            let (invocation_tt, _) = ast_to_token_tree(&it.token_tree().unwrap()).unwrap();
+            (id, invocation_tt)
+        })
+        .collect()
+}

crates/mbe/src/benchmark.rs

@@ -12,6 +12,9 @@ mod subtree_source;
 #[cfg(test)]
 mod tests;
 
+#[cfg(test)]
+mod benchmark;
+
 use std::fmt;
 
 use test_utils::mark;