Small optimization

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,