aboutsummaryrefslogtreecommitdiff
path: root/src/message/signalproxy/heartbeat.rs
blob: 32df3d423512acd81fa4313952c19ad2d14855c7 (plain)
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
use crate::error::ProtocolError;
use crate::message::MessageType;
use crate::primitive::{DateTime, Variant, VariantList};
use crate::{deserialize::Deserialize, serialize::Serialize};

#[derive(Clone, Debug, std::cmp::PartialEq)]
pub struct HeartBeat {
    timestamp: DateTime,
}

impl Serialize for HeartBeat {
    fn serialize(&self) -> Result<Vec<std::primitive::u8>, ProtocolError> {
        let mut res = VariantList::new();

        res.push(Variant::i32(MessageType::HeartBeat as i32));
        res.push(Variant::DateTime(self.timestamp.clone()));

        res.serialize()
    }
}

impl Deserialize for HeartBeat {
    fn parse(b: &[std::primitive::u8]) -> Result<(std::primitive::usize, Self), ProtocolError> {
        let (size, mut res) = VariantList::parse(&b)?;

        res.remove(0);

        Ok((
            size,
            Self {
                timestamp: match_variant!(res.remove(0), Variant::DateTime),
            },
        ))
    }
}

#[derive(Clone, Debug, std::cmp::PartialEq)]
pub struct HeartBeatReply {
    timestamp: DateTime,
}

impl Serialize for HeartBeatReply {
    fn serialize(&self) -> Result<Vec<std::primitive::u8>, ProtocolError> {
        let mut res = VariantList::new();

        res.push(Variant::i32(MessageType::HeartBeatReply as i32));
        res.push(Variant::DateTime(self.timestamp.clone()));

        res.serialize()
    }
}

impl Deserialize for HeartBeatReply {
    fn parse(b: &[std::primitive::u8]) -> Result<(std::primitive::usize, Self), ProtocolError> {
        let (size, mut res) = VariantList::parse(&b)?;

        res.remove(0);

        Ok((
            size,
            Self {
                timestamp: match_variant!(res.remove(0), Variant::DateTime),
            },
        ))
    }
}