Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Trace handling of messages

Browse source
This commit is contained in:
Nathan Sobo 2022-04-27 15:27:01 -06:00
parent 2db670308b
commit 6ef2d0fbec
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
crates/collab/src/rpc.rs
View file

@ -47,6 +47,7 @@ use tokio::{
time::Sleep, time::Sleep,
}; };
use tower::ServiceBuilder; use tower::ServiceBuilder;
use tracing::info_span;
use util::ResultExt; use util::ResultExt;
type MessageHandler = Box< type MessageHandler = Box<
@ -156,7 +157,11 @@ impl Server {
TypeId::of::<M>(), TypeId::of::<M>(),
Box::new(move |server, envelope| { Box::new(move |server, envelope| {
let envelope = envelope.into_any().downcast::<TypedEnvelope<M>>().unwrap(); let envelope = envelope.into_any().downcast::<TypedEnvelope<M>>().unwrap();
(handler)(server, *envelope).boxed() let span = info_span!(
"handle message",
payload_type = envelope.payload_type_name()
);
span.in_scope(|| (handler)(server, *envelope).boxed())
}), }),
); );
if prev_handler.is_some() { if prev_handler.is_some() {