lx/netapp
1
1
Fork 5
Code Issues 2 Pull requests 1 Releases 1 Wiki Activity

add streaming body to requests and responses #3

Merged
lx merged 64 commits from stream-body into main 2022-09-13 10:56:54 +00:00
Conversation 0 Commits 64 Files changed 22 +8 -4
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files
Showing only changes of commit 67ea3a48fa - Show all commits

12
src/message.rs
View file

@ -63,10 +63,6 @@ pub struct Req<M: Message> {
} }
impl<M: Message> Req<M> { impl<M: Message> Req<M> {
pub fn msg(&self) -> &M {
&self.msg
}
pub fn with_fixed_body(self, b: Bytes) -> Self { pub fn with_fixed_body(self, b: Bytes) -> Self {
Self { Self {
body: BodyData::Fixed(b), 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> { pub fn take_stream(&mut self) -> Option<ByteStream> {
std::mem::replace(&mut self.body, BodyData::None).into_stream() std::mem::replace(&mut self.body, BodyData::None).into_stream()
} }
@ -216,6 +216,10 @@ impl<M: Message> Resp<M> {
self.msg 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> { pub(crate) fn into_enc(self) -> Result<RespEnc, rmp_serde::encode::Error> {
Ok(RespEnc::Success { Ok(RespEnc::Success {
msg: rmp_to_vec_all_named(&self.msg)?.into(), msg: rmp_to_vec_all_named(&self.msg)?.into(),