Merge pull request #6920 from ant-media/getClientIpFromWS

Add clientIP to WebSocket Communication to use it in WebRTC
ant-media · Jan 13, 2025 · f946cc4 · f946cc4
2 parents dd05244 + 07aa805
commit f946cc4
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/src/main/java/io/antmedia/websocket/WebSocketCommunityHandler.java b/src/main/java/io/antmedia/websocket/WebSocketCommunityHandler.java
@@ -42,6 +42,9 @@ public class WebSocketCommunityHandler {
 
 	protected String userAgent = "N/A";
 
+	protected String clientIP = "N/A";
+
+
 	public WebSocketCommunityHandler(ApplicationContext appContext, Session session) {
 		this.appContext = appContext;
 		this.session = session;
@@ -496,4 +499,12 @@ public String getUserAgent() {
 	public void setUserAgent(String userAgent) {
 		this.userAgent = userAgent;
 	}
+
+	public String getClientIP() {
+		return clientIP;
+	}
+
+	public void setClientIP(String clientIP) {
+		this.clientIP = clientIP;
+	}
 }
diff --git a/src/test/java/io/antmedia/test/webrtc/adaptor/WebSocketCommunityHandlerTest.java b/src/test/java/io/antmedia/test/webrtc/adaptor/WebSocketCommunityHandlerTest.java
@@ -644,6 +644,15 @@ public void testUserAgent() {
 		wsHandler.setUserAgent(userAgent);
 		assertEquals(userAgent, wsHandler.getUserAgent());
 	}
+
+	@Test
+	public void testClientIP() {
+		assertEquals("N/A", wsHandler.getClientIP());
+
+		String clienTIP = "a.b.c.d";
+		wsHandler.setClientIP(clienTIP);
+		assertEquals(clienTIP, wsHandler.getClientIP());
+	}
 
 	@Test
 	public void testGetSDP() {