fastapi 使用Redis作为session存储 单点登录
时间: 2023-09-20 13:08:40 浏览: 98
FastAPI是一个现代化的Web框架,它具有高性能和易用性。要实现单点登录,可以使用Redis作为session存储。
下面是实现单点登录的一般步骤:
1. 创建一个Redis连接池
```python
import redis
from fastapi import FastAPI
app = FastAPI()
redis_pool = redis.ConnectionPool(host='localhost', port=6379, db=0)
```
2. 定义登录和注销路由
```python
@app.post("/login")
async def login(username: str, password: str):
# 检查用户名和密码是否正确
if username == "admin" and password == "password":
session_token = uuid.uuid4().hex
redis_conn = redis.Redis(connection_pool=redis_pool)
redis_conn.set(session_token, username)
return {"session_token": session_token}
else:
return {"error": "Invalid username or password"}
@app.post("/logout")
async def logout(session_token: str):
redis_conn = redis.Redis(connection_pool=redis_pool)
redis_conn.delete(session_token)
return {"message": "Logged out successfully"}
```
3. 定义需要登录才能访问的路由
```python
@app.get("/protected")
async def protected_route(session_token: str):
redis_conn = redis.Redis(connection_pool=redis_pool)
username = redis_conn.get(session_token)
if username:
return {"message": f"Hello, {username.decode('utf-8')}!"}
else:
return {"error": "Unauthorized"}
```
在这个例子中,当用户成功登录时,将在Redis中创建一个session token,并将其返回给客户端。客户端在访问需要登录才能访问的路由时,需要将session token作为参数传递。服务器将使用Redis检查session token是否存在,并返回相应的响应。
需要注意的是,这里的实现并没有考虑到会话过期和并发访问等问题,这些问题需要根据具体情况进行处理。