Add utility to write packet to byte vec

Signed-off-by: Matthias Beyer <mail@beyermatthias.de>

2 files changed, 23 insertions, 0 deletions

diff --git a/mqtt-tester/src/main.rs b/mqtt-tester/src/main.rs
index aec2d08..9562931 100644
--- a/mqtt-tester/src/main.rs
+++ b/mqtt-tester/src/main.rs
@@ -7,6 +7,7 @@
 mod client_report;
 mod command;
 mod report;
+mod util;
 
 use std::{path::PathBuf, process::exit};
 
diff --git a/mqtt-tester/src/util.rs b/mqtt-tester/src/util.rs
new file mode 100644
index 0000000..b4e19ec
--- /dev/null
+++ b/mqtt-tester/src/util.rs
@@ -0,0 +1,22 @@
+//
+//   This Source Code Form is subject to the terms of the Mozilla Public
+//   License, v. 2.0. If a copy of the MPL was not distributed with this
+//   file, You can obtain one at http://mozilla.org/MPL/2.0/.
+//
+
+use std::pin::Pin;
+
+use miette::IntoDiagnostic;
+use mqtt_format::v3::packet::MPacket;
+
+pub async fn packet_to_vec<'m>(mpacket: MPacket<'m>) -> miette::Result<Vec<u8>> {
+    let mut buf = vec![];
+    {
+        let mut cursor = futures::io::Cursor::new(&mut buf);
+        mpacket
+            .write_to(Pin::new(&mut cursor))
+            .await
+            .into_diagnostic()?;
+    }
+    Ok(buf)
+}