starlette_context_middleware_request_local_storage_quickstart.py

1import uvicorn
2from starlette.applications import Starlette
3from starlette.responses import JSONResponse
4from starlette.routing import Route
5
6from starlette_context import context, plugins
7from starlette_context.middleware import RawContextMiddleware
8
9
10async def index(request):
11    # context is a proxy to the data stored in the middleware
12    # it can be used like a dictionary
13    return JSONResponse(context.data)
14
15
16routes = [Route("/", index)]
17
18app = Starlette(debug=True, routes=routes)
19
20# Middleware is added here. 
21# You can use built-in plugins or create your own.
22app.add_middleware(
23    RawContextMiddleware,
24    plugins=(
25        plugins.RequestIdPlugin(),
26        plugins.CorrelationIdPlugin()
27    )
28)
29
30if __name__ == "__main__":
31    uvicorn.run(app, host="0.0.0.0", port=8000)