| 12345678910111213141516171819202122232425262728293031323334353637383940414243 |
- # -*- coding: utf-8 -*-
- # (c) Nelen & Schuurmans
- from contextvars import ContextVar
- from fastapi import Request
- __all__ = ["ctx", "RequestMiddleware"]
- class Context:
- def __init__(self):
- self._request_value: ContextVar[Request] = ContextVar("request_value")
- @property
- def request(self) -> Request:
- return self._request_value.get()
- @request.setter
- def request(self, value: Request) -> None:
- self._request_value.set(value)
- ctx = Context()
- class RequestMiddleware:
- """Save the current request in a context variable.
- We were experiencing database connections piling up until PostgreSQL's
- max_connections was hit, which has to do with BaseHTTPMiddleware not
- interacting properly with context variables. For more details, see:
- https://github.com/tiangolo/fastapi/issues/4719. Writing this
- middleware as generic ASGI middleware fixes the problem.
- """
- def __init__(self, app):
- self.app = app
- async def __call__(self, scope, receive, send):
- if scope["type"] == "http":
- ctx.request = Request(scope, receive)
- await self.app(scope, receive, send)
|
