Committing for reference

orchid-base/src/msg.rs

@@ -1,16 +1,22 @@
 use std::io;
+use std::pin::Pin;
 
+use async_std::io::{Read, ReadExt, Write, WriteExt};
 use orchid_api_traits::{Decode, Encode};
 
-pub fn send_msg(write: &mut impl io::Write, msg: &[u8]) -> io::Result<()> {
-	u32::try_from(msg.len()).unwrap().encode(write);
-	write.write_all(msg)?;
-	write.flush()
+pub async fn send_msg(mut write: Pin<&mut impl Write>, msg: &[u8]) -> io::Result<()> {
+	let mut len_buf = vec![];
+	u32::try_from(msg.len()).unwrap().encode(&mut len_buf);
+	write.write_all(&len_buf).await?;
+	write.write_all(msg).await?;
+	write.flush().await
 }
 
-pub fn recv_msg(read: &mut impl io::Read) -> io::Result<Vec<u8>> {
-	let len = u32::decode(read);
+pub async fn recv_msg(mut read: Pin<&mut impl Read>) -> io::Result<Vec<u8>> {
+	let mut len_buf = [0u8; (u32::BITS / 8) as usize];
+	read.read_exact(&mut len_buf).await?;
+	let len = u32::decode(&mut &len_buf[..]);
 	let mut msg = vec![0u8; len as usize];
-	read.read_exact(&mut msg)?;
+	read.read_exact(&mut msg).await?;
 	Ok(msg)
 }