aldrin_core/message/
shutdown.rs

1use super::message_ops::Sealed;
2use super::{
3    Message, MessageDeserializeError, MessageKind, MessageOps, MessageSerializeError,
4    MessageSerializer, MessageWithoutValueDeserializer,
5};
6use crate::{SerializedValue, SerializedValueSlice};
7use bytes::BytesMut;
8
9#[derive(Debug, Copy, Clone, PartialEq, Eq)]
10#[cfg_attr(feature = "fuzzing", derive(arbitrary::Arbitrary))]
11pub struct Shutdown;
12
13impl MessageOps for Shutdown {
14    fn kind(&self) -> MessageKind {
15        MessageKind::Shutdown
16    }
17
18    fn serialize_message(self) -> Result<BytesMut, MessageSerializeError> {
19        MessageSerializer::without_value(MessageKind::Shutdown).finish()
20    }
21
22    fn deserialize_message(buf: BytesMut) -> Result<Self, MessageDeserializeError> {
23        MessageWithoutValueDeserializer::new(buf, MessageKind::Shutdown)?.finish()?;
24        Ok(Self)
25    }
26
27    fn value(&self) -> Option<&SerializedValueSlice> {
28        None
29    }
30
31    fn value_mut(&mut self) -> Option<&mut SerializedValue> {
32        None
33    }
34}
35
36impl Sealed for Shutdown {}
37
38impl From<Shutdown> for Message {
39    fn from(msg: Shutdown) -> Self {
40        Self::Shutdown(msg)
41    }
42}
43
44#[cfg(test)]
45mod test {
46    use super::super::test::{assert_deserialize_eq, assert_serialize_eq};
47    use super::super::Message;
48    use super::Shutdown;
49
50    #[test]
51    fn shutdown() {
52        let serialized = [5, 0, 0, 0, 2];
53
54        let msg = Shutdown;
55        assert_serialize_eq(&msg, serialized);
56        assert_deserialize_eq(&msg, serialized);
57
58        let msg = Message::Shutdown(msg);
59        assert_serialize_eq(&msg, serialized);
60        assert_deserialize_eq(&msg, serialized);
61    }
62}