coerce 0.3.1-prerelease2

Async actor runtime and distributed systems framework
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

use crate::remote::net::proto::protocol::{
    NewNodeEvent, NodeRemovedEvent, SystemEvent as SysEvent,
};
use crate::remote::net::StreamMessage;
use crate::remote::stream::pubsub::Topic;
use protobuf::{parse_from_bytes, Message, ProtobufEnum, ProtobufError};
use uuid::Uuid;

pub struct SystemTopic;

pub enum ClusterEvent {
    NodeAdded(Uuid),
    NodeRemoved(Uuid),
}

pub enum SystemEvent {
    Cluster(ClusterEvent),
}

impl Topic for SystemTopic {
    type Message = SystemEvent;

    fn topic_name() -> &'static str {
        "coerce"
    }
}

impl From<NewNodeEvent> for SystemEvent {
    fn from(message: NewNodeEvent) -> Self {
        SystemEvent::Cluster(ClusterEvent::NodeAdded(
            Uuid::parse_str(message.get_node_id()).unwrap(),
        ))
    }
}

impl From<NodeRemovedEvent> for SystemEvent {
    fn from(message: NodeRemovedEvent) -> Self {
        SystemEvent::Cluster(ClusterEvent::NodeRemoved(
            Uuid::parse_str(message.get_node_id()).unwrap(),
        ))
    }
}

impl StreamMessage for SystemEvent {
    fn read_from_bytes(data: Vec<u8>) -> Option<Self> {
        match data.split_first() {
            Some((event, message)) => match SysEvent::from_i32(*event as i32) {
                Some(SysEvent::ClusterNodeRemoved) => Some(
                    parse_from_bytes::<NodeRemovedEvent>(message)
                        .unwrap()
                        .into(),
                ),
                Some(SysEvent::ClusterNewNode) => {
                    Some(parse_from_bytes::<NewNodeEvent>(message).unwrap().into())
                }
                None => None,
            },
            None => None,
        }
    }

    fn write_to_bytes(&self) -> Option<Vec<u8>> {
        match self {
            SystemEvent::Cluster(cluster) => match cluster {
                ClusterEvent::NodeAdded(id) => {
                    let event = NewNodeEvent {
                        node_id: id.to_string(),
                        ..NewNodeEvent::default()
                    };

                    write_event(SysEvent::ClusterNewNode, event.write_to_bytes())
                }
                ClusterEvent::NodeRemoved(id) => {
                    let event = NodeRemovedEvent {
                        node_id: id.to_string(),
                        ..NodeRemovedEvent::default()
                    };

                    write_event(SysEvent::ClusterNodeRemoved, event.write_to_bytes())
                }
            },
        }
    }
}

fn write_event(system_event: SysEvent, message: Result<Vec<u8>, ProtobufError>) -> Option<Vec<u8>> {
    let o = match message {
        Ok(mut message) => {
            let mut bytes = vec![];
            bytes.push(system_event as u8);
            bytes.append(&mut message);

            Some(bytes)
        }
        Err(_) => None,
    };

    o
}