在 FastAPI 中实现会话管理可以使用 SessionMiddleware 中间件来实现。下面是一个简单的例子:
from fastapi import FastAPI, Request from fastapi.middleware.session import SessionMiddleware from fastapi.responses import JSONResponse app = FastAPI() # 设置密钥,用于签名会话数据 SECRET_KEY = "mysecretkey" # 设置会话数据的存储方式 SESSION_COOKIE_NAME = "session" app.add_middleware(SessionMiddleware, secret_key=SECRET_KEY, cookie_name=SESSION_COOKIE_NAME) @app.get("/set_session") async def set_session(request: Request): session = request.session session["user_id"] = 1 return {"message": "Session set"} @app.get("/get_session") async def get_session(request: Request): session = request.session user_id = session.get("user_id") return JSONResponse(content={"user_id": user_id}) if __name__ == "__main__": import uvicorn uvicorn.run(app, host="127.0.0.1", port=8000) 在这个例子中,我们首先导入 SessionMiddleware 中间件,并在应用程序中添加该中间件。然后我们可以通过 request.session 来访问和修改会话数据。在 /set_session 路由中我们设置了一个会话数据,然后在 /get_session 路由中我们获取并返回了这个会话数据。
需要注意的是,在实际应用中,应该更加复杂和安全地管理会话数据,比如设置会话过期时间、限制会话存储大小等。FastAPI 的 SessionMiddleware 提供了更多功能来更好地管理会话数据,可以根据具体需求进行配置和使用。