FastAPI 学习之路（五十五）操作Redis

之前我们分享了操作关系型数据库，具体文章，

   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。