log stats on wgpu_room

livekit · Dec 8, 2023 · 5ce2ef3 · 5ce2ef3
1 parent 8bb0ed6
commit 5ce2ef3
Show file tree

Hide file tree

Showing 4 changed files with 43 additions and 2 deletions.
diff --git a/examples/wgpu_room/src/app.rs b/examples/wgpu_room/src/app.rs
@@ -162,8 +162,8 @@ impl LkApp {
             });
 
             ui.menu_button("Debug", |ui| {
-                if ui.button("Refresh stats").clicked() {
-                    // TODO
+                if ui.button("Log stats").clicked() {
+                    let _ = self.service.send(AsyncCmd::LogStats);
                 }
             });
         });

diff --git a/examples/wgpu_room/src/service.rs b/examples/wgpu_room/src/service.rs
@@ -33,6 +33,7 @@ pub enum AsyncCmd {
         publication: RemoteTrackPublication,
     },
     E2eeKeyRatchet,
+    LogStats,
 }
 
 #[derive(Debug)]
@@ -199,6 +200,32 @@ async fn service_task(inner: Arc<ServiceInner>, mut cmd_rx: mpsc::UnboundedRecei
                     }
                 }
             }
+            AsyncCmd::LogStats => {
+                if let Some(state) = running_state.as_ref() {
+                    for (_, publication) in state.room.local_participant().tracks() {
+                        if let Some(track) = publication.track() {
+                            log::info!(
+                                "track stats: LOCAL {:?} {:?}",
+                                track.sid(),
+                                track.get_stats().await,
+                            );
+                        }
+                    }
+
+                    for (_, participant) in state.room.participants() {
+                        for (_, publication) in participant.tracks() {
+                            if let Some(track) = publication.track() {
+                                log::info!(
+                                    "track stats: {:?} {:?} {:?}",
+                                    participant.identity(),
+                                    track.sid(),
+                                    track.get_stats().await,
+                                );
+                            }
+                        }
+                    }
+                }
+            }
         }
     }
 }

diff --git a/livekit/src/room/track/local_track.rs b/livekit/src/room/track/local_track.rs
@@ -42,6 +42,13 @@ impl LocalTrack {
             Self::Video(track) => track.rtc_track().into(),
         }
     }
+
+    pub async fn get_stats(&self) -> RoomResult<Vec<RtcStats>> {
+        match self {
+            Self::Audio(track) => track.get_stats().await,
+            Self::Video(track) => track.get_stats().await,
+        }
+    }
 }
 
 impl From<LocalTrack> for Track {

diff --git a/livekit/src/room/track/remote_track.rs b/livekit/src/room/track/remote_track.rs
@@ -37,6 +37,13 @@ impl RemoteTrack {
             Self::Video(track) => track.rtc_track().into(),
         }
     }
+
+    pub async fn get_stats(&self) -> RoomResult<Vec<RtcStats>> {
+        match self {
+            Self::Audio(track) => track.get_stats().await,
+            Self::Video(track) => track.get_stats().await,
+        }
+    }
 }
 
 pub(super) async fn get_stats(inner: &Arc<TrackInner>) -> RoomResult<Vec<RtcStats>> {