Skip to content

Commit 0359803

Browse files
authored
Merge pull request #10 from bybatkhuu/develop
⚡ Updated FastAPI middleware logger into async run_in_threadpool.
2 parents 83a24fb + 813b88f commit 0359803

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

beans_logging/fastapi/_middleware.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from uuid import uuid4
66

77
from fastapi import Request, Response
8+
from fastapi.concurrency import run_in_threadpool
89
from starlette.middleware.base import BaseHTTPMiddleware
910

1011
from beans_logging import logger
@@ -155,7 +156,8 @@ async def dispatch(self, request: Request, call_next) -> Response:
155156
_http_info["user_id"] = str(request.state.user_id)
156157

157158
_debug_msg = self.debug_format.format(**_http_info)
158-
_logger.debug(_debug_msg)
159+
# _logger.debug(_debug_msg)
160+
await run_in_threadpool(_logger.debug, _debug_msg)
159161

160162
_start_time = time.time()
161163
response = await call_next(request)
@@ -218,6 +220,7 @@ async def dispatch(self, request: Request, call_next) -> Response:
218220
)
219221

220222
_msg = _msg_format.format(**_http_info)
221-
_logger.bind(http_info=_http_info).log(_LEVEL, _msg)
223+
# _logger.bind(http_info=_http_info).log(_LEVEL, _msg)
224+
await run_in_threadpool(_logger.bind(http_info=_http_info).log, _LEVEL, _msg)
222225

223226
return response

0 commit comments

Comments
 (0)