之前我们分享了操作关系型数据库,具体文章,
FastAPI 学习之路(三十二)创建数据库
FastAPI 学习之路(三十三)操作数据库
FastAPI 学习之路(三十四)数据库多表操作
这次我们分享的是非关系型数据库--Redis。
首先,我们需要安装对应的依赖
pip install aioredis==1.3.1
接下来,我们去导入创建对应的链接
from starlette.requests import Request
from fastapi import FastAPI, Query
from aioredis import Redis, create_redis_pool
app = FastAPI()
async def create_redis() -> Redis:
return await create_redis_pool(f"redis://:@10.30.10.36/0?encoding=utf-8")
@app.on_event("startup")
async def startup_event():
app.state.redis = await create_redis()
print("init redis success")
@app.on_event("shutdown")
async def shutdown_event():
app.state.redis.close()
await app.state.redis.wait_closed()
print("redis closed")
这里我们就用到了上一篇中分享的startup和shutdown事件。接下来,我们去创建一个api操作redis,看是否能够生效:
@app.get("/test", summary="test redis")
async def test_redis(request: Request, num: int = Query(123, title="参数num")):
"""
等待redis写入,await异步变同步
如果不关心结果可以不用await,但是这里下一步要取值
必须得先等存完值,然后再取
"""
await request.app.state.redis.set("test", num)
# 等待读取
value = await request.app.state.redis.get("test")
print(value, type(value))
return {"value": value}
部署后请求接口
然后我们看下redis是否存储成功
可以看到,redis存储了结果,接口也正常返回了数据。这只是一个简单的demo,后续可以用来缓存,也可以用来存储token。