Add Resp::into_parts
This commit is contained in:
parent
b9df442f03
commit
67ea3a48fa
1 changed files with 8 additions and 4 deletions
|
@ -63,10 +63,6 @@ pub struct Req<M: Message> {
|
|||
}
|
||||
|
||||
impl<M: Message> Req<M> {
|
||||
pub fn msg(&self) -> &M {
|
||||
&self.msg
|
||||
}
|
||||
|
||||
pub fn with_fixed_body(self, b: Bytes) -> Self {
|
||||
Self {
|
||||
body: BodyData::Fixed(b),
|
||||
|
@ -81,6 +77,10 @@ impl<M: Message> Req<M> {
|
|||
}
|
||||
}
|
||||
|
||||
pub fn msg(&self) -> &M {
|
||||
&self.msg
|
||||
}
|
||||
|
||||
pub fn take_stream(&mut self) -> Option<ByteStream> {
|
||||
std::mem::replace(&mut self.body, BodyData::None).into_stream()
|
||||
}
|
||||
|
@ -216,6 +216,10 @@ impl<M: Message> Resp<M> {
|
|||
self.msg
|
||||
}
|
||||
|
||||
pub fn into_parts(self) -> (M::Response, Option<ByteStream>) {
|
||||
(self.msg, self.body.into_stream())
|
||||
}
|
||||
|
||||
pub(crate) fn into_enc(self) -> Result<RespEnc, rmp_serde::encode::Error> {
|
||||
Ok(RespEnc::Success {
|
||||
msg: rmp_to_vec_all_named(&self.msg)?.into(),
|
||||
|
|
Loading…
Reference in a new issue