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
use crate::call::MessageReader;
use crate::error::Result;
pub type DeserializeFn<T> = fn(MessageReader) -> Result<T>;
pub type SerializeFn<T> = fn(&T, &mut Vec<u8>);
pub struct Marshaller<T> {
pub ser: SerializeFn<T>,
pub de: DeserializeFn<T>,
}
#[cfg(feature = "protobuf-codec")]
pub mod pb_codec {
use protobuf::{CodedInputStream, Message};
use super::MessageReader;
use crate::error::Result;
#[inline]
pub fn ser<T: Message>(t: &T, buf: &mut Vec<u8>) {
t.write_to_vec(buf).unwrap()
}
#[inline]
pub fn de<T: Message>(mut reader: MessageReader) -> Result<T> {
let mut s = CodedInputStream::from_buffered_reader(&mut reader);
let mut m = T::new();
m.merge_from(&mut s)?;
Ok(m)
}
}
#[cfg(feature = "prost-codec")]
pub mod pr_codec {
use bytes::buf::BufMut;
use prost::Message;
use super::MessageReader;
use crate::error::Result;
#[inline]
pub fn ser<M: Message, B: BufMut>(msg: &M, buf: &mut B) {
msg.encode(buf).expect("Writing message to buffer failed");
}
#[inline]
pub fn de<M: Message + Default>(mut reader: MessageReader) -> Result<M> {
use bytes::buf::Buf;
reader.advance(0);
M::decode(reader).map_err(Into::into)
}
}