FastAPI+Redis

sudo docker run --name kes_redis -d -p 6379:6379 redis
redis-cli ping
conda activate abr

pip install redis

from fastapi import FastAPI, Depends,APIRouter
import redis.asyncio as aioredis
import uvicorn
from contextlib import asynccontextmanager

app = FastAPI()

# Redis 连接池配置
REDIS_URL = "redis://localhost:6379/0"

@asynccontextmanager
async def lifespan(app: FastAPI):
    # 初始化 Redis 客户端
    app.state.redis = await aioredis.from_url(REDIS_URL)
    yield
    # 关闭 Redis 连接
    await app.state.redis.close()

app.router.lifespan_context = lifespan

def get_redis_client():
    return app.state.redis

@app.on_event("startup")
async def startup():
    print("Application started")

@app.on_event("shutdown")
async def shutdown():
    print("Application shutdown")

router = APIRouter()

@router.get("/set/{key}/{value}")
async def set_key(key: str, value: str, redis: aioredis.Redis = Depends(get_redis_client)):
    await redis.set(key, value)
    return {"message": f"Key {key} set to {value}"}

@router.get("/get/{key}")
async def get_key(key: str, redis: aioredis.Redis = Depends(get_redis_client)):
    value = await redis.get(key)
    if value:
        return {"key": key, "value": value.decode()}
    else:
        return {"message": f"Key {key} not found"}

app.include_router(router)

if __name__ == "__main__":
    uvicorn.run("app1:app", host="0.0.0.0", port=8089, reload=True, )

curl -X GET "http://127.0.0.1:8089/set/hello/world"
curl -X GET "http://127.0.0.1:8089/get/hello"