metriki-core 0.0.1

A metrics library ported from dropwizard metrics
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

use std::collections::HashMap;
use std::sync::{Arc, RwLock};

use crate::metrics::{Meter, Metric};

/// Entrypoint of all metrics
///
#[derive(Default, Debug)]
pub struct MetricsRegistry {
    inner: Arc<RwLock<Inner>>,
}

#[derive(Default, Debug)]
struct Inner {
    metrics: HashMap<String, Metric>,
}

impl MetricsRegistry {
    /// Return `Meter` that has been registered or just created and resgitered.
    /// Panic if a metric is already register but is not meter
    pub fn meter(&self, name: &str) -> Arc<Meter> {
        let inner = self.inner.read().unwrap();

        if !inner.metrics.contains_key(name) {
            let mut inner_write = self.inner.write().unwrap();
            let meter = Meter::new();
            inner_write
                .metrics
                .insert(name.to_owned(), Metric::Meter(Arc::new(meter)));
        }

        let metric = inner.metrics.get(name).unwrap();
        match metric {
            Metric::Meter(ref m) => m.clone(),
            _ => panic!("A metric with same name and different type is already registered."),
        }
    }

    pub fn snapshots(&self) -> HashMap<String, Metric> {
        let inner = self.inner.read().unwrap();
        inner.metrics.clone()
    }
}