stratum_server/types/
session_id.rs

1use std::fmt::{self, Debug, Display};
2
3#[derive(Clone, PartialEq, Eq, Hash, Copy, Default)]
4pub struct SessionID([u8; 4]);
5
6impl SessionID {
7    #[must_use]
8    pub fn as_u32(&self) -> u32 {
9        u32::from_le_bytes(self.0)
10    }
11}
12
13impl From<u32> for SessionID {
14    fn from(value: u32) -> Self {
15        SessionID(value.to_le_bytes())
16    }
17}
18
19impl Display for SessionID {
20    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
21        write!(
22            f,
23            "{:02x}{:02x}{:02x}{:02x}",
24            self.0[3], self.0[2], self.0[1], self.0[0],
25        )
26    }
27}
28
29impl Debug for SessionID {
30    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
31        write!(
32            f,
33            "{:#04x}{:02x}{:02x}{:02x} ({})",
34            self.0[3],
35            self.0[2],
36            self.0[1],
37            self.0[0],
38            self.as_u32()
39        )
40    }
41}
42
43#[cfg(test)]
44mod test {
45    use super::*;
46
47    #[test]
48    fn test_session_id_u32_conversion() {
49        let id: u32 = 280;
50
51        let session_id = SessionID::from(id);
52
53        assert_eq!(id, session_id.as_u32());
54    }
55
56    #[test]
57    fn test_session_id_display() {
58        assert_eq!(format!("{}", SessionID::from(200)), "000000c8");
59        assert_eq!(format!("{}", SessionID::from(1)), "00000001");
60        assert_eq!(format!("{}", SessionID::from(2)), "00000002");
61        assert_eq!(format!("{}", SessionID::from(256)), "00000100");
62        assert_eq!(format!("{}", SessionID::from(65536)), "00010000");
63        assert_eq!(format!("{}", SessionID::from(16_777_216)), "01000000");
64        assert_eq!(format!("{}", SessionID::from(16_777_217)), "01000001");
65    }
66
67    #[test]
68    fn test_session_id_debug() {
69        assert_eq!(format!("{:?}", SessionID::from(200)), "0x000000c8 (200)");
70        assert_eq!(format!("{:?}", SessionID::from(1)), "0x00000001 (1)");
71        assert_eq!(format!("{:?}", SessionID::from(2)), "0x00000002 (2)");
72        assert_eq!(format!("{:?}", SessionID::from(256)), "0x00000100 (256)");
73        assert_eq!(
74            format!("{:?}", SessionID::from(65536)),
75            "0x00010000 (65536)"
76        );
77        assert_eq!(
78            format!("{:?}", SessionID::from(16_777_216)),
79            "0x01000000 (16777216)"
80        );
81        assert_eq!(
82            format!("{:?}", SessionID::from(16_777_217)),
83            "0x01000001 (16777217)"
84        );
85    }
86}