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
use prometheus::core::{Collector, Desc};
use prometheus::proto::MetricFamily;
use prometheus::{IntGaugeVec, Opts, Result};
pub fn monitor_allocator_stats<S: Into<String>>(namespace: S) -> Result<()> {
prometheus::register(Box::new(AllocStatsCollector::new(namespace)?))
}
struct AllocStatsCollector {
descs: Vec<Desc>,
metrics: IntGaugeVec,
}
impl AllocStatsCollector {
fn new<S: Into<String>>(namespace: S) -> Result<AllocStatsCollector> {
let stats = IntGaugeVec::new(
Opts::new("allocator_stats", "Allocator stats").namespace(namespace.into()),
&["type"],
)?;
Ok(AllocStatsCollector {
descs: stats.desc().into_iter().cloned().collect(),
metrics: stats,
})
}
}
impl Collector for AllocStatsCollector {
fn desc(&self) -> Vec<&Desc> {
self.descs.iter().collect()
}
fn collect(&self) -> Vec<MetricFamily> {
if let Ok(Some(stats)) = tikv_alloc::fetch_stats() {
for stat in stats {
self.metrics.with_label_values(&[stat.0]).set(stat.1 as i64);
}
}
self.metrics.collect()
}
}