fix: take first ip

jina-ai · May 23, 2023 · a30d8a3 · a30d8a3
1 parent 9c69055
commit a30d8a3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lcserve/backend/gateway.py b/lcserve/backend/gateway.py
@@ -1123,7 +1123,7 @@ async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
             if scope.get('headers'):
                 for header in scope['headers']:
                     if header[0].decode('latin-1') == 'x-forwarded-for':
-                        ip_address = header[1].decode('latin-1')
+                        ip_address = header[1].decode('latin-1').split(",")[0].strip()
                         break
 
             # Init the request/connection ID
@@ -1156,7 +1156,7 @@ async def custom_send(message: dict) -> None:
                 )
             elif scope["type"] == "websocket":
                 self.logger.info(
-                    f"WebSocket connection: {connection_id} - Path: {scope['path']} - Client IP: {ip_address} - Duration: {duration}"
+                    f"WebSocket connection: {connection_id} - Path: {scope['path']} - Client IP: {ip_address} - Duration: {duration} s"
                 )
 
         else: