grafbabe 0.1.1

Simple openmetrics stats collector
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

use serde::Serialize;
use std::time::Duration;

#[derive(Debug, Serialize)]
pub struct Series {
    pub name: String,
    pub label: Option<String>,
    // Optional to accomodate for the fact that a metric might not exist for every time slice
    pub values: Vec<Option<i64>>,
}

impl Series {
    pub fn new(name: String, label: Option<String>, capacity: usize) -> Self {
        Series {
            name,
            label,
            values: Vec::with_capacity(capacity),
        }
    }

    pub fn push(&mut self, value: Option<i64>) {
        self.values.push(value)
    }
}

#[derive(Debug, Default, Serialize)]
pub struct Metrics {
    pub timestamps: Vec<i64>,
    pub series: Vec<Series>,
}

const EVENT_DURATION_MINS: usize = 1;
const TOTAL_SAMPLES_15M: usize = 15 / EVENT_DURATION_MINS;
const TOTAL_SAMPLES_30M: usize = 30 / EVENT_DURATION_MINS;
const TOTAL_SAMPLES_1H: usize = 60 / EVENT_DURATION_MINS;
const TOTAL_SAMPLES_4H: usize = TOTAL_SAMPLES_1H * 4;
const TOTAL_SAMPLES_12H: usize = TOTAL_SAMPLES_1H * 12;
const TOTAL_SAMPLES_DAY: usize = TOTAL_SAMPLES_1H * 24;
const TOTAL_SAMPLES_WEEK: usize = TOTAL_SAMPLES_DAY * 7;
const TOTAL_SAMPLES_MONTH: usize = TOTAL_SAMPLES_DAY * 30;

#[derive(Copy, Clone, Debug)]
pub enum Window {
    QuarterHour,
    HalfHour,
    Hour,
    Hour4,
    Hour12,
    Day,
    Week,
    Month,
}

impl Window {
    pub fn total_samples(self) -> usize {
        match self {
            Window::QuarterHour => TOTAL_SAMPLES_15M,
            Window::HalfHour => TOTAL_SAMPLES_30M,
            Window::Hour => TOTAL_SAMPLES_1H,
            Window::Hour4 => TOTAL_SAMPLES_4H,
            Window::Hour12 => TOTAL_SAMPLES_12H,
            Window::Day => TOTAL_SAMPLES_DAY,
            Window::Week => TOTAL_SAMPLES_WEEK,
            Window::Month => TOTAL_SAMPLES_MONTH,
        }
    }

    pub fn duration(self) -> Duration {
        match self {
            Window::QuarterHour => Duration::from_mins(15),
            Window::HalfHour => Duration::from_mins(30),
            Window::Hour => Duration::from_hours(1),
            Window::Hour4 => Duration::from_hours(4),
            Window::Hour12 => Duration::from_hours(12),
            Window::Day => Duration::from_hours(24),
            Window::Week => Duration::from_hours(24) * 7,
            Window::Month => Duration::from_hours(24) * 30,
        }
    }

    pub fn as_ms(self) -> i64 {
        self.duration().as_millis() as i64
    }
}