zed-industries/zed
1
1
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2024-09-21 03:18:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

Remove peer message listeners when their receiver is dropped

Browse Source
This commit is contained in:
Max Brunsfeld 2021-06-16 17:02:40 -07:00
parent 9de4d73ffb
commit fb736d5e78
1 changed files with 17 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
zed-rpc/src/peer.rs
View File

@ -32,8 +32,9 @@ struct Connection {
_close_barrier: barrier::Sender,
}
type MessageHandler =
Box<dyn Send + Sync + Fn(&mut Option<proto::Envelope>, ConnectionId) -> Option<BoxFuture<()>>>;
type MessageHandler = Box<
dyn Send + Sync + Fn(&mut Option<proto::Envelope>, ConnectionId) -> Option<BoxFuture<bool>>,
>;
pub struct TypedEnvelope<T> {
id: u32,
@ -90,7 +91,8 @@ impl Peer {
connection_id,
payload: T::from_envelope(envelope).unwrap(),
})
.await;
.await
.is_err()
}
.boxed(),
)
@ -152,17 +154,24 @@ impl Peer {
);
}
} else {
let mut handled = false;
let mut envelope = Some(incoming);
for handler in this.message_handlers.read().await.iter() {
let mut handler_index = None;
let mut handler_was_dropped = false;
for (i, handler) in
this.message_handlers.read().await.iter().enumerate()
{
if let Some(future) = handler(&mut envelope, connection_id) {
future.await;
handled = true;
handler_was_dropped = future.await;
handler_index = Some(i);
break;
}
}
if !handled {
if let Some(handler_index) = handler_index {
if handler_was_dropped {
drop(this.message_handlers.write().await.remove(handler_index));
}
} else {
log::warn!("unhandled message: {:?}", envelope.unwrap().payload);
}
}