aldrin_core/message/
shutdown.rs1use 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}