FastAPI中怎么实现会话管理 - 问答

在 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 提供了更多功能来更好地管理会话数据，可以根据具体需求进行配置和使用。

0 赞

0 踩