turn: Add relay_addr to AllocationInfo (#630)

webrtc-rs · Nov 17, 2024 · 95949fa · 95949fa
1 parent c4f4031
commit 95949fa
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/turn/src/allocation/allocation_manager.rs b/turn/src/allocation/allocation_manager.rs
@@ -63,6 +63,7 @@ impl Manager {
                     AllocationInfo::new(
                         *five_tuple,
                         alloc.username.text.clone(),
+                        alloc.relay_addr,
                         #[cfg(feature = "metrics")]
                         alloc.relayed_bytes.load(Ordering::Acquire),
                     ),

diff --git a/turn/src/allocation/mod.rs b/turn/src/allocation/mod.rs
@@ -45,6 +45,9 @@ pub struct AllocationInfo {
     /// Username of this [`Allocation`].
     pub username: String,
 
+    /// Relay address of this [`Allocation`].
+    pub relay_addr: SocketAddr,
+
     /// Relayed bytes with this [`Allocation`].
     #[cfg(feature = "metrics")]
     pub relayed_bytes: usize,
@@ -55,11 +58,13 @@ impl AllocationInfo {
     pub fn new(
         five_tuple: FiveTuple,
         username: String,
+        relay_addr: SocketAddr,
         #[cfg(feature = "metrics")] relayed_bytes: usize,
     ) -> Self {
         Self {
             five_tuple,
             username,
+            relay_addr,
             #[cfg(feature = "metrics")]
             relayed_bytes,
         }
@@ -255,6 +260,7 @@ impl Allocation {
                 .send(AllocationInfo {
                     five_tuple: self.five_tuple,
                     username: self.username.text.clone(),
+                    relay_addr: self.relay_addr,
                     #[cfg(feature = "metrics")]
                     relayed_bytes: self.relayed_bytes.load(Ordering::Acquire),
                 })