23 lines
686 B
Rust
23 lines
686 B
Rust
use std::io;
|
|
use std::pin::Pin;
|
|
|
|
use async_std::io::{Read, ReadExt, Write, WriteExt};
|
|
use orchid_api_traits::{Decode, Encode};
|
|
|
|
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 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).await?;
|
|
Ok(msg)
|
|
}
|