noth/surrealpatch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add improved handling of different WebSocket message types

Browse source
This commit is contained in:
Tobie Morgan Hitchcock 2022-10-19 23:57:05 +01:00
parent f0eaf2bd19
commit 6ff1e250de
1 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
src/net/rpc.rs
View file

@ -102,14 +102,23 @@ impl Rpc {
while let Some(msg) = wrx.next().await { while let Some(msg) = wrx.next().await {
match msg { match msg {
// We've received a message from the client // We've received a message from the client
Ok(msg) => { Ok(msg) => match msg {
if msg.is_ping() { msg if msg.is_ping() => {
let _ = chn.send(Message::pong(vec![])); let _ = chn.send(Message::pong(vec![]));
} }
if msg.is_text() { msg if msg.is_text() => {
tokio::task::spawn(Rpc::call(rpc.clone(), msg, chn.clone())); tokio::task::spawn(Rpc::call(rpc.clone(), msg, chn.clone()));
} }
} msg if msg.is_close() => {
break;
}
msg if msg.is_pong() => {
continue;
}
_ => {
// Ignore everything else
}
},
// There was an error receiving the message // There was an error receiving the message
Err(err) => { Err(err) => {
// Output the WebSocket error to the logs // Output the WebSocket error to the logs