-
Notifications
You must be signed in to change notification settings - Fork 64
/
text.rs
95 lines (83 loc) · 3.04 KB
/
text.rs
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
// Benchmark inspired by https://github.com/tikv/rust-prometheus/blob/ab1ca7285d3463504381a5025ae1951e020d6796/benches/text_encoder.rs
use criterion::{black_box, criterion_group, criterion_main, Criterion};
use prometheus_client::encoding::{self, EncodeLabelSet, EncodeLabelValue, LabelValueEncoder};
use prometheus_client::metrics::counter::Counter;
use prometheus_client::metrics::family::Family;
use prometheus_client::metrics::histogram::{exponential_buckets, Histogram};
use prometheus_client::registry::Registry;
use std::fmt::Write;
pub fn text(c: &mut Criterion) {
c.bench_function("encode", |b| {
#[derive(Clone, Hash, PartialEq, Eq, EncodeLabelSet, Debug)]
struct Labels {
method: Method,
status: Status,
some_number: u64,
}
#[derive(Clone, Hash, PartialEq, Eq, EncodeLabelValue, Debug)]
enum Method {
Get,
#[allow(dead_code)]
Put,
}
#[derive(Clone, Hash, PartialEq, Eq, Debug)]
enum Status {
Two,
#[allow(dead_code)]
Four,
#[allow(dead_code)]
Five,
}
impl prometheus_client::encoding::EncodeLabelValue for Status {
fn encode(&self, writer: &mut LabelValueEncoder) -> Result<(), std::fmt::Error> {
let status = match self {
Status::Two => "200",
Status::Four => "400",
Status::Five => "500",
};
writer.write_str(status)?;
Ok(())
}
}
let mut registry = Registry::default();
for i in 0..100 {
let counter_family = Family::<Labels, Counter>::default();
let histogram_family = Family::<Labels, Histogram>::new_with_constructor(|| {
Histogram::new(exponential_buckets(1.0, 2.0, 10))
});
registry.register(
format!("my_counter_{}", i),
"My counter",
counter_family.clone(),
);
registry.register(
format!("my_histogram_{}", i),
"My histogram",
histogram_family.clone(),
);
for j in 0u32..100 {
counter_family
.get_or_create(&Labels {
method: Method::Get,
status: Status::Two,
some_number: j.into(),
})
.inc();
histogram_family
.get_or_create(&Labels {
method: Method::Get,
status: Status::Two,
some_number: j.into(),
})
.observe(j.into());
}
}
let mut buffer = String::new();
b.iter(|| {
encoding::text::encode(&mut buffer, ®istry).unwrap();
black_box(&mut buffer);
})
});
}
criterion_group!(benches, text);
criterion_main!(benches);