im-core 0.1.0

Rust IM SDK for Awiki clients built on Agent Network Protocol (ANP)
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

use std::time::Duration;

#[derive(Debug, Clone, PartialEq, Eq)]
pub(crate) struct RuntimeTimeouts {
    connect_timeout: Duration,
    request_timeout: Duration,
    websocket_idle_timeout: Duration,
    websocket_ping_timeout: Duration,
    attachment_transfer_timeout: Duration,
    db_request_timeout: Duration,
}

impl RuntimeTimeouts {
    pub(crate) fn new(
        connect_timeout: Duration,
        request_timeout: Duration,
        websocket_idle_timeout: Duration,
        websocket_ping_timeout: Duration,
        attachment_transfer_timeout: Duration,
        db_request_timeout: Duration,
    ) -> Self {
        Self {
            connect_timeout,
            request_timeout,
            websocket_idle_timeout,
            websocket_ping_timeout,
            attachment_transfer_timeout,
            db_request_timeout,
        }
    }

    pub(crate) fn connect_timeout(&self) -> Duration {
        self.connect_timeout
    }

    pub(crate) fn request_timeout(&self) -> Duration {
        self.request_timeout
    }

    pub(crate) fn websocket_idle_timeout(&self) -> Duration {
        self.websocket_idle_timeout
    }

    pub(crate) fn websocket_ping_timeout(&self) -> Duration {
        self.websocket_ping_timeout
    }

    pub(crate) fn attachment_transfer_timeout(&self) -> Duration {
        self.attachment_transfer_timeout
    }

    pub(crate) fn db_request_timeout(&self) -> Duration {
        self.db_request_timeout
    }
}

impl Default for RuntimeTimeouts {
    fn default() -> Self {
        Self::new(
            Duration::from_secs(10),
            Duration::from_secs(30),
            Duration::from_secs(90),
            Duration::from_secs(15),
            Duration::from_secs(60 * 30),
            Duration::from_secs(30),
        )
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn runtime_timeout_default_values_are_stable() {
        let timeouts = RuntimeTimeouts::default();

        assert_eq!(timeouts.connect_timeout(), Duration::from_secs(10));
        assert_eq!(timeouts.request_timeout(), Duration::from_secs(30));
        assert_eq!(timeouts.websocket_idle_timeout(), Duration::from_secs(90));
        assert_eq!(timeouts.websocket_ping_timeout(), Duration::from_secs(15));
        assert_eq!(
            timeouts.attachment_transfer_timeout(),
            Duration::from_secs(60 * 30)
        );
        assert_eq!(timeouts.db_request_timeout(), Duration::from_secs(30));
    }

    #[test]
    fn runtime_timeouts_can_be_overridden() {
        let timeouts = RuntimeTimeouts::new(
            Duration::from_secs(1),
            Duration::from_secs(2),
            Duration::from_secs(3),
            Duration::from_secs(4),
            Duration::from_secs(5),
            Duration::from_secs(6),
        );

        assert_eq!(timeouts.connect_timeout(), Duration::from_secs(1));
        assert_eq!(timeouts.request_timeout(), Duration::from_secs(2));
        assert_eq!(timeouts.websocket_idle_timeout(), Duration::from_secs(3));
        assert_eq!(timeouts.websocket_ping_timeout(), Duration::from_secs(4));
        assert_eq!(
            timeouts.attachment_transfer_timeout(),
            Duration::from_secs(5)
        );
        assert_eq!(timeouts.db_request_timeout(), Duration::from_secs(6));
    }
}