mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-21 20:55:14 +00:00
46 lines
2.0 KiB
Python
46 lines
2.0 KiB
Python
import asyncio
|
|
from collections.abc import AsyncGenerator
|
|
from typing import Annotated, Any
|
|
|
|
from fastapi import APIRouter, Depends, Request
|
|
from fastapi.responses import JSONResponse
|
|
from fastapi.sse import EventSourceResponse, ServerSentEvent
|
|
|
|
from app.common.response import ResponseSchema, SuccessResponse
|
|
from app.config.setting import Settings, get_settings
|
|
from app.core.logger import logger
|
|
from app.core.router_class import OperationLogRoute
|
|
|
|
from .schema import ServiceInfoOut
|
|
from .service import HealthService
|
|
|
|
HealthRouter = APIRouter(route_class=OperationLogRoute, prefix="/health", tags=["健康检查"])
|
|
|
|
|
|
@HealthRouter.get("/check", summary="健康检查", response_model=ResponseSchema[ServiceInfoOut])
|
|
async def health_check(request: Request, settings: Annotated[Settings, Depends(get_settings)]) -> JSONResponse:
|
|
"""健康检查：实时探测 DB / Redis 网络连通状态。"""
|
|
redis: Any | None = getattr(request.app.state, "redis", None)
|
|
info = await HealthService.collect(redis)
|
|
ok = info.db_status == 1 and info.redis_status == 1
|
|
return SuccessResponse(data=info, msg=f"系统健康-{settings.VERSION}" if ok else f"服务异常-{settings.VERSION}")
|
|
|
|
|
|
|
|
@HealthRouter.get("/stream", summary="健康检查实时流(SSE)", response_class=EventSourceResponse)
|
|
async def health_stream_controller(
|
|
request: Request,
|
|
) -> AsyncGenerator[ServerSentEvent, None]:
|
|
"""健康检查实时流（SSE）：每 30s 实时探测 DB / Redis 后推送一帧。"""
|
|
redis: Any | None = getattr(request.app.state, "redis", None)
|
|
while True:
|
|
info = await HealthService.collect(redis)
|
|
# raw_data 原样发送；用 data= 会被 fastapi.sse 再 JSON 编码一次，前端需二次解析
|
|
logger.debug(
|
|
"健康检查 SSE 推送: DB={} Redis={}",
|
|
"正常" if info.db_status == 1 else "异常",
|
|
"正常" if info.redis_status == 1 else "异常",
|
|
)
|
|
yield ServerSentEvent(raw_data=info.model_dump_json())
|
|
await asyncio.sleep(30) # SSE 推送间隔（秒）
|