rtyler/meows
rtyler
/
meows
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Invoke the default handler when an unknown message type is sent 

For well formed messages which do _not_ have a known message type, the previous
behavior was to simply drop the message. This change ensures that the default
handler has a chance to execute even on well formed messages
This commit is contained in:
R Tyler Croy 2020-07-05 15:02:16 -07:00
parent 0f28c3e472
commit 12b4948355
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/lib.rs
View File

@ -349,6 +349,7 @@ impl<ServerState: 'static + Send + Sync, ClientState: 'static + Default + Send +
break;
}
let message = message.to_string();
let mut invoke_default = false;
if let Ok(envelope) = serde_json::from_str::<Envelope>(&message) {
debug!("Envelope deserialized: {:?}", envelope);
@ -359,6 +360,7 @@ impl<ServerState: 'static + Send + Sync, ClientState: 'static + Default + Send +
Some(handler.clone())
} else {
debug!("No handler found for message type: {}", envelope.ttype);
invoke_default = true;
None
}
}
@ -378,6 +380,10 @@ impl<ServerState: 'static + Send + Sync, ClientState: 'static + Default + Send +
}
}
} else {
invoke_default = true;
}
if invoke_default {
if let Some(response) = default.call(message, state.clone()).await {
channel_tx.send(response).await;
}