新增pg和timescaledb的复合查询方法，用于查询 scada 关联的模拟、原始、清洗数据

timescaledb/composite_queries.py

@@ -0,0 +1,126 @@
+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
+        )