aldrin_core/message/
emit_event.rs

1use super::message_ops::Sealed;
2use super::{Message, MessageKind, MessageOps};
3use crate::error::SerializeError;
4use crate::ids::ServiceCookie;
5use crate::message_deserializer::{MessageDeserializeError, MessageWithValueDeserializer};
6use crate::message_serializer::{MessageSerializeError, MessageSerializer};
7use crate::serialized_value::{SerializedValue, SerializedValueSlice};
8use crate::value_serializer::Serialize;
9use bytes::BytesMut;
10
11#[derive(Debug, Clone, PartialEq, Eq)]
12#[cfg_attr(feature = "fuzzing", derive(arbitrary::Arbitrary))]
13pub struct EmitEvent {
14    pub service_cookie: ServiceCookie,
15    pub event: u32,
16    pub value: SerializedValue,
17}
18
19impl EmitEvent {
20    pub fn with_serialize_value<T: Serialize + ?Sized>(
21        service_cookie: ServiceCookie,
22        event: u32,
23        value: &T,
24    ) -> Result<Self, SerializeError> {
25        let value = SerializedValue::serialize(value)?;
26        Ok(Self {
27            service_cookie,
28            event,
29            value,
30        })
31    }
32}
33
34impl MessageOps for EmitEvent {
35    fn kind(&self) -> MessageKind {
36        MessageKind::EmitEvent
37    }
38
39    fn serialize_message(self) -> Result<BytesMut, MessageSerializeError> {
40        let mut serializer = MessageSerializer::with_value(self.value, MessageKind::EmitEvent)?;
41
42        serializer.put_uuid(self.service_cookie.0);
43        serializer.put_varint_u32_le(self.event);
44
45        serializer.finish()
46    }
47
48    fn deserialize_message(buf: BytesMut) -> Result<Self, MessageDeserializeError> {
49        let mut deserializer = MessageWithValueDeserializer::new(buf, MessageKind::EmitEvent)?;
50
51        let service_cookie = deserializer.try_get_uuid().map(ServiceCookie)?;
52        let event = deserializer.try_get_varint_u32_le()?;
53        let value = deserializer.finish()?;
54
55        Ok(Self {
56            service_cookie,
57            event,
58            value,
59        })
60    }
61
62    fn value(&self) -> Option<&SerializedValueSlice> {
63        Some(&self.value)
64    }
65}
66
67impl Sealed for EmitEvent {}
68
69impl From<EmitEvent> for Message {
70    fn from(msg: EmitEvent) -> Self {
71        Self::EmitEvent(msg)
72    }
73}
74
75#[cfg(test)]
76mod test {
77    use super::super::test::{assert_deserialize_eq_with_value, assert_serialize_eq};
78    use super::super::Message;
79    use super::EmitEvent;
80    use crate::ids::ServiceCookie;
81    use uuid::uuid;
82
83    #[test]
84    fn emit_event() {
85        let serialized = [
86            28, 0, 0, 0, 16, 2, 0, 0, 0, 3, 4, 0x02, 0x6c, 0x31, 0x42, 0x53, 0x0b, 0x4d, 0x65,
87            0x85, 0x0d, 0xa2, 0x97, 0xdc, 0xc2, 0xfe, 0xcb, 1,
88        ];
89        let value = 4u8;
90
91        let msg = EmitEvent::with_serialize_value(
92            ServiceCookie(uuid!("026c3142-530b-4d65-850d-a297dcc2fecb")),
93            1,
94            &value,
95        )
96        .unwrap();
97        assert_serialize_eq(&msg, serialized);
98        assert_deserialize_eq_with_value(&msg, serialized, &value);
99
100        let msg = Message::EmitEvent(msg);
101        assert_serialize_eq(&msg, serialized);
102        assert_deserialize_eq_with_value(&msg, serialized, &value);
103    }
104}
aldrin_core/message/emit_event.rs

aldrin_core/message/
emit_event.rs
