From 6ef2d0fbec8c05c1a92212aaf296825df66f1c30 Mon Sep 17 00:00:00 2001
From: Nathan Sobo <nathan@zed.dev>
Date: Wed, 27 Apr 2022 15:27:01 -0600
Subject: [PATCH] Trace handling of messages

---
 crates/collab/src/rpc.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/crates/collab/src/rpc.rs b/crates/collab/src/rpc.rs
index 6c4775ba6d3229d030e02273e14734ed83ed9032..8b77eba26001ae415620172283f32e9c008a852b 100644
--- a/crates/collab/src/rpc.rs
+++ b/crates/collab/src/rpc.rs
@@ -47,6 +47,7 @@ use tokio::{
     time::Sleep,
 };
 use tower::ServiceBuilder;
+use tracing::info_span;
 use util::ResultExt;
 
 type MessageHandler = Box<
@@ -156,7 +157,11 @@ impl Server {
             TypeId::of::<M>(),
             Box::new(move |server, envelope| {
                 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() {