robotrt-middleware-core 0.1.0-beta.2

RobotRT modular robotics runtime and middleware components.
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

use std::time::Duration;

use transport_core::Endpoint;

use super::SimpleDiscovery;
use super::types::DiscoveryState;

impl SimpleDiscovery {
    pub fn register_topic(&mut self, name: impl Into<String>) {
        let name = name.into();
        self.topics.insert(name.clone());
        self.topic_state.insert(name, DiscoveryState::default());
    }

    pub fn register_service(&mut self, name: impl Into<String>) {
        let name = name.into();
        self.services.insert(name.clone());
        self.service_state.insert(name, DiscoveryState::default());
    }

    pub fn register_mission(&mut self, name: impl Into<String>) {
        let name = name.into();
        self.missions.insert(name.clone());
        self.mission_state.insert(name, DiscoveryState::default());
    }

    pub fn register_topic_with_ttl(&mut self, name: impl Into<String>, ttl: Duration) {
        let name = name.into();
        self.topics.insert(name.clone());
        self.topic_state.insert(name, DiscoveryState::with_ttl(ttl));
    }

    pub fn register_service_with_ttl(&mut self, name: impl Into<String>, ttl: Duration) {
        let name = name.into();
        self.services.insert(name.clone());
        self.service_state
            .insert(name, DiscoveryState::with_ttl(ttl));
    }

    pub fn register_mission_with_ttl(&mut self, name: impl Into<String>, ttl: Duration) {
        let name = name.into();
        self.missions.insert(name.clone());
        self.mission_state
            .insert(name, DiscoveryState::with_ttl(ttl));
    }

    pub fn add_labels(&mut self, key: impl Into<String>, labels: Vec<String>) {
        self.labels.insert(key.into(), labels);
    }

    pub fn register_endpoint(&mut self, name: impl Into<String>, endpoint: Endpoint) {
        let name = name.into();
        self.endpoints.insert(name.clone(), endpoint);
        self.endpoint_state.insert(name, DiscoveryState::default());
    }

    pub fn register_endpoint_with_ttl(
        &mut self,
        name: impl Into<String>,
        endpoint: Endpoint,
        ttl: Duration,
    ) {
        let name = name.into();
        self.endpoints.insert(name.clone(), endpoint);
        self.endpoint_state
            .insert(name, DiscoveryState::with_ttl(ttl));
    }

    pub fn update_endpoint_labels(&mut self, name: &str, labels: Vec<String>) -> bool {
        let Some(endpoint) = self.endpoints.get_mut(name) else {
            return false;
        };
        endpoint.labels = labels;
        true
    }

    pub fn unregister_endpoint(&mut self, name: &str) -> bool {
        let removed_endpoint = self.endpoints.remove(name).is_some();
        let removed_state = self.endpoint_state.remove(name).is_some();
        if removed_endpoint || removed_state {
            self.remove_labels_if_unused(name);
        }
        removed_endpoint || removed_state
    }
}