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

use crate::{database, parse_prometheus};
use prometheus_scraper::ParseError;
use rusqlite::Connection;
use std::{fmt, fs::File, io::Read};

pub fn seed_database(mut connection: Connection, mock_data_path: &str) -> Result<(), SeedError> {
    log::info!("Seeding database");
    let mut f = File::open(mock_data_path).map_err(SeedError::IO)?;
    let mut s = String::new();
    f.read_to_string(&mut s).map_err(SeedError::IO)?;
    let metrics = parse_prometheus(&s).map_err(SeedError::Parse)?;

    // for n in 0..(30 * 24 * 60) {
    //     if n % 100 == 0 {
    //         log::debug!("Seed progress: {n}");
    //     }
    database::store_snapshot(&mut connection, &metrics).map_err(SeedError::DB)?;
    // }

    Ok(())
}

#[derive(Debug)]
pub enum SeedError {
    IO(std::io::Error),
    Parse(ParseError),
    DB(rusqlite::Error),
}
impl fmt::Display for SeedError {
    fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
        match self {
            SeedError::IO(e) => e.fmt(fmt),
            SeedError::Parse(e) => e.fmt(fmt),
            SeedError::DB(e) => e.fmt(fmt),
        }
    }
}