127 lines
4.3 KiB
Python
127 lines
4.3 KiB
Python
from typing import List, Optional, Dict, Any
|
||
from datetime import datetime
|
||
from psycopg import AsyncConnection
|
||
|
||
from postgresql.scada_info import ScadaRepository as PostgreScadaRepository
|
||
from timescaledb.schemas.realtime import RealtimeRepository
|
||
from timescaledb.schemas.scada import ScadaRepository
|
||
|
||
|
||
class CompositeQueries:
|
||
"""
|
||
复合查询类，提供跨表查询功能
|
||
"""
|
||
|
||
@staticmethod
|
||
async def get_scada_associated_simulation_data(
|
||
timescale_conn: AsyncConnection,
|
||
postgres_conn: AsyncConnection,
|
||
device_id: str,
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
field: str,
|
||
) -> Optional[Any]:
|
||
"""
|
||
获取 SCADA 关联的 link/node 模拟值
|
||
|
||
根据传入的 SCADA device_id，找到关联的 link/node，
|
||
并根据对应的 type，查询对应的模拟数据
|
||
|
||
Args:
|
||
timescale_conn: TimescaleDB 异步连接
|
||
postgres_conn: PostgreSQL 异步连接
|
||
device_id: SCADA 设备ID
|
||
start_time: 开始时间
|
||
end_time: 结束时间
|
||
field: 要查询的字段名
|
||
|
||
Returns:
|
||
模拟数据值，如果没有找到则返回 None
|
||
|
||
Raises:
|
||
ValueError: 当 SCADA 设备未找到或字段无效时
|
||
"""
|
||
# 1. 查询所有 SCADA 信息
|
||
scada_infos = await PostgreScadaRepository.get_scadas(postgres_conn)
|
||
|
||
# 2. 根据 device_id 找到对应的 SCADA 信息
|
||
target_scada = None
|
||
for scada in scada_infos:
|
||
if scada["id"] == device_id:
|
||
target_scada = scada
|
||
break
|
||
|
||
if not target_scada:
|
||
raise ValueError(f"SCADA device {device_id} not found")
|
||
|
||
# 3. 根据 type 和 associated_element_id 查询对应的模拟数据
|
||
element_id = target_scada["associated_element_id"]
|
||
scada_type = target_scada["type"]
|
||
|
||
if scada_type.lower() == "pipe_flow":
|
||
# 查询 link 模拟数据
|
||
return await RealtimeRepository.get_link_field_by_time_range(
|
||
timescale_conn, start_time, end_time, element_id, field
|
||
)
|
||
elif scada_type.lower() == "pressure":
|
||
# 查询 node 模拟数据
|
||
return await RealtimeRepository.get_node_field_by_time_range(
|
||
timescale_conn, start_time, end_time, element_id, field
|
||
)
|
||
else:
|
||
raise ValueError(f"Unknown SCADA type: {scada_type}")
|
||
|
||
@staticmethod
|
||
async def get_element_associated_scada_data(
|
||
timescale_conn: AsyncConnection,
|
||
postgres_conn: AsyncConnection,
|
||
element_id: str,
|
||
start_time: datetime,
|
||
end_time: datetime,
|
||
use_cleaned: bool = False,
|
||
) -> Optional[Any]:
|
||
"""
|
||
获取 link/node 关联的 SCADA 监测值
|
||
|
||
根据传入的 link/node id，匹配 SCADA 信息，
|
||
如果存在关联的 SCADA device_id，获取实际的监测数据
|
||
|
||
Args:
|
||
timescale_conn: TimescaleDB 异步连接
|
||
postgres_conn: PostgreSQL 异步连接
|
||
element_id: link 或 node 的 ID
|
||
start_time: 开始时间
|
||
end_time: 结束时间
|
||
use_cleaned: 是否使用清洗后的数据 (True: "cleaned_value", False: "monitored_value")
|
||
|
||
Returns:
|
||
SCADA 监测数据值，如果没有找到则返回 None
|
||
|
||
Raises:
|
||
ValueError: 当元素类型无效时
|
||
"""
|
||
|
||
# 1. 查询所有 SCADA 信息
|
||
scada_infos = await PostgreScadaRepository.get_scadas(postgres_conn)
|
||
|
||
# 2. 根据 element_type 和 element_id 找到关联的 SCADA 设备
|
||
associated_scada = None
|
||
for scada in scada_infos:
|
||
if scada["associated_element_id"] == element_id:
|
||
associated_scada = scada
|
||
break
|
||
|
||
if not associated_scada:
|
||
# 没有找到关联的 SCADA 设备
|
||
return None
|
||
|
||
# 3. 通过 SCADA device_id 获取监测数据
|
||
device_id = associated_scada["id"]
|
||
|
||
# 根据 use_cleaned 参数选择字段
|
||
data_field = "cleaned_value" if use_cleaned else "monitored_value"
|
||
|
||
return await ScadaRepository.get_scada_field_by_id_time_range(
|
||
timescale_conn, device_id, start_time, end_time, data_field
|
||
)
|