471 lines
14 KiB
Python
471 lines
14 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, Query
|
||
from typing import List, Optional
|
||
from datetime import datetime
|
||
from psycopg import AsyncConnection
|
||
|
||
from .database import get_database_instance
|
||
from .schemas.realtime import RealtimeRepository
|
||
from .schemas.scheme import SchemeRepository
|
||
from .schemas.scada import ScadaRepository
|
||
from .composite_queries import CompositeQueries
|
||
from postgresql.database import get_database_instance as get_postgres_database_instance
|
||
|
||
router = APIRouter(prefix="/timescaledb", tags=["TimescaleDB"])
|
||
|
||
|
||
# 创建支持数据库选择的连接依赖函数
|
||
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
|
||
|
||
|
||
# PostgreSQL 数据库连接依赖函数
|
||
async def get_postgres_connection(
|
||
db_name: Optional[str] = Query(
|
||
None, description="指定要连接的 PostgreSQL 数据库名称，为空时使用默认数据库"
|
||
)
|
||
):
|
||
"""获取 PostgreSQL 数据库连接，支持通过查询参数指定数据库名称"""
|
||
instance = await get_postgres_database_instance(db_name)
|
||
async with instance.get_connection() as conn:
|
||
yield conn
|
||
|
||
|
||
# --- Realtime Endpoints ---
|
||
|
||
|
||
@router.post("/realtime/links/batch", status_code=201)
|
||
async def insert_realtime_links(
|
||
data: List[dict], conn: AsyncConnection = Depends(get_database_connection)
|
||
):
|
||
await RealtimeRepository.insert_links_batch(conn, data)
|
||
return {"message": f"Inserted {len(data)} records"}
|
||
|
||
|
||
@router.get("/realtime/links")
|
||
async def get_realtime_links(
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
return await RealtimeRepository.get_links_by_time_range(conn, start_time, end_time)
|
||
|
||
|
||
@router.delete("/realtime/links")
|
||
async def delete_realtime_links(
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
await RealtimeRepository.delete_links_by_time_range(conn, start_time, end_time)
|
||
return {"message": "Deleted successfully"}
|
||
|
||
|
||
@router.patch("/realtime/links/{link_id}/field")
|
||
async def update_realtime_link_field(
|
||
link_id: str,
|
||
time: datetime,
|
||
field: str,
|
||
value: float, # Assuming float for now, could be Any but FastAPI needs type
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
try:
|
||
await RealtimeRepository.update_link_field(conn, time, link_id, field, value)
|
||
return {"message": "Updated successfully"}
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
|
||
|
||
@router.post("/realtime/nodes/batch", status_code=201)
|
||
async def insert_realtime_nodes(
|
||
data: List[dict], conn: AsyncConnection = Depends(get_database_connection)
|
||
):
|
||
await RealtimeRepository.insert_nodes_batch(conn, data)
|
||
return {"message": f"Inserted {len(data)} records"}
|
||
|
||
|
||
@router.get("/realtime/nodes")
|
||
async def get_realtime_nodes(
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
return await RealtimeRepository.get_nodes_by_time_range(conn, start_time, end_time)
|
||
|
||
|
||
@router.delete("/realtime/nodes")
|
||
async def delete_realtime_nodes(
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
await RealtimeRepository.delete_nodes_by_time_range(conn, start_time, end_time)
|
||
return {"message": "Deleted successfully"}
|
||
|
||
|
||
@router.post("/realtime/simulation/store", status_code=201)
|
||
async def store_realtime_simulation_result(
|
||
node_result_list: List[dict],
|
||
link_result_list: List[dict],
|
||
result_start_time: str,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
"""Store realtime simulation results to TimescaleDB"""
|
||
await RealtimeRepository.store_realtime_simulation_result(
|
||
conn, node_result_list, link_result_list, result_start_time
|
||
)
|
||
return {"message": "Simulation results stored successfully"}
|
||
|
||
|
||
@router.get("/realtime/query/by-time-property")
|
||
async def query_realtime_records_by_time_property(
|
||
query_time: str,
|
||
type: str,
|
||
property: str,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
"""Query all realtime records by time and property"""
|
||
try:
|
||
return await RealtimeRepository.query_all_record_by_time_property(
|
||
conn, query_time, type, property
|
||
)
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
|
||
|
||
@router.get("/realtime/query/by-id-time")
|
||
async def query_realtime_simulation_by_id_time(
|
||
ID: str,
|
||
type: str,
|
||
query_time: str,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
"""Query realtime simulation results by ID and time"""
|
||
try:
|
||
return await RealtimeRepository.query_simulation_result_by_ID_time(
|
||
conn, ID, type, query_time
|
||
)
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
|
||
|
||
# --- Scheme Endpoints ---
|
||
|
||
|
||
@router.post("/scheme/links/batch", status_code=201)
|
||
async def insert_scheme_links(
|
||
data: List[dict], conn: AsyncConnection = Depends(get_database_connection)
|
||
):
|
||
await SchemeRepository.insert_links_batch(conn, data)
|
||
return {"message": f"Inserted {len(data)} records"}
|
||
|
||
|
||
@router.get("/scheme/links")
|
||
async def get_scheme_links(
|
||
scheme_type: str,
|
||
scheme_name: str,
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
return await SchemeRepository.get_links_by_scheme_and_time_range(
|
||
conn, scheme_type, scheme_name, start_time, end_time
|
||
)
|
||
|
||
|
||
@router.get("/scheme/links/{link_id}/field")
|
||
async def get_scheme_link_field(
|
||
scheme_type: str,
|
||
scheme_name: str,
|
||
link_id: str,
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
field: str,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
try:
|
||
return await SchemeRepository.get_link_field_by_scheme_and_time_range(
|
||
conn, scheme_type, scheme_name, start_time, end_time, link_id, field
|
||
)
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
|
||
|
||
@router.patch("/scheme/links/{link_id}/field")
|
||
async def update_scheme_link_field(
|
||
scheme_type: str,
|
||
scheme_name: str,
|
||
link_id: str,
|
||
time: datetime,
|
||
field: str,
|
||
value: float,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
try:
|
||
await SchemeRepository.update_link_field(
|
||
conn, time, scheme_type, scheme_name, link_id, field, value
|
||
)
|
||
return {"message": "Updated successfully"}
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
|
||
|
||
@router.delete("/scheme/links")
|
||
async def delete_scheme_links(
|
||
scheme_type: str,
|
||
scheme_name: str,
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
await SchemeRepository.delete_links_by_scheme_and_time_range(
|
||
conn, scheme_type, scheme_name, start_time, end_time
|
||
)
|
||
return {"message": "Deleted successfully"}
|
||
|
||
|
||
@router.post("/scheme/nodes/batch", status_code=201)
|
||
async def insert_scheme_nodes(
|
||
data: List[dict], conn: AsyncConnection = Depends(get_database_connection)
|
||
):
|
||
await SchemeRepository.insert_nodes_batch(conn, data)
|
||
return {"message": f"Inserted {len(data)} records"}
|
||
|
||
|
||
@router.get("/scheme/nodes/{node_id}/field")
|
||
async def get_scheme_node_field(
|
||
scheme_type: str,
|
||
scheme_name: str,
|
||
node_id: str,
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
field: str,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
try:
|
||
return await SchemeRepository.get_node_field_by_scheme_and_time_range(
|
||
conn, scheme_type, scheme_name, start_time, end_time, node_id, field
|
||
)
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
|
||
|
||
@router.patch("/scheme/nodes/{node_id}/field")
|
||
async def update_scheme_node_field(
|
||
scheme_type: str,
|
||
scheme_name: str,
|
||
node_id: str,
|
||
time: datetime,
|
||
field: str,
|
||
value: float,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
try:
|
||
await SchemeRepository.update_node_field(
|
||
conn, time, scheme_type, scheme_name, node_id, field, value
|
||
)
|
||
return {"message": "Updated successfully"}
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
|
||
|
||
@router.delete("/scheme/nodes")
|
||
async def delete_scheme_nodes(
|
||
scheme_type: str,
|
||
scheme_name: str,
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
await SchemeRepository.delete_nodes_by_scheme_and_time_range(
|
||
conn, scheme_type, scheme_name, start_time, end_time
|
||
)
|
||
return {"message": "Deleted successfully"}
|
||
|
||
|
||
@router.post("/scheme/simulation/store", status_code=201)
|
||
async def store_scheme_simulation_result(
|
||
scheme_type: str,
|
||
scheme_name: str,
|
||
node_result_list: List[dict],
|
||
link_result_list: List[dict],
|
||
result_start_time: str,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
"""Store scheme simulation results to TimescaleDB"""
|
||
await SchemeRepository.store_scheme_simulation_result(
|
||
conn,
|
||
scheme_type,
|
||
scheme_name,
|
||
node_result_list,
|
||
link_result_list,
|
||
result_start_time,
|
||
)
|
||
return {"message": "Scheme simulation results stored successfully"}
|
||
|
||
|
||
@router.get("/scheme/query/by-scheme-time-property")
|
||
async def query_scheme_records_by_scheme_time_property(
|
||
scheme_type: str,
|
||
scheme_name: str,
|
||
query_time: str,
|
||
type: str,
|
||
property: str,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
"""Query all scheme records by scheme, time and property"""
|
||
try:
|
||
return await SchemeRepository.query_all_record_by_scheme_time_property(
|
||
conn, scheme_type, scheme_name, query_time, type, property
|
||
)
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
|
||
|
||
@router.get("/scheme/query/by-id-time")
|
||
async def query_scheme_simulation_by_id_time(
|
||
scheme_type: str,
|
||
scheme_name: str,
|
||
ID: str,
|
||
type: str,
|
||
query_time: str,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
"""Query scheme simulation results by ID and time"""
|
||
try:
|
||
return await SchemeRepository.query_scheme_simulation_result_by_ID_time(
|
||
conn, scheme_type, scheme_name, ID, type, query_time
|
||
)
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
|
||
|
||
# --- SCADA Endpoints ---
|
||
|
||
|
||
@router.post("/scada/batch", status_code=201)
|
||
async def insert_scada_data(
|
||
data: List[dict], conn: AsyncConnection = Depends(get_database_connection)
|
||
):
|
||
await ScadaRepository.insert_scada_batch(conn, data)
|
||
return {"message": f"Inserted {len(data)} records"}
|
||
|
||
|
||
@router.get("/scada")
|
||
async def get_scada_by_id_time_range(
|
||
device_id: str,
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
return await ScadaRepository.get_scada_by_id_time_range(
|
||
conn, device_id, start_time, end_time
|
||
)
|
||
|
||
|
||
@router.get("/scada/{device_id}/field")
|
||
async def get_scada_field_by_id_time_range(
|
||
device_id: str,
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
field: str,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
try:
|
||
return await ScadaRepository.get_scada_field_by_id_time_range(
|
||
conn, device_id, start_time, end_time, field
|
||
)
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
|
||
|
||
@router.patch("/scada/{device_id}/field")
|
||
async def update_scada_field(
|
||
device_id: str,
|
||
time: datetime,
|
||
field: str,
|
||
value: float,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
try:
|
||
await ScadaRepository.update_scada_field(conn, time, device_id, field, value)
|
||
return {"message": "Updated successfully"}
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
|
||
|
||
@router.delete("/scada")
|
||
async def delete_scada_data(
|
||
device_id: str,
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
conn: AsyncConnection = Depends(get_database_connection),
|
||
):
|
||
await ScadaRepository.delete_scada_by_id_time_range(
|
||
conn, device_id, start_time, end_time
|
||
)
|
||
return {"message": "Deleted successfully"}
|
||
|
||
|
||
# --- Composite Query Endpoints ---
|
||
|
||
|
||
@router.get("/composite/scada-simulation")
|
||
async def get_scada_associated_simulation_data(
|
||
device_id: str,
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
field: str,
|
||
timescale_conn: AsyncConnection = Depends(get_database_connection),
|
||
postgres_conn: AsyncConnection = Depends(get_postgres_connection),
|
||
):
|
||
"""
|
||
获取 SCADA 关联的 link/node 模拟值
|
||
|
||
根据传入的 SCADA device_id，找到关联的 link/node，
|
||
并根据对应的 type，查询对应的模拟数据
|
||
"""
|
||
try:
|
||
result = await CompositeQueries.get_scada_associated_simulation_data(
|
||
timescale_conn, postgres_conn, device_id, start_time, end_time, field
|
||
)
|
||
if result is None:
|
||
raise HTTPException(status_code=404, detail="No simulation data found")
|
||
return result
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|
||
|
||
|
||
@router.get("/composite/element-scada")
|
||
async def get_element_associated_scada_data(
|
||
element_id: str,
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
use_cleaned: bool = Query(False, description="是否使用清洗后的数据"),
|
||
timescale_conn: AsyncConnection = Depends(get_database_connection),
|
||
postgres_conn: AsyncConnection = Depends(get_postgres_connection),
|
||
):
|
||
"""
|
||
获取 link/node 关联的 SCADA 监测值
|
||
|
||
根据传入的 link/node id，匹配 SCADA 信息，
|
||
如果存在关联的 SCADA device_id，获取实际的监测数据
|
||
"""
|
||
try:
|
||
result = await CompositeQueries.get_element_associated_scada_data(
|
||
timescale_conn, postgres_conn, element_id, start_time, end_time, use_cleaned
|
||
)
|
||
if result is None:
|
||
raise HTTPException(
|
||
status_code=404, detail="No associated SCADA data found"
|
||
)
|
||
return result
|
||
except ValueError as e:
|
||
raise HTTPException(status_code=400, detail=str(e))
|