marxml 0.1.3

Fast markdown + XML query and mutation. Rust core for the marxml ecosystem.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

//! Parse + select + serialize benches.
//!
//! Run locally with `cargo bench`. In CI, `CodSpeed`'s GitHub Action wraps
//! this same binary via the `codspeed-criterion-compat` shim and produces
//! instruction-count measurements that don't depend on runner load.

#![allow(missing_docs)]

use std::fmt::Write as _;

use codspeed_criterion_compat::{
    black_box, criterion_group, criterion_main, Criterion, Throughput,
};

fn corpus_small() -> String {
    r#"<phase id="1" status="todo">
<task id="1.1"><status>todo</status></task>
<task id="1.2"><status>done</status></task>
</phase>"#
        .to_string()
}

fn corpus_medium() -> String {
    let mut s = String::with_capacity(64 * 1024);
    s.push_str("# Mixed doc\n\nIntro paragraph.\n\n");
    for i in 0..200 {
        writeln!(
            s,
            r#"<phase id="{i}"><task id="{i}.1" status="todo"><status>todo</status></task><task id="{i}.2" status="done"><status>done</status></task></phase>"#,
        )
        .unwrap();
    }
    s
}

fn corpus_large() -> String {
    let mut s = String::with_capacity(1_000_000);
    for i in 0..5000 {
        let state = if i % 2 == 0 { "todo" } else { "done" };
        writeln!(
            s,
            r#"<task id="{i}" status="{state}"><status>{state}</status></task>"#,
        )
        .unwrap();
    }
    s
}

fn bench_parse(c: &mut Criterion) {
    let mut g = c.benchmark_group("parse");
    for (label, src) in [
        ("small_~200B", corpus_small()),
        ("medium_~50KB", corpus_medium()),
        ("large_~500KB", corpus_large()),
    ] {
        g.throughput(Throughput::Bytes(src.len() as u64));
        g.bench_function(label, |b| {
            b.iter(|| {
                let doc = marxml::parse(black_box(&src)).expect("clean corpus");
                black_box(doc);
            });
        });
    }
    g.finish();
}

fn bench_select(c: &mut Criterion) {
    let src = corpus_medium();
    let doc = marxml::parse(&src).expect("clean corpus");
    let sel_tag = marxml::Selector::parse("task").unwrap();
    let sel_attr = marxml::Selector::parse(r#"task[status="todo"]"#).unwrap();
    let sel_deep = marxml::Selector::parse("phase task").unwrap();

    let mut g = c.benchmark_group("select");
    g.bench_function("by_tag", |b| {
        b.iter(|| black_box(doc.select(&sel_tag).count()));
    });
    g.bench_function("by_attr_eq", |b| {
        b.iter(|| black_box(doc.select(&sel_attr).count()));
    });
    g.bench_function("descendant", |b| {
        b.iter(|| black_box(doc.select(&sel_deep).count()));
    });
    g.finish();
}

fn bench_serialize(c: &mut Criterion) {
    let src = corpus_medium();
    let doc = marxml::parse(&src).expect("clean corpus");
    let mut g = c.benchmark_group("serialize");
    g.bench_function("to_xml_tight", |b| {
        b.iter(|| black_box(doc.to_xml(&marxml::SerializeOpts::default())));
    });
    g.bench_function("to_xml_pretty", |b| {
        b.iter(|| black_box(doc.to_xml(&marxml::SerializeOpts::pretty())));
    });
    g.finish();
}

fn bench_mutate(c: &mut Criterion) {
    let src = corpus_medium();
    let doc = marxml::parse(&src).expect("clean corpus");
    let sel = marxml::Selector::parse(r#"task[status="todo"]"#).unwrap();
    let mut g = c.benchmark_group("mutate");
    g.bench_function("update_attr_all_matches", |b| {
        b.iter(|| black_box(doc.update(&sel, &[("status", "done")])));
    });
    g.finish();
}

criterion_group!(
    benches,
    bench_parse,
    bench_select,
    bench_serialize,
    bench_mutate
);
criterion_main!(benches);