更新方法以支持多id查询

timescaledb/composite_queries.py

@@ -17,133 +17,140 @@ class CompositeQueries:
     async def get_scada_associated_realtime_simulation_data(
         timescale_conn: AsyncConnection,
         postgres_conn: AsyncConnection,
-        device_id: str,
+        device_ids: List[str],
         start_time: datetime,
         end_time: datetime,
         field: str,
-    ) -> Optional[Any]:
+    ) -> List[Optional[Any]]:
         """
         获取 SCADA 关联的 link/node 模拟值
 
-        根据传入的 SCADA device_id，找到关联的 link/node，
+        根据传入的 SCADA device_ids，找到关联的 link/node，
         并根据对应的 type，查询对应的模拟数据
 
         Args:
             timescale_conn: TimescaleDB 异步连接
             postgres_conn: PostgreSQL 异步连接
-            device_id: SCADA 设备ID
+            device_ids: SCADA 设备ID列表
             start_time: 开始时间
             end_time: 结束时间
             field: 要查询的字段名
 
         Returns:
-            模拟数据值，如果没有找到则返回 None
+            模拟数据值列表，如果没有找到则对应位置返回 None
 
         Raises:
             ValueError: 当 SCADA 设备未找到或字段无效时
         """
+        results = []
         # 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
+        for device_id in device_ids:
+            # 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")
+            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"]
+            # 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}")
+            if scada_type.lower() == "pipe_flow":
+                # 查询 link 模拟数据
+                res = await RealtimeRepository.get_link_field_by_time_range(
+                    timescale_conn, start_time, end_time, element_id, field
+                )
+            elif scada_type.lower() == "pressure":
+                # 查询 node 模拟数据
+                res = 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}")
+            results.append(res)
+        return results
 
     @staticmethod
     async def get_scada_associated_scheme_simulation_data(
         timescale_conn: AsyncConnection,
         postgres_conn: AsyncConnection,
-        device_id: str,
+        device_ids: List[str],
         start_time: datetime,
         end_time: datetime,
         scheme_type: str,
         scheme_name: str,
-        field: str,
-    ) -> Optional[Any]:
+    ) -> List[Optional[Any]]:
         """
         获取 SCADA 关联的 link/node 模拟值
 
-        根据传入的 SCADA device_id，找到关联的 link/node，
+        根据传入的 SCADA device_ids，找到关联的 link/node，
         并根据对应的 type，查询对应的模拟数据
 
         Args:
             timescale_conn: TimescaleDB 异步连接
             postgres_conn: PostgreSQL 异步连接
-            device_id: SCADA 设备ID
+            device_ids: SCADA 设备ID列表
             start_time: 开始时间
             end_time: 结束时间
             field: 要查询的字段名
 
         Returns:
-            模拟数据值，如果没有找到则返回 None
+            模拟数据值列表，如果没有找到则对应位置返回 None
 
         Raises:
             ValueError: 当 SCADA 设备未找到或字段无效时
         """
+        results = []
         # 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
+        for device_id in device_ids:
+            # 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")
+            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"]
+            # 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 SchemeRepository.get_link_field_by_scheme_and_time_range(
-                timescale_conn,
-                scheme_type,
-                scheme_name,
-                start_time,
-                end_time,
-                element_id,
-                field,
-            )
-        elif scada_type.lower() == "pressure":
-            # 查询 node 模拟数据
-            return await SchemeRepository.get_node_field_by_scheme_and_time_range(
-                timescale_conn,
-                scheme_type,
-                scheme_name,
-                start_time,
-                end_time,
-                element_id,
-                field,
-            )
-        else:
-            raise ValueError(f"Unknown SCADA type: {scada_type}")
+            if scada_type.lower() == "pipe_flow":
+                # 查询 link 模拟数据
+                res = await SchemeRepository.get_link_field_by_scheme_and_time_range(
+                    timescale_conn,
+                    scheme_type,
+                    scheme_name,
+                    start_time,
+                    end_time,
+                    element_id,
+                    "flow",
+                )
+            elif scada_type.lower() == "pressure":
+                # 查询 node 模拟数据
+                res = await SchemeRepository.get_node_field_by_scheme_and_time_range(
+                    timescale_conn,
+                    scheme_type,
+                    scheme_name,
+                    start_time,
+                    end_time,
+                    element_id,
+                    "pressure",
+                )
+            else:
+                raise ValueError(f"Unknown SCADA type: {scada_type}")
+            results.append(res)
+        return results
 
     @staticmethod
     async def get_element_associated_scada_data(