74 lines
1.7 KiB
Python
74 lines
1.7 KiB
Python
import asyncio
|
|
from contextlib import asynccontextmanager
|
|
from fastapi import FastAPI
|
|
|
|
from app.loader import load_services
|
|
from app.store import all, get, history, init_db
|
|
from app.scheduler import run_service
|
|
from app.utils.schema import load_schema, render_response
|
|
from app.middleware.auth import auth_middleware
|
|
|
|
|
|
services = load_services()
|
|
TASKS = []
|
|
|
|
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
# ===== STARTUP =====
|
|
init_db()
|
|
|
|
for svc in services:
|
|
task = asyncio.create_task(run_service(svc))
|
|
TASKS.append(task)
|
|
|
|
yield # 👉 App chạy ở đây
|
|
|
|
# ===== SHUTDOWN =====
|
|
for task in TASKS:
|
|
task.cancel()
|
|
|
|
app = FastAPI(
|
|
title="Health Agent",
|
|
lifespan=lifespan
|
|
)
|
|
|
|
app.middleware("http")(auth_middleware)
|
|
|
|
@app.get("/health")
|
|
def health():
|
|
health_schema = load_schema("health_response")
|
|
|
|
services = []
|
|
for name, data in all().items():
|
|
checks = data.get("checks", [])
|
|
failed = sum(1 for c in checks if not c.get("ok"))
|
|
|
|
services.append({
|
|
"message": f"Service {name} is {data.get('status')}",
|
|
"name": name,
|
|
"status": data.get("status") == "HEALTHY",
|
|
"total_checks": len(checks),
|
|
"failed_checks": failed,
|
|
"last_updated": data.get("last_updated"),
|
|
})
|
|
|
|
return {
|
|
"code": 200,
|
|
"data": render_response(health_schema, services)
|
|
}
|
|
|
|
@app.get("/services")
|
|
def services_list():
|
|
return all()
|
|
|
|
@app.get("/services/{name}")
|
|
def service_detail(name: str):
|
|
return get(name)
|
|
|
|
@app.get("/services/{name}/history")
|
|
def service_history(name: str, limit: int = 20):
|
|
return history(name, limit)
|