重构现代化 FastAPI 后端项目框架

app/infra/db/postgresql/router.py

@@ -0,0 +1,90 @@
+from fastapi import APIRouter, Depends, HTTPException, Query
+from typing import Optional
+from psycopg import AsyncConnection
+
+from .database import get_database_instance
+from .scada_info import ScadaRepository
+from .scheme import SchemeRepository
+
+router = APIRouter(prefix="/postgresql", tags=["postgresql"])
+
+
+# 创建支持数据库选择的连接依赖函数
+async def get_database_connection(
+    db_name: Optional[str] = Query(
+        None, description="指定要连接的数据库名称，为空时使用默认数据库"
+    )
+):
+    """获取数据库连接，支持通过查询参数指定数据库名称"""
+    instance = await get_database_instance(db_name)
+    async with instance.get_connection() as conn:
+        yield conn
+
+
+@router.get("/scada-info")
+async def get_scada_info_with_connection(
+    conn: AsyncConnection = Depends(get_database_connection),
+):
+    """
+    使用连接池查询所有SCADA信息
+    """
+    try:
+        # 使用ScadaRepository查询SCADA信息
+        scada_data = await ScadaRepository.get_scadas(conn)
+        return {"success": True, "data": scada_data, "count": len(scada_data)}
+    except Exception as e:
+        raise HTTPException(
+            status_code=500, detail=f"查询SCADA信息时发生错误: {str(e)}"
+        )
+
+
+@router.get("/scheme-list")
+async def get_scheme_list_with_connection(
+    conn: AsyncConnection = Depends(get_database_connection),
+):
+    """
+    使用连接池查询所有方案信息
+    """
+    try:
+        # 使用SchemeRepository查询方案信息
+        scheme_data = await SchemeRepository.get_schemes(conn)
+        return {"success": True, "data": scheme_data, "count": len(scheme_data)}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"查询方案信息时发生错误: {str(e)}")
+
+
+@router.get("/burst-locate-result")
+async def get_burst_locate_result_with_connection(
+    conn: AsyncConnection = Depends(get_database_connection),
+):
+    """
+    使用连接池查询所有爆管定位结果
+    """
+    try:
+        # 使用SchemeRepository查询爆管定位结果
+        burst_data = await SchemeRepository.get_burst_locate_results(conn)
+        return {"success": True, "data": burst_data, "count": len(burst_data)}
+    except Exception as e:
+        raise HTTPException(
+            status_code=500, detail=f"查询爆管定位结果时发生错误: {str(e)}"
+        )
+
+
+@router.get("/burst-locate-result/{burst_incident}")
+async def get_burst_locate_result_by_incident(
+    burst_incident: str,
+    conn: AsyncConnection = Depends(get_database_connection),
+):
+    """
+    根据 burst_incident 查询爆管定位结果
+    """
+    try:
+        # 使用SchemeRepository查询爆管定位结果
+        return await SchemeRepository.get_burst_locate_result_by_incident(
+            conn, burst_incident
+        )
+    except Exception as e:
+        raise HTTPException(
+            status_code=500,
+            detail=f"根据 burst_incident 查询爆管定位结果时发生错误: {str(e)}",
+        )