1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
use super::Transport;
use crate::encoding::Message;
use anyhow::Result;
use log::debug;
use std::str::FromStr;
use tokio::{
io::{AsyncBufReadExt, AsyncReadExt, AsyncWriteExt, BufReader, BufWriter},
net::{
tcp::{OwnedReadHalf, OwnedWriteHalf},
TcpStream,
},
sync::Mutex,
};
pub struct TcpTransport {
write: Mutex<BufWriter<OwnedWriteHalf>>,
read: Mutex<BufReader<OwnedReadHalf>>,
}
impl TcpTransport {
pub async fn new(stream: TcpStream) -> Result<Self> {
let (read, write) = stream.into_split();
Ok(Self {
write: BufWriter::new(write).into(),
read: BufReader::new(read).into(),
})
}
}
impl Transport for TcpTransport {
async fn recv(&self) -> Result<Message> {
let mut g = self.read.lock().await;
let mut message = String::new();
while !message.ends_with("\r\n\r\n") {
g.read_line(&mut message).await?;
}
let mut mesg = Message::from_str(message.trim())?;
debug!("<- {mesg}");
let cl = mesg
.content_length()
.transpose()?
.map(|e| e.0)
.unwrap_or_default();
let mut body = vec![0u8; cl];
g.read_exact(&mut body).await?;
*mesg.body_mut() = String::from_utf8_lossy_owned(body);
Ok(mesg)
}
async fn send(&self, request: Message) -> Result<()> {
debug!("-> {request}");
let mut g = self.write.lock().await;
g.write_all(format!("{request}").as_bytes()).await?;
g.flush().await?;
Ok(())
}
}
|
