Skip to content

Commit f7a12a2

Browse files
committed
test headers
1 parent 6e1b776 commit f7a12a2

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

api/src/middleware/request_context_middleware.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from starlette.types import ASGIApp, Receive, Scope, Send
44

55
from middleware.request_context import RequestContext, _request_context
6-
from utils.logger import HttpRequest, API_ACCESS_LOG
6+
from utils.logger import HttpRequest, API_ACCESS_LOG, get_logger
77

88

99
class RequestContextMiddleware:
@@ -12,7 +12,7 @@ class RequestContextMiddleware:
1212
"""
1313

1414
def __init__(self, app: ASGIApp) -> None:
15-
self.logger = logging.getLogger(API_ACCESS_LOG)
15+
self.logger = get_logger(API_ACCESS_LOG)
1616
self.app = app
1717

1818
@staticmethod
@@ -63,8 +63,13 @@ def log_api_access(
6363
"""
6464
latency = time.time() - start_time
6565
request = self.create_http_request(scope, request_context, status_code, content_length, latency)
66+
headers = {
67+
k.decode().lower(): v.decode()
68+
for k, v in scope.get("headers", [])
69+
}
6670
self.logger.info(
67-
"API Access Log",
71+
{ "user_id": request_context.user_id if request_context.user_id else "NO_AUTH",
72+
"headers": headers },
6873
extra={
6974
"context": {
7075
"http_request": request,

0 commit comments

Comments
 (0)