jiang
5966d039de
Reorganize algorithm packages by business responsibility, move orchestration into services, and keep database access behind pooled repositories. Harden analysis API validation, remove unsafe legacy simulation endpoints, and add regression and architecture boundary coverage. BREAKING CHANGE: legacy algorithm module paths and obsolete simulation endpoints are removed.
74 lines
3.3 KiB
Python
74 lines
3.3 KiB
Python
from typing import Any
|
||
from datetime import datetime
|
||
|
||
from typing import Literal
|
||
from uuid import UUID
|
||
|
||
from fastapi import APIRouter, Depends, HTTPException, Body
|
||
from pydantic import BaseModel, Field
|
||
from starlette.concurrency import run_in_threadpool
|
||
|
||
from app.auth.keycloak_dependencies import get_current_keycloak_username
|
||
from app.services.burst_location import (
|
||
run_burst_location_by_network,
|
||
)
|
||
|
||
router = APIRouter()
|
||
|
||
|
||
class BurstLocationRequest(BaseModel):
|
||
"""爆管定位请求模型"""
|
||
network: str = Field(..., description="管网名称（或数据库名称）")
|
||
data_source: Literal["monitoring", "simulation"] = Field("monitoring", description="数据来源：monitoring（监测）或simulation（模拟）")
|
||
pressure_scada_ids: list[str] | None = Field(None, description="压力SCADA传感器ID列表")
|
||
burst_pressure: dict[str, float] | list[dict[str, Any]] | None = Field(None, description="爆管时的压力数据")
|
||
normal_pressure: dict[str, float] | list[dict[str, Any]] | None = Field(None, description="正常时的压力数据")
|
||
burst_leakage: float = Field(..., description="爆管时的漏水量")
|
||
flow_scada_ids: list[str] | None = Field(None, description="流量SCADA传感器ID列表")
|
||
burst_flow: dict[str, float] | list[dict[str, Any]] | None = Field(None, description="爆管时的流量数据")
|
||
normal_flow: dict[str, float] | list[dict[str, Any]] | None = Field(None, description="正常时的流量数据")
|
||
min_dpressure: float = Field(2.0, description="最小压力差（bar）")
|
||
basic_pressure: float = Field(10.0, description="基准压力（bar）")
|
||
scada_burst_start: datetime | None = Field(None, description="爆管/模拟方案开始时间")
|
||
scada_burst_end: datetime | None = Field(None, description="爆管/模拟方案结束时间")
|
||
scada_normal_start: datetime | None = Field(None, description="监测数据正常工况开始时间")
|
||
scada_normal_end: datetime | None = Field(None, description="监测数据正常工况结束时间")
|
||
use_scada_flow: bool = Field(False, description="是否使用SCADA流量数据")
|
||
scheme_name: str | None = Field(None, description="爆管定位运行名称")
|
||
simulation_run_id: UUID | None = Field(None, description="分析模拟运行 ID")
|
||
|
||
|
||
@router.post(
|
||
"/burst-locations",
|
||
summary="执行爆管定位",
|
||
description="基于压力和流量数据定位管网中的爆管位置"
|
||
)
|
||
async def locate_burst(
|
||
data: BurstLocationRequest = Body(..., description="爆管定位请求数据"),
|
||
username: str = Depends(get_current_keycloak_username),
|
||
) -> dict[str, Any]:
|
||
"""
|
||
执行爆管定位分析。
|
||
|
||
使用压力和流量SCADA数据，通过对比爆管和正常状态下的数据差异，
|
||
定位管网中的爆管位置。
|
||
|
||
Args:
|
||
data: 包含管网名称（或数据库名称）、压力、流量数据及相关参数的请求体
|
||
username: 当前认证用户名
|
||
|
||
Returns:
|
||
包含定位结果的字典
|
||
|
||
Raises:
|
||
HTTPException: 当数据类型或值不正确时
|
||
"""
|
||
try:
|
||
return await run_in_threadpool(
|
||
run_burst_location_by_network,
|
||
**data.model_dump(),
|
||
username=username,
|
||
)
|
||
except (TypeError, ValueError) as exc:
|
||
raise HTTPException(status_code=400, detail=str(exc))
|