aldrin_core/message/
subscribe_all_events.rs

1use super::message_ops::Sealed;
2use super::{
3    Message, MessageDeserializeError, MessageKind, MessageOps, MessageSerializeError,
4    MessageSerializer, MessageWithoutValueDeserializer, OptionKind,
5};
6use crate::{SerializedValue, SerializedValueSlice, ServiceCookie};
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    fn value_mut(&mut self) -> Option<&mut SerializedValue> {
63        None
64    }
65}
66
67impl Sealed for SubscribeAllEvents {}
68
69impl From<SubscribeAllEvents> for Message {
70    fn from(msg: SubscribeAllEvents) -> Self {
71        Self::SubscribeAllEvents(msg)
72    }
73}
74
75#[cfg(test)]
76mod test {
77    use super::super::test::{assert_deserialize_eq, assert_serialize_eq};
78    use super::super::Message;
79    use super::SubscribeAllEvents;
80    use crate::ServiceCookie;
81    use uuid::uuid;
82
83    #[test]
84    fn no_serial() {
85        let serialized = [
86            22, 0, 0, 0, 58, 0, 0x94, 0x5f, 0xc6, 0xe4, 0xe8, 0x9c, 0x49, 0x61, 0xb7, 0xbc, 0x4e,
87            0x0e, 0x84, 0x80, 0xdf, 0xad,
88        ];
89
90        let msg = SubscribeAllEvents {
91            serial: None,
92            service_cookie: ServiceCookie(uuid!("945fc6e4-e89c-4961-b7bc-4e0e8480dfad")),
93        };
94        assert_serialize_eq(&msg, serialized);
95        assert_deserialize_eq(&msg, serialized);
96
97        let msg = Message::SubscribeAllEvents(msg);
98        assert_serialize_eq(&msg, serialized);
99        assert_deserialize_eq(&msg, serialized);
100    }
101
102    #[test]
103    fn serial() {
104        let serialized = [
105            23, 0, 0, 0, 58, 1, 2, 0x94, 0x5f, 0xc6, 0xe4, 0xe8, 0x9c, 0x49, 0x61, 0xb7, 0xbc,
106            0x4e, 0x0e, 0x84, 0x80, 0xdf, 0xad,
107        ];
108
109        let msg = SubscribeAllEvents {
110            serial: Some(2),
111            service_cookie: ServiceCookie(uuid!("945fc6e4-e89c-4961-b7bc-4e0e8480dfad")),
112        };
113        assert_serialize_eq(&msg, serialized);
114        assert_deserialize_eq(&msg, serialized);
115
116        let msg = Message::SubscribeAllEvents(msg);
117        assert_serialize_eq(&msg, serialized);
118        assert_deserialize_eq(&msg, serialized);
119    }
120}
aldrin_core/message/subscribe_all_events.rs

aldrin_core/message/
subscribe_all_events.rs
