mecha10-core 0.8.0

Mecha10 core runtime and foundational types
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
118
119
120
121

//! Infrastructure configuration types
//!
//! Configuration structures for Redis, PostgreSQL, MLflow, MinIO, and other
//! infrastructure services used by the Mecha10 framework.

use serde::Deserialize;

#[derive(Debug, Deserialize, Clone)]
pub struct InfrastructureConfig {
    pub redis: RedisConfig,
    #[serde(default)]
    pub control_plane_redis: Option<RedisConfig>,
    #[serde(default)]
    pub bridge: Option<BridgeConfig>,
    #[serde(default)]
    pub postgres: Option<PostgresConfig>,
    #[serde(default)]
    pub mlflow: Option<MlflowConfig>,
    #[serde(default)]
    pub minio: Option<MinioConfig>,
}

#[derive(Debug, Deserialize, Clone)]
pub struct RedisConfig {
    pub url: String,
    #[serde(default = "default_max_connections")]
    pub max_connections: usize,
    #[serde(default = "default_connection_timeout")]
    pub connection_timeout_ms: u64,
}

impl Default for RedisConfig {
    fn default() -> Self {
        Self {
            url: "redis://localhost:6379".to_string(),
            max_connections: default_max_connections(),
            connection_timeout_ms: default_connection_timeout(),
        }
    }
}

#[derive(Debug, Deserialize, Clone)]
pub struct PostgresConfig {
    pub host: String,
    pub port: u16,
    pub database: String,
    pub username: String,
    pub password: String,
    #[serde(default = "default_max_connections")]
    pub max_connections: usize,
}

#[derive(Debug, Deserialize, Clone)]
pub struct MlflowConfig {
    pub tracking_uri: String,
    pub artifact_location: String,
}

#[derive(Debug, Deserialize, Clone)]
pub struct MinioConfig {
    pub endpoint: String,
    pub access_key: String,
    pub secret_key: String,
    #[serde(default)]
    pub secure: bool,
}

#[derive(Debug, Deserialize, Clone)]
pub struct BridgeConfig {
    #[serde(default = "default_bridge_enabled")]
    pub enabled: bool,

    #[serde(default = "default_local_topics")]
    pub local_topics: Vec<String>,

    #[serde(default = "default_bridged_topics")]
    pub bridged_topics: Vec<String>,

    #[serde(default)]
    pub robot_id: Option<String>,
}

impl Default for BridgeConfig {
    fn default() -> Self {
        Self {
            enabled: default_bridge_enabled(),
            local_topics: default_local_topics(),
            bridged_topics: default_bridged_topics(),
            robot_id: None,
        }
    }
}

fn default_bridge_enabled() -> bool {
    true
}

fn default_local_topics() -> Vec<String> {
    vec![
        "/sensor/camera/*".to_string(),
        "/sensor/lidar/*".to_string(),
        "/sensor/imu/*".to_string(),
    ]
}

fn default_bridged_topics() -> Vec<String> {
    vec![
        "/system/*".to_string(),
        "/telemetry/*".to_string(),
        "/logs/*".to_string(),
        "/fleet/*".to_string(),
    ]
}

fn default_max_connections() -> usize {
    10
}

fn default_connection_timeout() -> u64 {
    5000
}