From 67ea3a48fa1c9c462f1c4912c9658ad002d3336d Mon Sep 17 00:00:00 2001
From: Alex Auvolat <alex@adnab.me>
Date: Fri, 22 Jul 2022 13:40:06 +0200
Subject: Add Resp::into_parts

---
 src/message.rs | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/message.rs b/src/message.rs
index 0ac4cb8..8e4bc2f 100644
--- a/src/message.rs
+++ b/src/message.rs
@@ -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(),
-- 
cgit v1.2.3