deboa-extras 0.0.9-beta.11

deboa extras (serialization, compression, websockets, streams, catchers (middleware) and sse support).
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

use crate::errors::DeboaExtrasError;
use bytes::Bytes;

#[derive(Debug)]
pub struct ServerEvent {
    id: Option<String>,
    event: Option<String>,
    data: Vec<String>,
    retry: Option<u64>,
}

impl Default for ServerEvent {
    fn default() -> Self {
        Self::new()
    }
}

impl ServerEvent {
    pub fn new() -> Self {
        Self { id: None, event: None, data: Vec::new(), retry: None }
    }

    pub fn id(&self) -> &Option<String> {
        &self.id
    }

    pub fn event(&self) -> &Option<String> {
        &self.event
    }

    pub fn data(&self) -> &Vec<String> {
        &self.data
    }

    pub fn add_data(&mut self, data: String) {
        self.data.push(data);
    }

    pub fn retry(&self) -> &Option<u64> {
        &self.retry
    }

    pub fn parse(data: &Bytes) -> Result<ServerEvent, DeboaExtrasError> {
        let data = String::from_utf8_lossy(data.as_ref());
        let text_message = data;
        let lines = text_message.lines();
        let mut event = ServerEvent::new();
        for line in lines {
            if let Some(stripped) = line.strip_prefix("id: ") {
                event.id = Some(stripped.to_string());
            }

            if let Some(stripped) = line.strip_prefix("event: ") {
                event.event = Some(stripped.to_string());
            }

            if let Some(stripped) = line.strip_prefix("retry: ") {
                event.retry = Some(
                    stripped
                        .parse::<u64>()
                        .unwrap(),
                );
            }

            if let Some(stripped) = line.strip_prefix("data: ") {
                if stripped == "[DONE]" {
                    break;
                }

                event.add_data(stripped.to_string());
            }
        }

        Ok(event)
    }
}