British-Information-Technologies
/
ChatKit-server
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added auto connection closing to Connection

This commit is contained in:
michael-bailey 2022-06-14 17:56:00 +02:00
parent 3729aa3b02
commit 6b6a86168e
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
server/src/network/connection.rs
View File

@ -36,7 +36,7 @@ pub enum ConnectionMessage {
#[derive(Message)]
#[rtype(result = "()")]
pub(crate) enum ConnectionOuput {
pub enum ConnectionOuput {
RecvData(Addr<Connection>, SocketAddr, String),
ConnectionClosed(Addr<Connection>),
}
@ -98,9 +98,18 @@ impl Actor for Connection {
let mut reader = BufReader::new(read_half);
let mut buffer_string = String::new();
while let Ok(_) = reader.read_line(&mut buffer_string).await {
while let Ok(len) = reader.read_line(&mut buffer_string).await {
use SelfMessage::{UpdateObserversWithData};
use ConnectionMessage::CloseConnection;
if len == 0 {
println!("[Connection] connection closed");
addr.send(CloseConnection)
.await
.expect("[Connection] failed to send close message to self");
return
}
println!("[Connection] read line");
use SelfMessage::UpdateObserversWithData;
addr
.send(UpdateObserversWithData(buffer_string.clone()))
.await;