mltop 0.6.3

Resource monitor for ML engineers written in Rust
Documentation 
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

use serde::Deserialize;
use std::env::home_dir;
use std::fs::read_to_string;
use std::sync::OnceLock;

pub mod theme;

pub use theme::Theme;

pub const REFRESH_RATE_MILLIS: u64 = 1000;
pub const GRAPH_X_AXIS_WINDOW_IN_SECONDS: usize = 120;
pub const MAX_STORED_SNAPSHOTS: usize = 300;
pub const MESSAGE_EXPIRATION_IN_SECONDS: u64 = 10;

const DEFAULT_CONFIG_FILE: &str = ".config/mltop/config.toml";

pub static CONFIG: OnceLock<Config> = OnceLock::new();

#[derive(Default, Deserialize, Debug)]
pub struct Config {
    pub theme: Theme,
}

impl Config {
    pub fn get() -> Config {
        let Some(home) = home_dir() else {
            return Config::default();
        };

        let config_file = home.join(DEFAULT_CONFIG_FILE);
        match config_file.is_file() {
            false => Self::default(),
            true => {
                let Ok(contents) = read_to_string(&config_file) else {
                    return Config::default();
                };
                toml::from_str(&contents).unwrap_or(Config::default())
            }
        }
    }
}

pub fn init_config() {
    let config = Config::get();
    CONFIG
        .set(config)
        .expect("config should be initialized only once");
}

pub fn get_config() -> &'static Config {
    CONFIG.get().expect("config should be initialized")
}