56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
"""Agent 上报接收路由（数据写入 metrics.db）"""
|
||
|
||
from datetime import datetime
|
||
|
||
from fastapi import APIRouter, Depends
|
||
from sqlmodel import Session, select
|
||
|
||
from app.core.security import require_agent_key
|
||
from app.database import get_metrics_session
|
||
from app.models.monitor import EventLog, MetricPoint, SecurityCheck, ServerInfo
|
||
from app.schemas.agent import AgentReport
|
||
|
||
router = APIRouter(prefix="/api/agent", tags=["agent"])
|
||
|
||
|
||
@router.post(
|
||
"/report",
|
||
summary="Agent 上报资源/信息/安全状态",
|
||
dependencies=[Depends(require_agent_key)],
|
||
)
|
||
def report(data: AgentReport, session: Session = Depends(get_metrics_session)) -> dict:
|
||
# 资源监控时序
|
||
if data.metrics:
|
||
session.add(MetricPoint(asset_id=data.asset_id, **data.metrics.model_dump()))
|
||
|
||
# 服务器信息快照（每资产一条，覆盖更新）
|
||
if data.server_info:
|
||
info = data.server_info.model_dump()
|
||
info["last_seen"] = datetime.utcnow()
|
||
existing = session.exec(
|
||
select(ServerInfo).where(ServerInfo.asset_id == data.asset_id)
|
||
).first()
|
||
if existing:
|
||
for key, value in info.items():
|
||
setattr(existing, key, value)
|
||
session.add(existing)
|
||
else:
|
||
session.add(ServerInfo(asset_id=data.asset_id, **info))
|
||
|
||
# 安全检查项
|
||
if data.security:
|
||
for check in data.security:
|
||
session.add(SecurityCheck(asset_id=data.asset_id, **check.model_dump()))
|
||
|
||
# 事件日志
|
||
session.add(
|
||
EventLog(
|
||
level="info",
|
||
source="agent",
|
||
asset_id=data.asset_id,
|
||
message=f"Agent 上报 asset_id={data.asset_id}",
|
||
)
|
||
)
|
||
session.commit()
|
||
return {"status": "ok", "asset_id": data.asset_id}
|