mecha10-core 0.7.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

//! Node metadata types and builders for service discovery

use serde::{Deserialize, Serialize};
use std::collections::HashMap;

#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "lowercase")]
pub enum NodeStatus {
    Starting,
    Running,
    Stopping,
    Stopped,
    Unhealthy,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct NodeMetadata {
    pub node_id: String,

    #[serde(default)]
    pub node_type: String,

    #[serde(default = "default_status")]
    pub status: NodeStatus,

    #[serde(default)]
    pub started_at: u64,

    #[serde(default)]
    pub last_heartbeat: u64,

    #[serde(default)]
    pub publishes: Vec<String>,

    #[serde(default)]
    pub subscribes: Vec<String>,

    #[serde(default)]
    pub capabilities: Vec<String>,

    #[serde(default)]
    pub metadata: HashMap<String, String>,
}

fn default_status() -> NodeStatus {
    NodeStatus::Running
}

impl Default for NodeMetadata {
    fn default() -> Self {
        Self {
            node_id: String::new(),
            node_type: String::new(),
            status: NodeStatus::Running,
            started_at: current_timestamp(),
            last_heartbeat: current_timestamp(),
            publishes: Vec::new(),
            subscribes: Vec::new(),
            capabilities: Vec::new(),
            metadata: HashMap::new(),
        }
    }
}

impl NodeMetadata {
    /// Create new metadata for a node
    pub fn new(node_id: impl Into<String>) -> Self {
        Self {
            node_id: node_id.into(),
            ..Default::default()
        }
    }

    crate::builder_setter!(
        /// Set node type
        into, with_type, node_type, impl Into<String>
    );

    crate::builder_setter!(
        /// Add a published topic
        push, publishes, publishes, impl Into<String>
    );

    crate::builder_setter!(
        /// Add a subscribed topic
        push, subscribes, subscribes, impl Into<String>
    );

    crate::builder_setter!(
        /// Add a capability
        push, capability, capabilities, impl Into<String>
    );

    crate::builder_setter!(
        /// Add metadata
        insert, metadata, metadata, impl Into<String>, impl Into<String>
    );
}

/// Get current Unix timestamp in seconds
pub(crate) fn current_timestamp() -> u64 {
    crate::prelude::now_secs()
}