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
119
120
121
122

use std::sync::Arc;

use metrics::KeyName;
use metrics::{Counter, CounterFn, Gauge, GaugeFn, Histogram, HistogramFn, Key, Recorder, Unit};
use tracing::trace;

use super::common::KeyLabels;
use super::{METRIC_COUNTER, METRIC_DESCRIPTION, METRIC_GAUGE, METRIC_HISTOGRAM, METRIC_TARGET};

#[derive(Default)]
pub(crate) struct MetricRecorder;

impl MetricRecorder {
    fn register_description(&self, name: &str, description: &str) {
        trace!(
            target: METRIC_TARGET,
            name = name,
            metric_type = METRIC_DESCRIPTION,
            description = description
        );
    }
}

impl Recorder for MetricRecorder {
    fn describe_counter(&self, key_name: KeyName, _unit: Option<Unit>, description: &'static str) {
        self.register_description(key_name.as_str(), description);
    }

    fn describe_gauge(&self, key_name: KeyName, _unit: Option<Unit>, description: &'static str) {
        self.register_description(key_name.as_str(), description);
    }

    fn describe_histogram(&self, key_name: KeyName, _unit: Option<Unit>, description: &'static str) {
        self.register_description(key_name.as_str(), description);
    }

    fn register_counter(&self, key: &Key) -> Counter {
        Counter::from_arc(Arc::new(MetricHandle(key.clone())))
    }

    fn register_gauge(&self, key: &Key) -> Gauge {
        Gauge::from_arc(Arc::new(MetricHandle(key.clone())))
    }

    fn register_histogram(&self, key: &Key) -> Histogram {
        Histogram::from_arc(Arc::new(MetricHandle(key.clone())))
    }
}

pub(crate) struct MetricHandle(Key);

impl CounterFn for MetricHandle {
    fn increment(&self, value: u64) {
        let keylabels: KeyLabels = self.0.clone().into();
        let json_string = serde_json::to_string(&keylabels).unwrap();
        trace!(
            target: METRIC_TARGET,
            key_labels = json_string.as_str(),
            metric_type = METRIC_COUNTER,
            increment = value,
        );
    }

    fn absolute(&self, value: u64) {
        let keylabels: KeyLabels = self.0.clone().into();
        let json_string = serde_json::to_string(&keylabels).unwrap();
        trace!(
            target: METRIC_TARGET,
            key_labels = json_string.as_str(),
            metric_type = METRIC_COUNTER,
            absolute = value,
        );
    }
}

impl GaugeFn for MetricHandle {
    fn increment(&self, value: f64) {
        let keylabels: KeyLabels = self.0.clone().into();
        let json_string = serde_json::to_string(&keylabels).unwrap();
        trace!(
            target: METRIC_TARGET,
            key_labels = json_string.as_str(),
            metric_type = METRIC_GAUGE,
            gauge_inc = value,
        );
    }

    fn decrement(&self, value: f64) {
        let keylabels: KeyLabels = self.0.clone().into();
        let json_string = serde_json::to_string(&keylabels).unwrap();
        trace!(
            target: METRIC_TARGET,
            key_labels = json_string.as_str(),
            metric_type = METRIC_GAUGE,
            gauge_dec = value,
        );
    }

    fn set(&self, value: f64) {
        let keylabels: KeyLabels = self.0.clone().into();
        let json_string = serde_json::to_string(&keylabels).unwrap();
        trace!(
            target: METRIC_TARGET,
            key_labels = json_string.as_str(),
            metric_type = METRIC_GAUGE,
            gauge_set = value,
        );
    }
}

impl HistogramFn for MetricHandle {
    fn record(&self, value: f64) {
        let keylabels: KeyLabels = self.0.clone().into();
        let json_string = serde_json::to_string(&keylabels).unwrap();
        trace!(
            target: METRIC_TARGET,
            key_labels = json_string.as_str(),
            metric_type = METRIC_HISTOGRAM,
            hist_record = value,
        );
    }
}