aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorAlex Auvolat <alex@adnab.me>2022-07-22 13:32:08 +0200
committerAlex Auvolat <alex@adnab.me>2022-07-22 13:32:08 +0200
commitb9df442f035e6648e80adf8f9bf86b4943508ae5 (patch)
tree7c1c7179cb2c06f944476e6d9518ee88f37752a3 /src
parent50627c206043edf5ce8755893944a6f17a54fb85 (diff)
downloadnetapp-b9df442f035e6648e80adf8f9bf86b4943508ae5.tar.gz
netapp-b9df442f035e6648e80adf8f9bf86b4943508ae5.zip
Small optimization
Diffstat (limited to 'src')
-rw-r--r--src/endpoint.rs5
-rw-r--r--src/message.rs4
2 files changed, 8 insertions, 1 deletions
diff --git a/src/endpoint.rs b/src/endpoint.rs
index 31500aa..588f7e3 100644
--- a/src/endpoint.rs
+++ b/src/endpoint.rs
@@ -52,7 +52,10 @@ where
T: EndpointHandler<M>,
M: Message + 'static,
{
- async fn handle(self: &Arc<Self>, m: Req<M>, from: NodeID) -> Resp<M> {
+ async fn handle(self: &Arc<Self>, mut m: Req<M>, from: NodeID) -> Resp<M> {
+ // Immediately drop stream to avoid backpressure if a stream was sent
+ // (this will make all data sent to the stream be ignored immediately)
+ drop(m.take_stream());
Resp::new(EndpointHandler::handle(self, m.msg(), from).await)
}
}
diff --git a/src/message.rs b/src/message.rs
index ba06551..0ac4cb8 100644
--- a/src/message.rs
+++ b/src/message.rs
@@ -81,6 +81,10 @@ impl<M: Message> Req<M> {
}
}
+ pub fn take_stream(&mut self) -> Option<ByteStream> {
+ std::mem::replace(&mut self.body, BodyData::None).into_stream()
+ }
+
pub(crate) fn into_enc(
self,
prio: RequestPriority,