1 use criterion::black_box;
2
3 use winnow::combinator::alt;
4 use winnow::combinator::repeat;
5 use winnow::prelude::*;
6 use winnow::token::take_till;
7 use winnow::token::take_while;
8
contains_token(c: &mut criterion::Criterion)9 fn contains_token(c: &mut criterion::Criterion) {
10 let data = [
11 ("contiguous", CONTIGUOUS),
12 ("interleaved", INTERLEAVED),
13 ("canada", CANADA),
14 ];
15 let mut group = c.benchmark_group("contains_token");
16 for (name, sample) in data {
17 let len = sample.len();
18 group.throughput(criterion::Throughput::Bytes(len as u64));
19
20 group.bench_with_input(criterion::BenchmarkId::new("slice", name), &len, |b, _| {
21 b.iter(|| black_box(parser_slice.parse_peek(black_box(sample)).unwrap()));
22 });
23 group.bench_with_input(criterion::BenchmarkId::new("array", name), &len, |b, _| {
24 b.iter(|| black_box(parser_array.parse_peek(black_box(sample)).unwrap()));
25 });
26 group.bench_with_input(criterion::BenchmarkId::new("tuple", name), &len, |b, _| {
27 b.iter(|| black_box(parser_tuple.parse_peek(black_box(sample)).unwrap()));
28 });
29 group.bench_with_input(
30 criterion::BenchmarkId::new("closure-or", name),
31 &len,
32 |b, _| {
33 b.iter(|| black_box(parser_closure_or.parse_peek(black_box(sample)).unwrap()));
34 },
35 );
36 group.bench_with_input(
37 criterion::BenchmarkId::new("closure-matches", name),
38 &len,
39 |b, _| {
40 b.iter(|| {
41 black_box(
42 parser_closure_matches
43 .parse_peek(black_box(sample))
44 .unwrap(),
45 )
46 });
47 },
48 );
49 }
50 group.finish();
51 }
52
parser_slice(input: &mut &str) -> PResult<usize>53 fn parser_slice(input: &mut &str) -> PResult<usize> {
54 let contains = &['0', '1', '2', '3', '4', '5', '6', '7', '8', '9'][..];
55 repeat(
56 0..,
57 alt((take_while(1.., contains), take_till(1.., contains))),
58 )
59 .parse_next(input)
60 }
61
parser_array(input: &mut &str) -> PResult<usize>62 fn parser_array(input: &mut &str) -> PResult<usize> {
63 let contains = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9'];
64 repeat(
65 0..,
66 alt((take_while(1.., contains), take_till(1.., contains))),
67 )
68 .parse_next(input)
69 }
70
parser_tuple(input: &mut &str) -> PResult<usize>71 fn parser_tuple(input: &mut &str) -> PResult<usize> {
72 let contains = ('0', '1', '2', '3', '4', '5', '6', '7', '8', '9');
73 repeat(
74 0..,
75 alt((take_while(1.., contains), take_till(1.., contains))),
76 )
77 .parse_next(input)
78 }
79
parser_closure_or(input: &mut &str) -> PResult<usize>80 fn parser_closure_or(input: &mut &str) -> PResult<usize> {
81 let contains = |c: char| {
82 c == '0'
83 || c == '1'
84 || c == '2'
85 || c == '3'
86 || c == '4'
87 || c == '5'
88 || c == '6'
89 || c == '7'
90 || c == '8'
91 || c == '9'
92 };
93 repeat(
94 0..,
95 alt((take_while(1.., contains), take_till(1.., contains))),
96 )
97 .parse_next(input)
98 }
99
parser_closure_matches(input: &mut &str) -> PResult<usize>100 fn parser_closure_matches(input: &mut &str) -> PResult<usize> {
101 let contains = |c: char| matches!(c, '0' | '1' | '2' | '3' | '4' | '5' | '6' | '7' | '8' | '9');
102 repeat(
103 0..,
104 alt((take_while(1.., contains), take_till(1.., contains))),
105 )
106 .parse_next(input)
107 }
108
109 const CONTIGUOUS: &str = "012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789";
110 const INTERLEAVED: &str = "0123456789abc0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab0123456789ab";
111 const CANADA: &str = include_str!("../third_party/nativejson-benchmark/data/canada.json");
112
113 criterion::criterion_group!(benches, contains_token);
114 criterion::criterion_main!(benches);
115