YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-03-31 17:55:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

aquatic_ws: don't handle connection closed same as other errors

Browse source
This commit is contained in:
Joakim Frostegård 2020-05-23 14:17:03 +02:00
parent ae92785484
commit 572aa632b6
2 changed files with 9 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
aquatic_ws/src/lib/network/mod.rs
View file

@ -182,7 +182,12 @@ pub fn run_handshakes_and_read_messages(
},
Err(Io(err)) if err.kind() == ErrorKind::WouldBlock => {
break;
}
},
Err(tungstenite::Error::ConnectionClosed) => {
remove_connection_if_exists(connections, poll_token);
break
},
Err(err) => {
eprintln!("error reading messages: {}", err);
@ -229,8 +234,9 @@ pub fn send_out_messages(
match established_ws.ws.write_message(out_message.to_ws_message()){
Ok(()) => {},
Err(Io(err)) if err.kind() == ErrorKind::WouldBlock => {
continue;
Err(Io(err)) if err.kind() == ErrorKind::WouldBlock => {},
Err(tungstenite::Error::ConnectionClosed) => {
remove_connection_if_exists(connections, meta.poll_token);
},
Err(err) => {
eprintln!("error writing ws message: {}", err);