Files
fastapi-cache/examples/in_memory/main.py

53 lines
1.0 KiB
Python
Raw Normal View History

import pendulum
2020-08-26 18:04:57 +08:00
import uvicorn
from fastapi import FastAPI
from starlette.requests import Request
from starlette.responses import Response
from fastapi_cache import FastAPICache
2022-02-24 10:07:33 -06:00
from fastapi_cache.backends.inmemory import InMemoryBackend
2020-10-08 15:10:34 +08:00
from fastapi_cache.decorator import cache
2020-08-26 18:04:57 +08:00
app = FastAPI()
2020-10-08 15:10:34 +08:00
ret = 0
2020-08-26 18:04:57 +08:00
2020-10-08 15:10:34 +08:00
2020-11-03 18:08:06 +08:00
@cache(namespace="test", expire=1)
2020-10-08 15:10:34 +08:00
async def get_ret():
global ret
ret = ret + 1
return ret
2020-08-26 18:04:57 +08:00
@app.get("/")
@cache(namespace="test", expire=10)
async def index():
2020-10-08 15:10:34 +08:00
return dict(ret=await get_ret())
2020-08-26 18:04:57 +08:00
2020-11-03 18:08:06 +08:00
@app.get("/clear")
async def clear():
return await FastAPICache.clear(namespace="test")
2021-10-09 16:51:05 +08:00
@app.get("/date")
@cache(namespace="test", expire=10)
async def get_data():
return pendulum.today()
2021-10-09 16:51:05 +08:00
@app.get("/datetime")
@cache(namespace="test", expire=2)
2021-10-09 16:51:05 +08:00
async def get_datetime(request: Request, response: Response):
return {"now": pendulum.now()}
2021-10-09 16:51:05 +08:00
2020-08-26 18:04:57 +08:00
@app.on_event("startup")
async def startup():
2022-02-24 10:07:33 -06:00
FastAPICache.init(InMemoryBackend())
2020-08-26 18:04:57 +08:00
if __name__ == "__main__":
uvicorn.run("main:app", debug=True, reload=True)