aldrin_core/message/
send_item.rs

1use super::message_ops::Sealed;
2use super::{Message, MessageKind, MessageOps};
3use crate::error::SerializeError;
4use crate::ids::ChannelCookie;
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 SendItem {
14    pub cookie: ChannelCookie,
15    pub value: SerializedValue,
16}
17
18impl SendItem {
19    pub fn with_serialize_value<T: Serialize + ?Sized>(
20        cookie: ChannelCookie,
21        value: &T,
22    ) -> Result<Self, SerializeError> {
23        let value = SerializedValue::serialize(value)?;
24        Ok(Self { cookie, value })
25    }
26}
27
28impl MessageOps for SendItem {
29    fn kind(&self) -> MessageKind {
30        MessageKind::SendItem
31    }
32
33    fn serialize_message(self) -> Result<BytesMut, MessageSerializeError> {
34        let mut serializer = MessageSerializer::with_value(self.value, MessageKind::SendItem)?;
35
36        serializer.put_uuid(self.cookie.0);
37
38        serializer.finish()
39    }
40
41    fn deserialize_message(buf: BytesMut) -> Result<Self, MessageDeserializeError> {
42        let mut deserializer = MessageWithValueDeserializer::new(buf, MessageKind::SendItem)?;
43
44        let cookie = deserializer.try_get_uuid().map(ChannelCookie)?;
45        let value = deserializer.finish()?;
46
47        Ok(Self { cookie, value })
48    }
49
50    fn value(&self) -> Option<&SerializedValueSlice> {
51        Some(&self.value)
52    }
53}
54
55impl Sealed for SendItem {}
56
57impl From<SendItem> for Message {
58    fn from(msg: SendItem) -> Self {
59        Self::SendItem(msg)
60    }
61}
62
63#[cfg(test)]
64mod test {
65    use super::super::test::{assert_deserialize_eq_with_value, assert_serialize_eq};
66    use super::super::Message;
67    use super::SendItem;
68    use crate::ids::ChannelCookie;
69    use uuid::uuid;
70
71    #[test]
72    fn send_item() {
73        let serialized = [
74            27, 0, 0, 0, 27, 2, 0, 0, 0, 3, 4, 0x02, 0x6c, 0x31, 0x42, 0x53, 0x0b, 0x4d, 0x65,
75            0x85, 0x0d, 0xa2, 0x97, 0xdc, 0xc2, 0xfe, 0xcb,
76        ];
77        let value = 4u8;
78
79        let msg = SendItem::with_serialize_value(
80            ChannelCookie(uuid!("026c3142-530b-4d65-850d-a297dcc2fecb")),
81            &value,
82        )
83        .unwrap();
84        assert_serialize_eq(&msg, serialized);
85        assert_deserialize_eq_with_value(&msg, serialized, &value);
86
87        let msg = Message::SendItem(msg);
88        assert_serialize_eq(&msg, serialized);
89        assert_deserialize_eq_with_value(&msg, serialized, &value);
90    }
91}
aldrin_core/message/send_item.rs

aldrin_core/message/
send_item.rs
