aldrin_core/message/
subscribe_all_events.rs

1use super::message_ops::Sealed;
2use super::{Message, MessageKind, MessageOps, OptionKind};
3use crate::ids::ServiceCookie;
4use crate::message_deserializer::{MessageDeserializeError, MessageWithoutValueDeserializer};
5use crate::message_serializer::{MessageSerializeError, MessageSerializer};
6use crate::serialized_value::SerializedValueSlice;
7use bytes::BytesMut;
8
9#[derive(Debug, Copy, Clone, PartialEq, Eq)]
10#[cfg_attr(feature = "fuzzing", derive(arbitrary::Arbitrary))]
11pub struct SubscribeAllEvents {
12    pub serial: Option<u32>,
13    pub service_cookie: ServiceCookie,
14}
15
16impl MessageOps for SubscribeAllEvents {
17    fn kind(&self) -> MessageKind {
18        MessageKind::SubscribeAllEvents
19    }
20
21    fn serialize_message(self) -> Result<BytesMut, MessageSerializeError> {
22        let mut serializer = MessageSerializer::without_value(MessageKind::SubscribeAllEvents);
23
24        match self.serial {
25            None => {
26                serializer.put_discriminant_u8(OptionKind::None);
27            }
28
29            Some(serial) => {
30                serializer.put_discriminant_u8(OptionKind::Some);
31                serializer.put_varint_u32_le(serial);
32            }
33        }
34
35        serializer.put_uuid(self.service_cookie.0);
36
37        serializer.finish()
38    }
39
40    fn deserialize_message(buf: BytesMut) -> Result<Self, MessageDeserializeError> {
41        let mut deserializer =
42            MessageWithoutValueDeserializer::new(buf, MessageKind::SubscribeAllEvents)?;
43
44        let serial = match deserializer.try_get_discriminant_u8()? {
45            OptionKind::None => None,
46            OptionKind::Some => deserializer.try_get_varint_u32_le().map(Some)?,
47        };
48
49        let service_cookie = deserializer.try_get_uuid().map(ServiceCookie)?;
50
51        deserializer.finish()?;
52        Ok(Self {
53            serial,
54            service_cookie,
55        })
56    }
57
58    fn value(&self) -> Option<&SerializedValueSlice> {
59        None
60    }
61}
62
63impl Sealed for SubscribeAllEvents {}
64
65impl From<SubscribeAllEvents> for Message {
66    fn from(msg: SubscribeAllEvents) -> Self {
67        Self::SubscribeAllEvents(msg)
68    }
69}
70
71#[cfg(test)]
72mod test {
73    use super::super::test::{assert_deserialize_eq, assert_serialize_eq};
74    use super::super::Message;
75    use super::SubscribeAllEvents;
76    use crate::ids::ServiceCookie;
77    use uuid::uuid;
78
79    #[test]
80    fn no_serial() {
81        let serialized = [
82            22, 0, 0, 0, 58, 0, 0x94, 0x5f, 0xc6, 0xe4, 0xe8, 0x9c, 0x49, 0x61, 0xb7, 0xbc, 0x4e,
83            0x0e, 0x84, 0x80, 0xdf, 0xad,
84        ];
85
86        let msg = SubscribeAllEvents {
87            serial: None,
88            service_cookie: ServiceCookie(uuid!("945fc6e4-e89c-4961-b7bc-4e0e8480dfad")),
89        };
90        assert_serialize_eq(&msg, serialized);
91        assert_deserialize_eq(&msg, serialized);
92
93        let msg = Message::SubscribeAllEvents(msg);
94        assert_serialize_eq(&msg, serialized);
95        assert_deserialize_eq(&msg, serialized);
96    }
97
98    #[test]
99    fn serial() {
100        let serialized = [
101            23, 0, 0, 0, 58, 1, 2, 0x94, 0x5f, 0xc6, 0xe4, 0xe8, 0x9c, 0x49, 0x61, 0xb7, 0xbc,
102            0x4e, 0x0e, 0x84, 0x80, 0xdf, 0xad,
103        ];
104
105        let msg = SubscribeAllEvents {
106            serial: Some(2),
107            service_cookie: ServiceCookie(uuid!("945fc6e4-e89c-4961-b7bc-4e0e8480dfad")),
108        };
109        assert_serialize_eq(&msg, serialized);
110        assert_deserialize_eq(&msg, serialized);
111
112        let msg = Message::SubscribeAllEvents(msg);
113        assert_serialize_eq(&msg, serialized);
114        assert_deserialize_eq(&msg, serialized);
115    }
116}
aldrin_core/message/subscribe_all_events.rs

aldrin_core/message/
subscribe_all_events.rs
