2 files changed, 247 insertions, 0 deletions
diff --git a/third_party/rust/fluent-bundle/benches/resolver.rs b/third_party/rust/fluent-bundle/benches/resolver.rs
new file mode 100644
index 0000000000..2116d1b91e
--- /dev/null
+++ b/third_party/rust/fluent-bundle/benches/resolver.rs
@@ -0,0 +1,168 @@
+use criterion::criterion_group;
+use criterion::criterion_main;
+use criterion::BenchmarkId;
+use criterion::Criterion;
+use std::collections::HashMap;
+use std::fs::File;
+use std::io;
+use std::io::Read;
+use std::rc::Rc;
+
+use fluent_bundle::{FluentArgs, FluentBundle, FluentResource, FluentValue};
+use fluent_syntax::ast;
+use unic_langid::langid;
+
+fn read_file(path: &str) -> Result<String, io::Error> {
+    let mut f = File::open(path)?;
+    let mut s = String::new();
+    f.read_to_string(&mut s)?;
+    Ok(s)
+}
+
+fn get_strings(tests: &[&'static str]) -> HashMap<&'static str, String> {
+    let mut ftl_strings = HashMap::new();
+    for test in tests {
+        let path = format!("./benches/{}.ftl", test);
+        ftl_strings.insert(*test, read_file(&path).expect("Couldn't load file"));
+    }
+    return ftl_strings;
+}
+
+fn get_ids(res: &FluentResource) -> Vec<String> {
+    res.entries()
+        .filter_map(|entry| match entry {
+            ast::Entry::Message(ast::Message { id, .. }) => Some(id.name.to_owned()),
+            _ => None,
+        })
+        .collect()
+}
+
+fn get_args(name: &str) -> Option<FluentArgs> {
+    match name {
+        "preferences" => {
+            let mut prefs_args = FluentArgs::new();
+            prefs_args.set("name", FluentValue::from("John"));
+            prefs_args.set("tabCount", FluentValue::from(5));
+            prefs_args.set("count", FluentValue::from(3));
+            prefs_args.set("version", FluentValue::from("65.0"));
+            prefs_args.set("path", FluentValue::from("/tmp"));
+            prefs_args.set("num", FluentValue::from(4));
+            prefs_args.set("email", FluentValue::from("john@doe.com"));
+            prefs_args.set("value", FluentValue::from(4.5));
+            prefs_args.set("unit", FluentValue::from("mb"));
+            prefs_args.set("service-name", FluentValue::from("Mozilla Disk"));
+            Some(prefs_args)
+        }
+        _ => None,
+    }
+}
+
+fn add_functions<R>(name: &'static str, bundle: &mut FluentBundle<R>) {
+    match name {
+        "preferences" => {
+            bundle
+                .add_function("PLATFORM", |_args, _named_args| {
+                    return "linux".into();
+                })
+                .expect("Failed to add a function to the bundle.");
+        }
+        _ => {}
+    }
+}
+
+fn get_bundle(name: &'static str, source: &str) -> (FluentBundle<FluentResource>, Vec<String>) {
+    let res = FluentResource::try_new(source.to_owned()).expect("Couldn't parse an FTL source");
+    let ids = get_ids(&res);
+    let lids = vec![langid!("en")];
+    let mut bundle = FluentBundle::new(lids);
+    bundle
+        .add_resource(res)
+        .expect("Couldn't add FluentResource to the FluentBundle");
+    add_functions(name, &mut bundle);
+    (bundle, ids)
+}
+
+fn resolver_bench(c: &mut Criterion) {
+    let tests = &[
+        #[cfg(feature = "all-benchmarks")]
+        "simple",
+        "preferences",
+        #[cfg(feature = "all-benchmarks")]
+        "menubar",
+        #[cfg(feature = "all-benchmarks")]
+        "unescape",
+    ];
+    let ftl_strings = get_strings(tests);
+
+    let mut group = c.benchmark_group("construct");
+    for name in tests {
+        let source = ftl_strings.get(name).expect("Failed to find the source.");
+        group.bench_with_input(BenchmarkId::from_parameter(name), &source, |b, source| {
+            let res = Rc::new(
+                FluentResource::try_new(source.to_string()).expect("Couldn't parse an FTL source"),
+            );
+            b.iter(|| {
+                let lids = vec![langid!("en")];
+                let mut bundle = FluentBundle::new(lids);
+                bundle
+                    .add_resource(res.clone())
+                    .expect("Couldn't add FluentResource to the FluentBundle");
+                add_functions(name, &mut bundle);
+            })
+        });
+    }
+    group.finish();
+
+    let mut group = c.benchmark_group("resolve");
+    for name in tests {
+        let source = ftl_strings.get(name).expect("Failed to find the source.");
+        group.bench_with_input(BenchmarkId::from_parameter(name), &source, |b, source| {
+            let (bundle, ids) = get_bundle(name, source);
+            let args = get_args(name);
+            b.iter(|| {
+                let mut s = String::new();
+                for id in &ids {
+                    let msg = bundle.get_message(id).expect("Message found");
+                    let mut errors = vec![];
+                    if let Some(value) = msg.value() {
+                        let _ = bundle.write_pattern(&mut s, value, args.as_ref(), &mut errors);
+                        s.clear();
+                    }
+                    for attr in msg.attributes() {
+                        let _ =
+                            bundle.write_pattern(&mut s, attr.value(), args.as_ref(), &mut errors);
+                        s.clear();
+                    }
+                    assert!(errors.len() == 0, "Resolver errors: {:#?}", errors);
+                }
+            })
+        });
+    }
+    group.finish();
+
+    let mut group = c.benchmark_group("resolve_to_str");
+    for name in tests {
+        let source = ftl_strings.get(name).expect("Failed to find the source.");
+        group.bench_with_input(BenchmarkId::from_parameter(name), &source, |b, source| {
+            let (bundle, ids) = get_bundle(name, source);
+            let args = get_args(name);
+            b.iter(|| {
+                for id in &ids {
+                    let msg = bundle.get_message(id).expect("Message found");
+                    let mut errors = vec![];
+                    if let Some(value) = msg.value() {
+                        let _ = bundle.format_pattern(value, args.as_ref(), &mut errors);
+                    }
+                    for attr in msg.attributes() {
+                        let _ = bundle.format_pattern(attr.value(), args.as_ref(), &mut errors);
+                    }
+                    assert!(errors.len() == 0, "Resolver errors: {:#?}", errors);
+                }
+            })
+        });
+    }
+    group.finish();
+}
+
+criterion_group!(benches, resolver_bench);
+criterion_main!(benches);
diff --git a/third_party/rust/fluent-bundle/benches/resolver_iai.rs b/third_party/rust/fluent-bundle/benches/resolver_iai.rs
new file mode 100644
index 0000000000..10212f6f39
--- /dev/null
+++ b/third_party/rust/fluent-bundle/benches/resolver_iai.rs
@@ -0,0 +1,79 @@
+use fluent_bundle::{FluentArgs, FluentBundle, FluentResource, FluentValue};
+use fluent_syntax::ast;
+use unic_langid::{langid, LanguageIdentifier};
+
+const LANG_EN: LanguageIdentifier = langid!("en");
+
+fn add_functions<R>(name: &'static str, bundle: &mut FluentBundle<R>) {
+    match name {
+        "preferences" => {
+            bundle
+                .add_function("PLATFORM", |_args, _named_args| {
+                    return "linux".into();
+                })
+                .expect("Failed to add a function to the bundle.");
+        }
+        _ => {}
+    }
+}
+
+fn get_args(name: &str) -> Option<FluentArgs> {
+    match name {
+        "preferences" => {
+            let mut prefs_args = FluentArgs::new();
+            prefs_args.set("name", FluentValue::from("John"));
+            prefs_args.set("tabCount", FluentValue::from(5));
+            prefs_args.set("count", FluentValue::from(3));
+            prefs_args.set("version", FluentValue::from("65.0"));
+            prefs_args.set("path", FluentValue::from("/tmp"));
+            prefs_args.set("num", FluentValue::from(4));
+            prefs_args.set("email", FluentValue::from("john@doe.com"));
+            prefs_args.set("value", FluentValue::from(4.5));
+            prefs_args.set("unit", FluentValue::from("mb"));
+            prefs_args.set("service-name", FluentValue::from("Mozilla Disk"));
+            Some(prefs_args)
+        }
+        _ => None,
+    }
+}
+
+fn get_ids(res: &FluentResource) -> Vec<String> {
+    res.entries()
+        .filter_map(|entry| match entry {
+            ast::Entry::Message(ast::Message { id, .. }) => Some(id.name.to_owned()),
+            _ => None,
+        })
+        .collect()
+}
+
+fn iai_resolve_preferences() {
+    let files = &[include_str!("preferences.ftl")];
+    for source in files {
+        let res = FluentResource::try_new(source.to_string()).expect("failed to parse FTL.");
+        let ids = get_ids(&res);
+        let mut bundle = FluentBundle::new(vec![LANG_EN]);
+        bundle.add_resource(res).expect("Failed to add a resource.");
+        add_functions("preferences", &mut bundle);
+        let args = get_args("preferences");
+        let mut s = String::new();
+        for id in &ids {
+            let msg = bundle.get_message(id).expect("Message found");
+            let mut errors = vec![];
+            if let Some(value) = msg.value() {
+                bundle
+                    .write_pattern(&mut s, value, args.as_ref(), &mut errors)
+                    .expect("Failed to write a pattern.");
+                s.clear();
+            }
+            for attr in msg.attributes() {
+                bundle
+                    .write_pattern(&mut s, attr.value(), args.as_ref(), &mut errors)
+                    .expect("Failed to write a pattern.");
+                s.clear();
+            }
+            assert!(errors.len() == 0, "Resolver errors: {:#?}", errors);
+        }
+    }
+}
+
+iai::main!(iai_resolve_preferences);