1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
#[cfg(feature = "json")]
pub mod json {
use crate::error::MessageError;
use crate::message::{Message, Action, PacketBytes};
use bytes::BytesRead;
use serde::{Serialize, de::DeserializeOwned};
pub fn encode<T, A, B>(value: T) -> Result<Message<A, B>, MessageError>
where
T: Serialize,
A: Action,
B: PacketBytes
{
let mut msg = Message::new();
serde_json::to_writer(msg.body_mut(), &value)
.map_err(MessageError::Json)?;
Ok(msg)
}
pub fn decode<A, B, T>(msg: Message<A, B>) -> Result<T, MessageError>
where
A: Action,
B: PacketBytes,
T: DeserializeOwned
{
serde_json::from_slice(msg.body().as_slice())
.map_err(MessageError::Json)
}
}
#[cfg(feature = "protobuf")]
pub mod protobuf {
use crate::error::MessageError;
use crate::message::{Message, Action, PacketBytes};
use bytes::BytesRead;
use fire_protobuf::encode::{EncodeMessage, MessageEncoder};
use fire_protobuf::decode::DecodeMessage;
pub fn encode<T, A, B>(mut value: T) -> Result<Message<A, B>, MessageError>
where
T: EncodeMessage,
A: Action,
B: PacketBytes
{
let mut msg = Message::new();
let mut encoder = MessageEncoder::new(msg.body_mut());
value.encode(None, &mut encoder)
.map_err(MessageError::EncodeError)?;
Ok(msg)
}
pub fn decode<A, B, T>(msg: Message<A, B>) -> Result<T, MessageError>
where
A: Action,
B: PacketBytes,
T: for<'a> DecodeMessage<'a>
{
T::parse_from_bytes(msg.body().as_slice())
.map_err(MessageError::DecodeError)
}
}