fuel-web-utils 0.0.27

Fuel library for web utils
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

use async_trait::async_trait;
use prometheus::Registry;
use rand::{distr::Alphanumeric, Rng};

#[async_trait]
pub trait TelemetryMetrics: Send + Sync + 'static {
    fn registry(&self) -> &Registry;

    fn metrics(&self) -> Option<Self>
    where
        Self: std::marker::Sized;

    fn generate_random_prefix() -> String {
        rand::rng()
            .sample_iter(&Alphanumeric)
            .filter(|c| c.is_ascii_alphabetic())
            .take(6)
            .map(char::from)
            .collect()
    }

    fn gather_metrics(&self) -> String {
        use prometheus::Encoder;

        let encoder = prometheus::TextEncoder::new();
        let mut buffer = Vec::new();

        if let Err(e) = encoder.encode(&self.registry().gather(), &mut buffer) {
            tracing::error!("could not encode custom metrics: {}", e);
        }

        String::from_utf8(buffer).unwrap_or_default()
    }
}