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

use crate::core::input::InputKind;
use crate::core::label::Labels;
use crate::core::name::MetricName;
use crate::core::void::Void;
use crate::core::{Flush, MetricValue};

use std::rc::Rc;

/// Define metrics, write values and flush them.
pub trait OutputScope: Flush {
    /// Define a raw metric of the specified type.
    fn new_metric(&self, name: MetricName, kind: InputKind) -> OutputMetric;
}

/// Output metrics are not thread safe.
#[derive(Clone)]
pub struct OutputMetric {
    inner: Rc<dyn Fn(MetricValue, Labels)>,
}

impl OutputMetric {
    /// Utility constructor
    pub fn new<F: Fn(MetricValue, Labels) + 'static>(metric: F) -> OutputMetric {
        OutputMetric {
            inner: Rc::new(metric),
        }
    }

    /// Some may prefer the `metric.write(value)` form to the `(metric)(value)` form.
    /// This shouldn't matter as metrics should be of type Counter, Marker, etc.
    #[inline]
    pub fn write(&self, value: MetricValue, labels: Labels) {
        (self.inner)(value, labels)
    }
}

/// A function trait that opens a new metric capture scope.
pub trait Output: Send + Sync + 'static + OutputDyn {
    /// The type of Scope returned byt this output.
    type SCOPE: OutputScope;

    /// Open a new scope for this output.
    fn new_scope(&self) -> Self::SCOPE;

    /// Open a new scope for this output.
    #[deprecated(since = "0.7.2", note = "Use new_scope()")]
    fn output(&self) -> Self::SCOPE {
        self.new_scope()
    }
}

/// A function trait that opens a new metric capture scope.
pub trait OutputDyn {
    /// Open a new scope from this output.
    fn output_dyn(&self) -> Rc<dyn OutputScope + 'static>;
}

/// Blanket impl of dyn output trait
impl<T: Output + Send + Sync + 'static> OutputDyn for T {
    fn output_dyn(&self) -> Rc<dyn OutputScope + 'static> {
        Rc::new(self.new_scope())
    }
}

/// Discard all metric values sent to it.
pub fn output_none() -> Void {
    Void {}
}