reudp 0.0.1

A reliable layer on top of UDP.
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

const HEADER_SIZE: usize = 9; // 8 bytes for sequence number, 1 byte for message type

#[derive(Debug, PartialEq, Clone)]
pub enum MessageType {
    Data,
    Ack,
    Heartbeat,
    Unknown(u8),
}

#[derive(Debug, Clone)]
pub struct Message {
    pub sequence: u64,
    pub message_type: MessageType,
    pub payload: Vec<u8>,
}

impl Message {
    pub fn new(sequence: u64, message_type: MessageType, payload: Vec<u8>) -> Self {
        Self {
            sequence,
            message_type,
            payload,
        }
    }

    pub fn to_bytes(&self) -> Vec<u8> {
        let mut bytes = Vec::with_capacity(HEADER_SIZE + self.payload.len());
        bytes.extend_from_slice(&self.sequence.to_be_bytes());
        bytes.push(match self.message_type {
            MessageType::Data => 0,
            MessageType::Ack => 1,
            MessageType::Heartbeat => 2,
            MessageType::Unknown(t) => t,
        });
        bytes.extend_from_slice(&self.payload);
        bytes
    }

    pub fn from_bytes(bytes: &[u8]) -> Self {
        let sequence = u64::from_be_bytes(bytes[..8].try_into().unwrap());
        let message_type = match bytes[8] {
            0 => MessageType::Data,
            1 => MessageType::Ack,
            2 => MessageType::Heartbeat,
            t => {
                eprintln!("Unknown message type: {}", t);
                MessageType::Unknown(t)
            }
        };
        let payload = bytes[9..].to_vec();
        Self {
            sequence,
            message_type,
            payload,
        }
    }
}