add auth and redis endpoints

This commit is contained in:
Jakub Miazek 2023-07-24 15:07:29 +02:00
parent 0b2d2e7933
commit 32346cd7e7
5 changed files with 53 additions and 12 deletions

16
app/api/health.py Normal file
View File

@ -0,0 +1,16 @@
import logging
from fastapi import APIRouter, status, Request
router = APIRouter()
@router.get("/redis", status_code=status.HTTP_200_OK)
async def redis_check(request: Request):
_redis = await request.app.state.redis
_info = None
try:
_info = await _redis.info()
except Exception as e:
logging.error(f"Redis error: {e}")
return _info

View File

@ -1,24 +1,35 @@
from fastapi import FastAPI
from contextlib import asynccontextmanager
from fastapi import FastAPI, Depends
from app.api.nonsense import router as nonsense_router
from app.api.shakespeare import router as shakespeare_router
from app.api.stuff import router as stuff_router
from app.utils.logging import AppLogger
from app.api.user import router as user_router
from app.api.health import router as health_router
from app.redis import get_redis
from app.services.auth import AuthBearer
logger = AppLogger.__call__().get_logger()
app = FastAPI(title="Stuff And Nonsense API", version="0.5")
@asynccontextmanager
async def lifespan(app: FastAPI):
# Load the redis connection
app.state.redis = await get_redis()
yield
# close redis connection and release the resources
app.state.redis.close()
app = FastAPI(title="Stuff And Nonsense API", version="0.6", lifespan=lifespan)
app.include_router(stuff_router)
app.include_router(nonsense_router)
app.include_router(shakespeare_router)
app.include_router(user_router)
@app.on_event("startup")
async def startup_event():
logger.info("Starting up...")
@app.on_event("shutdown")
async def shutdown_event():
logger.info("Shutting down...")
app.include_router(health_router, prefix="/v1/public/health", tags=["Health, Public"])
app.include_router(health_router, prefix="/v1/health", tags=["Health, Bearer"], dependencies=[Depends(AuthBearer())])

14
app/redis.py Normal file
View File

@ -0,0 +1,14 @@
import redis.asyncio as redis
from app import config
global_settings = config.get_settings()
async def get_redis():
return await redis.from_url(
global_settings.redis_url.unicode_string(),
encoding="utf-8",
decode_responses=True,
)

View File

@ -38,7 +38,7 @@ async def create_access_token(user: User, request: Request):
_payload = {
"email": user.email,
"expiry": time.time() + global_settings.jwt_expire,
"platform": request.headers.get("User-Agent")
"platform": request.headers.get("User-Agent"),
}
_token = jwt.encode(_payload, str(user.password), algorithm=global_settings.jwt_algorithm)

View File

@ -1,6 +1,6 @@
[tool.poetry]
name = "fastapi-sqlalchemy-asyncpg"
version = "0.0.5"
version = "0.0.6"
description = ""
authors = ["Jakub Miazek <the@grillazz.com>"]
packages = []