From f7bb1f74b38fd7632fcd5ed107c3bb8b84f817ff Mon Sep 17 00:00:00 2001
From: Derek <alexanderderekrein@gmail.com>
Date: Sat, 5 Aug 2023 09:00:04 +0200
Subject: [PATCH] feat: create logging span

---
 src/websocket_service/mod.rs | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/websocket_service/mod.rs b/src/websocket_service/mod.rs
index 2aa3b41..f1d9eae 100644
--- a/src/websocket_service/mod.rs
+++ b/src/websocket_service/mod.rs
@@ -98,21 +98,22 @@ async fn handle_msg(
     client: &Arc<relay_client::websocket::Client>,
 ) -> Result<()> {
     info!("Websocket service received message: {:?}", msg);
+    let topic = msg.topic.clone();
+    let _span = tracing::warn_span!(
+        "", topic = %topic, rpc_id = %msg.message_id,
+    );
     match msg.tag {
         4004 => {
-            let topic = msg.topic.clone();
             info!("Received push delete for topic: {}", topic);
             push_delete::handle(msg, state, client).await?;
             info!("Finished processing push delete for topic: {}", topic);
         }
         4006 => {
-            let topic = msg.topic.clone();
             info!("Received push subscribe on topic: {}", &topic);
             push_subscribe::handle(msg, state, client).await?;
             info!("Finished processing push subscribe for topic: {}", topic);
         }
         4008 => {
-            let topic = msg.topic.clone();
             info!("Received push update on topic: {}", &topic);
             push_update::handle(msg, state, client).await?;
             info!("Finished processing push update for topic: {}", topic);