kozabrada123/chorus
kozabrada123
/
chorus
mirror of https://github.com/polyphony-chat/chorus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update Websocket to fix premature closing w 1006

This commit is contained in:
kozabrada123 2023-05-06 10:39:58 +02:00
parent 6fdff97722
commit 172c000813
1 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/gateway.rs
View File

@ -341,18 +341,25 @@ impl<'a> WebSocketConnection {
let (mut ws_tx, mut ws_rx) = ws_stream.split();
// Send messages from the send channel
while let Some(msg) = send_channel_read.recv().await {
ws_tx.send(msg).await.unwrap();
}
loop {
// Write received messages to the receive channel
while let Some(msg) = ws_rx.next().await {
receive_channel_write
.send(msg.unwrap())
.await
.unwrap();
};
// Write received messages to the receive channel
let msg = ws_rx.next().await;
if msg.as_ref().is_some() {
let msg_unwrapped = msg.unwrap().unwrap();
receive_channel_write
.send(msg_unwrapped)
.await
.unwrap();
};
// Send messages from the send channel
let msg = send_channel_read.recv().await;
if msg.as_ref().is_some() {
let msg_unwrapped = msg.unwrap();
ws_tx.send(msg_unwrapped).await.unwrap();
}
}
});
WebSocketConnection {