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

use std::{net::SocketAddr, time::Duration};

use serde::Deserialize;

#[derive(Debug, Deserialize)]
pub(crate) struct Config {
    /// The sink's type.
    pub(crate) sink: Sink,
    /// The address to expose for scraping.
    pub(crate) address: SocketAddr,
    /// How long samples should be considered in summaries.
    #[serde(default)]
    pub(crate) retention: Retention,
    /// Quantiles to use for aggregating distribution metrics into a summary.
    #[serde(default = "default_quantiles")]
    pub(crate) quantiles: Vec<f64>,
    /// Labels that will be added to all metrics.
    #[serde(default)]
    pub(crate) global_labels: Vec<(String, String)>,
    /// The maximum time between compaction ticks.
    #[serde(with = "humantime_serde", default = "default_compaction_interval")]
    pub(crate) compaction_interval: Duration,
}

#[derive(Debug, PartialEq, Deserialize)]
pub(crate) enum Sink {
    Prometheus,
}

#[derive(Debug, PartialEq, Deserialize)]
pub(crate) enum Retention {
    Forever,
    ResetOnScrape,
    // TODO: `SlidingWindow`
}

impl Default for Retention {
    fn default() -> Self {
        Self::ResetOnScrape
    }
}

fn default_quantiles() -> Vec<f64> {
    vec![0.75, 0.9, 0.95, 0.99]
}

fn default_compaction_interval() -> Duration {
    // 1m, 30s, 15s, 10s are often used values of prometheus's `scrape_interval`.
    // 1.1s is a good value that splits the scrape interval uniformly enough.
    Duration::from_millis(1100)
}