fix: media-gateway rtpengine missing clear timeout (#470)

* fix: media-gateway rtpengine missing clear timeout * added test
8xFF · Nov 26, 2024 · b9fc447 · b9fc447
1 parent c124a3b
commit b9fc447
Showing 1 changed file with 85 additions and 0 deletions.
diff --git a/packages/media_gateway/src/store.rs b/packages/media_gateway/src/store.rs
@@ -53,6 +53,7 @@ impl GatewayStore {
 
     pub fn on_tick(&mut self, now: u64) {
         self.webrtc.on_tick(now);
+        self.rtpengine.on_tick(now);
 
         let ping = PingEvent {
             cpu: self.node.cpu,
@@ -315,4 +316,88 @@ mod tests {
             })
         );
     }
+
+    #[test]
+    fn clear_timeout() {
+        let mut store = GatewayStore::new(ZoneId(0), Location { lat: 1.0, lon: 1.0 }, 60, 80, 90);
+        store.on_ping(
+            0,
+            1,
+            PingEvent {
+                cpu: 10,
+                memory: 20,
+                disk: 30,
+                origin: Origin::Media(MediaOrigin {}),
+                webrtc: Some(ServiceStats { live: 100, max: 1000, active: true }),
+                rtpengine: None,
+            },
+        );
+
+        store.on_ping(
+            0,
+            257,
+            PingEvent {
+                cpu: 10,
+                memory: 20,
+                disk: 30,
+                origin: Origin::Gateway(GatewayOrigin {
+                    location: Some(Location { lat: 2.0, lon: 2.0 }),
+                    zone: 1,
+                }),
+                webrtc: Some(ServiceStats { live: 100, max: 1000, active: true }),
+                rtpengine: None,
+            },
+        );
+
+        // Verify nodes are registered
+        assert_eq!(store.best_for(ServiceKind::Webrtc, None), Some(1));
+
+        // Trigger timeout
+        store.on_tick(5000); // PING_TIMEOUT is 5000
+
+        // Verify nodes are cleared
+        assert_eq!(store.best_for(ServiceKind::Webrtc, None), None);
+    }
+
+    #[test]
+    fn clear_timeout_rtpengine() {
+        let mut store = GatewayStore::new(ZoneId(0), Location { lat: 1.0, lon: 1.0 }, 60, 80, 90);
+        store.on_ping(
+            0,
+            1,
+            PingEvent {
+                cpu: 10,
+                memory: 20,
+                disk: 30,
+                origin: Origin::Media(MediaOrigin {}),
+                webrtc: None,
+                rtpengine: Some(ServiceStats { live: 100, max: 1000, active: true }),
+            },
+        );
+
+        store.on_ping(
+            0,
+            257,
+            PingEvent {
+                cpu: 10,
+                memory: 20,
+                disk: 30,
+                origin: Origin::Gateway(GatewayOrigin {
+                    location: Some(Location { lat: 2.0, lon: 2.0 }),
+                    zone: 1,
+                }),
+                webrtc: None,
+                rtpengine: Some(ServiceStats { live: 100, max: 1000, active: true }),
+            },
+        );
+
+        // Verify nodes are registered
+        assert_eq!(store.best_for(ServiceKind::RtpEngine, None), Some(1));
+
+        // Trigger timeout
+        store.on_tick(5000); // PING_TIMEOUT is 5000
+
+        // Verify nodes are cleared
+        assert_eq!(store.best_for(ServiceKind::RtpEngine, None), None);
+    }
 }