Fix gateway

gateway/app/middleware/logging.py

@@ -5,44 +5,53 @@ Logging middleware for gateway
 import logging
 import time
 from fastapi import Request
-import json
+from starlette.middleware.base import BaseHTTPMiddleware
+from starlette.responses import Response
+import uuid
 
 logger = logging.getLogger(__name__)
 
-async def logging_middleware(request: Request, call_next):
-    """Logging middleware"""
+class LoggingMiddleware(BaseHTTPMiddleware):
+    """Logging middleware class"""
     
-    start_time = time.time()
-    
-    # Log request
-    logger.info(
-        f"Request: {request.method} {request.url.path}",
-        extra={
-            "method": request.method,
-            "url": request.url.path,
-            "query_params": str(request.query_params),
-            "client_host": request.client.host,
-            "user_agent": request.headers.get("user-agent", ""),
-            "request_id": getattr(request.state, 'request_id', None)
-        }
-    )
-    
-    # Process request
-    response = await call_next(request)
-    
-    # Calculate duration
-    duration = time.time() - start_time
-    
-    # Log response
-    logger.info(
-        f"Response: {response.status_code} in {duration:.3f}s",
-        extra={
-            "status_code": response.status_code,
-            "duration": duration,
-            "method": request.method,
-            "url": request.url.path,
-            "request_id": getattr(request.state, 'request_id', None)
-        }
-    )
-    
-    return response
+    async def dispatch(self, request: Request, call_next) -> Response:
+        """Process request with logging"""
+        
+        start_time = time.time()
+        
+        # Generate request ID
+        request_id = str(uuid.uuid4())
+        request.state.request_id = request_id
+        
+        # Log request
+        logger.info(
+            f"Request: {request.method} {request.url.path}",
+            extra={
+                "method": request.method,
+                "url": request.url.path,
+                "query_params": str(request.query_params),
+                "client_host": request.client.host if request.client else "unknown",
+                "user_agent": request.headers.get("user-agent", ""),
+                "request_id": request_id
+            }
+        )
+        
+        # Process request
+        response = await call_next(request)
+        
+        # Calculate duration
+        duration = time.time() - start_time
+        
+        # Log response
+        logger.info(
+            f"Response: {response.status_code} in {duration:.3f}s",
+            extra={
+                "status_code": response.status_code,
+                "duration": duration,
+                "method": request.method,
+                "url": request.url.path,
+                "request_id": request_id
+            }
+        )
+        
+        return response