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

// Copyright (c) Aptos
// SPDX-License-Identifier: Apache-2.0

#![forbid(unsafe_code)]

use crate::{build_information::get_build_information, system_information::collect_system_info};
use prometheus::proto::MetricFamily;
use std::collections::BTreeMap;

/// Used to expose system and build information
pub fn get_system_and_build_information(chain_id: Option<String>) -> BTreeMap<String, String> {
    let mut system_and_build_information = get_build_information(chain_id);
    collect_system_info(&mut system_and_build_information);
    system_and_build_information
}

/// Inserts an optional value into the given map iff the value exists
pub(crate) fn insert_optional_value(
    map: &mut BTreeMap<String, String>,
    key: &str,
    value: Option<String>,
) {
    if let Some(value) = value {
        map.insert(key.to_string(), value);
    }
}

/// Sums all gauge counts in the given set of metric families
pub fn sum_all_gauges(metric_families: &Vec<MetricFamily>) -> f64 {
    let mut gauge_sum = 0.0;
    for metric_family in metric_families {
        for metric in metric_family.get_metric() {
            gauge_sum += metric.get_gauge().get_value();
        }
    }
    gauge_sum
}

/// Sums all histogram sample counts in the given set of metric families
pub fn sum_all_histogram_counts(metric_families: &Vec<MetricFamily>) -> f64 {
    let mut count_sum = 0.0;
    for metric_family in metric_families {
        for metric in metric_family.get_metric() {
            count_sum += metric.get_histogram().get_sample_count() as f64
        }
    }
    count_sum
}

/// Sums all histogram sample sums in the given set of metric families
pub fn sum_all_histogram_sums(metric_families: &Vec<MetricFamily>) -> f64 {
    let mut count_sum = 0.0;
    for metric_family in metric_families {
        for metric in metric_family.get_metric() {
            count_sum += metric.get_histogram().get_sample_sum() as f64
        }
    }
    count_sum
}