autometrics 3.0.0

Easily add metrics to your code that actually help you spot and debug issues in production. Built on Prometheus and OpenTelemetry.
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

#[cfg(debug_assertions)]
use crate::__private::FunctionDescription;
use crate::constants::*;
use crate::labels::{BuildInfoLabels, CounterLabels, GaugeLabels, HistogramLabels};
use crate::tracker::TrackMetrics;
use metrics::{
    counter, describe_counter, describe_gauge, describe_histogram, gauge, histogram, Gauge, Unit,
};
use std::{sync::Once, time::Instant};

static DESCRIBE_METRICS: Once = Once::new();
static SET_BUILD_INFO: Once = Once::new();

fn describe_metrics() {
    DESCRIBE_METRICS.call_once(|| {
        describe_counter!(COUNTER_NAME_PROMETHEUS, COUNTER_DESCRIPTION);
        describe_histogram!(
            HISTOGRAM_NAME_PROMETHEUS,
            Unit::Seconds,
            HISTOGRAM_DESCRIPTION
        );
        describe_gauge!(GAUGE_NAME_PROMETHEUS, GAUGE_DESCRIPTION);
        describe_gauge!(BUILD_INFO_NAME, BUILD_INFO_DESCRIPTION);
    });
}

pub struct MetricsTracker {
    gauge: Option<Gauge>,
    start: Instant,
}

impl TrackMetrics for MetricsTracker {
    fn start(gauge_labels: Option<&GaugeLabels>) -> Self {
        describe_metrics();

        let gauge = if let Some(gauge_labels) = gauge_labels {
            let gauge = gauge!(GAUGE_NAME, &gauge_labels.to_array());
            gauge.increment(1.0);
            Some(gauge)
        } else {
            None
        };

        Self {
            gauge,
            start: Instant::now(),
        }
    }

    fn finish(self, counter_labels: &CounterLabels, histogram_labels: &HistogramLabels) {
        let duration = self.start.elapsed().as_secs_f64();
        counter!(COUNTER_NAME_PROMETHEUS, &counter_labels.to_vec()).increment(1);
        histogram!(HISTOGRAM_NAME_PROMETHEUS, &histogram_labels.to_vec()).record(duration);
        if let Some(gauge) = self.gauge {
            gauge.decrement(1.0);
        }
    }

    fn set_build_info(build_info_labels: &BuildInfoLabels) {
        SET_BUILD_INFO.call_once(|| {
            gauge!(BUILD_INFO_NAME, &build_info_labels.to_vec()).set(1.0);
        });
    }

    #[cfg(debug_assertions)]
    fn intitialize_metrics(function_descriptions: &[FunctionDescription]) {
        for function in function_descriptions {
            let labels = &CounterLabels::from(function).to_vec();
            counter!(COUNTER_NAME, labels).increment(0);
        }
    }
}