neuronic 0.1.0

Real-time graphical visualization of Caryatid message bus flow
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

//! Configuration loading for Neuronic.
//!
//! Configuration is loaded from TOML files with environment variable overrides.
//! The loading order is:
//!
//! 1. `config.default.toml` (if it exists) - shipped defaults
//! 2. User-specified config file (e.g., `neuronic.toml`)
//! 3. Environment variables with `NEURONIC_` prefix

use anyhow::Result;
use config::{Config, Environment, File};
use serde::Deserialize;
use std::path::Path;

/// Default configuration file name.
pub const DEFAULT_CONFIG_FILE: &str = "config.default.toml";

/// Application configuration.
#[derive(Debug, Clone, Deserialize, Default)]
pub struct NeuronicConfig {
    /// Filter settings.
    #[serde(default)]
    pub filter: FilterConfig,

    /// Graph layout settings.
    #[serde(default)]
    pub graph: GraphConfig,
}

/// Filter configuration.
#[derive(Debug, Clone, Deserialize)]
pub struct FilterConfig {
    /// Topic prefixes to ignore.
    #[serde(default = "default_ignored_topics")]
    pub ignored_topics: Vec<String>,
}

impl Default for FilterConfig {
    fn default() -> Self {
        Self {
            ignored_topics: default_ignored_topics(),
        }
    }
}

fn default_ignored_topics() -> Vec<String> {
    vec!["cardano.query.".to_string()]
}

/// Graph layout configuration.
#[derive(Debug, Clone, Deserialize)]
pub struct GraphConfig {
    /// Backlog warning threshold.
    #[serde(default = "default_backlog_warning")]
    pub backlog_warning: u64,

    /// Backlog critical threshold.
    #[serde(default = "default_backlog_critical")]
    pub backlog_critical: u64,

    /// Pending time warning threshold in milliseconds.
    #[serde(default = "default_pending_warning_ms")]
    pub pending_warning_ms: u64,

    /// Pending time critical threshold in milliseconds.
    #[serde(default = "default_pending_critical_ms")]
    pub pending_critical_ms: u64,
}

impl Default for GraphConfig {
    fn default() -> Self {
        Self {
            backlog_warning: default_backlog_warning(),
            backlog_critical: default_backlog_critical(),
            pending_warning_ms: default_pending_warning_ms(),
            pending_critical_ms: default_pending_critical_ms(),
        }
    }
}

fn default_backlog_warning() -> u64 {
    100
}
fn default_backlog_critical() -> u64 {
    1000
}
fn default_pending_warning_ms() -> u64 {
    500
}
fn default_pending_critical_ms() -> u64 {
    2000
}

impl NeuronicConfig {
    /// Load configuration from a file.
    ///
    /// Loads configuration in the following order (later sources override earlier):
    /// 1. `config.default.toml` in the current directory (if present)
    /// 2. The specified config file path
    /// 3. Environment variables with `NEURONIC_` prefix
    ///
    /// If no config files exist, sensible defaults are used.
    pub fn load(path: &Path) -> Result<Self> {
        let config = Config::builder()
            // First, try to load the default config file
            .add_source(File::with_name(DEFAULT_CONFIG_FILE).required(false))
            // Then load the user-specified config file
            .add_source(File::from(path).required(false))
            // Finally, override with environment variables
            .add_source(Environment::with_prefix("NEURONIC").separator("_"))
            .build()?;

        let neuronic_config: NeuronicConfig = config.try_deserialize().unwrap_or_default();

        Ok(neuronic_config)
    }
}