30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
|
"""NL2SQL 依赖健康检查。"""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import asyncio
|
||
|
|
import time
|
||
|
|
from collections.abc import Awaitable, Callable
|
||
|
|
|
||
|
|
|
||
|
|
async def _check_one(checker: Callable[[], Awaitable[None]]) -> dict:
|
||
|
|
started = time.perf_counter()
|
||
|
|
try:
|
||
|
|
await checker()
|
||
|
|
except Exception as exc: # noqa: BLE001 对外只返回异常类型
|
||
|
|
return {
|
||
|
|
"status": f"down: {type(exc).__name__}",
|
||
|
|
"ms": round((time.perf_counter() - started) * 1000, 1),
|
||
|
|
}
|
||
|
|
return {"status": "ok", "ms": round((time.perf_counter() - started) * 1000, 1)}
|
||
|
|
|
||
|
|
|
||
|
|
async def check_nl2sql_health(checkers: dict[str, Callable[[], Awaitable[None]]]) -> dict:
|
||
|
|
"""并行检查依赖，返回不含异常详情的状态摘要。"""
|
||
|
|
names = ("mysql", "redis", "milvus", "llm")
|
||
|
|
values = await asyncio.gather(
|
||
|
|
*(_check_one(checkers[name]) for name in names if name in checkers)
|
||
|
|
)
|
||
|
|
result = {name: value for name, value in zip((name for name in names if name in checkers), values)}
|
||
|
|
result["ready"] = bool(result) and all(item["status"] == "ok" for item in result.values())
|
||
|
|
return result
|