Build lightweight AI agent admin

backend-fastapi/core/database_monitor/api.py

@@ -0,0 +1,106 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+数据库监控 API
+
+监控维度与「数据库连接管理」一致：每个启用连接（code）为一个监控目标，
+凭证通过 ConnectionResolver 解析，支持 PostgreSQL / MySQL / SQL Server / Oracle。
+"""
+import logging
+from typing import List, Optional
+
+from fastapi import APIRouter, Depends, HTTPException, Query
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.database import get_db
+from core.database_monitor.config_resolver import (
+    create_monitor_collector,
+    list_monitor_configs,
+    resolve_monitor_target,
+)
+from core.database_monitor.schema import (
+    DatabaseConfigSchema,
+    DatabaseConnectionTestSchema,
+    DatabaseOverviewSchema,
+    DatabaseRealtimeStatsSchema,
+)
+
+logger = logging.getLogger(__name__)
+
+router = APIRouter(prefix="/database_monitor", tags=["数据库监控"])
+
+
+async def get_database_configs(db: AsyncSession) -> List[dict]:
+    """获取全部可监控连接配置（供 WebSocket 等模块复用）。"""
+    return await list_monitor_configs(db)
+
+
+@router.get("/configs", response_model=List[DatabaseConfigSchema], summary="获取数据库监控目标列表")
+async def get_database_monitor_configs(db: AsyncSession = Depends(get_db)):
+    """获取已注册的数据库连接列表，作为监控目标。"""
+    configs = await list_monitor_configs(db)
+    return [
+        DatabaseConfigSchema(
+            name=config["name"],
+            db_name=config["db_name"],
+            db_type=config["db_type"],
+            host=config["host"],
+            port=config["port"],
+            database=config["database"],
+            user=config["user"],
+            has_password=config["has_password"],
+            is_system=config.get("is_system", False),
+        )
+        for config in configs
+    ]
+
+
+@router.get("/{connection_code}/overview", response_model=DatabaseOverviewSchema, summary="获取数据库概览信息")
+async def get_database_overview(
+    connection_code: str,
+    database: Optional[str] = Query(None, description="覆盖连接默认库/服务名"),
+    db: AsyncSession = Depends(get_db),
+):
+    """获取指定连接的监控概览。"""
+    try:
+        target = await resolve_monitor_target(db, connection_code, database)
+    except ValueError as exc:
+        raise HTTPException(status_code=404, detail=str(exc)) from exc
+
+    collector = create_monitor_collector(target)
+    data = await collector.get_all_info(connection_code, target["name"])
+    return DatabaseOverviewSchema(**data)
+
+
+@router.get("/{connection_code}/realtime", response_model=DatabaseRealtimeStatsSchema, summary="获取数据库实时统计")
+async def get_database_realtime_stats(
+    connection_code: str,
+    database: Optional[str] = Query(None, description="覆盖连接默认库/服务名"),
+    db: AsyncSession = Depends(get_db),
+):
+    """获取指定连接的实时统计。"""
+    try:
+        target = await resolve_monitor_target(db, connection_code, database)
+    except ValueError as exc:
+        raise HTTPException(status_code=404, detail=str(exc)) from exc
+
+    collector = create_monitor_collector(target)
+    data = await collector.get_realtime_stats(connection_code)
+    return DatabaseRealtimeStatsSchema(**data)
+
+
+@router.post("/{connection_code}/test", response_model=DatabaseConnectionTestSchema, summary="测试数据库连接")
+async def test_database_connection(
+    connection_code: str,
+    database: Optional[str] = Query(None, description="覆盖连接默认库/服务名"),
+    db: AsyncSession = Depends(get_db),
+):
+    """测试指定连接是否可用。"""
+    try:
+        target = await resolve_monitor_target(db, connection_code, database)
+    except ValueError as exc:
+        raise HTTPException(status_code=404, detail=str(exc)) from exc
+
+    collector = create_monitor_collector(target)
+    result = await collector.test_connection()
+    return DatabaseConnectionTestSchema(**result)