新增内部查询方法，替换simulation中scada设备查询方法

timescaledb/internal_queries.py

@@ -4,6 +4,8 @@ from fastapi.logger import logger
 from timescaledb.schemas.scheme import SchemeRepository
 from timescaledb.schemas.realtime import RealtimeRepository
 import timescaledb.timescaledb_info as timescaledb_info
+from datetime import datetime, timedelta
+from timescaledb.schemas.scada import ScadaRepository
 import psycopg
 import time
 
@@ -75,3 +77,46 @@ class InternalStorage:
                     time.sleep(1)  # 重试前等待
                 else:
                     raise  # 达到最大重试次数后抛出异常
+
+    @staticmethod
+    def query_scada_by_ids_time(
+        device_ids: List[str],
+        query_time: str,
+        db_name: str = None,
+        max_retries: int = 3,
+    ) -> dict:
+        """查询指定时间点的 SCADA 数据"""
+
+        # 解析时间，假设是北京时间
+        beijing_time = datetime.fromisoformat(query_time)
+        start_time = beijing_time - timedelta(seconds=1)
+        end_time = beijing_time + timedelta(seconds=1)
+
+        for attempt in range(max_retries):
+            try:
+                conn_string = (
+                    timescaledb_info.get_pgconn_string(db_name=db_name)
+                    if db_name
+                    else timescaledb_info.get_pgconn_string()
+                )
+                with psycopg.Connection.connect(conn_string) as conn:
+                    rows = ScadaRepository.get_scada_by_ids_time_range_sync(
+                        conn, device_ids, start_time, end_time
+                    )
+                    # 处理结果，返回每个 device_id 的第一个值
+                    result = {}
+                    for device_id in device_ids:
+                        device_rows = [
+                            row for row in rows if row["device_id"] == device_id
+                        ]
+                        if device_rows:
+                            result[device_id] = device_rows[0]["monitored_value"]
+                        else:
+                            result[device_id] = None
+                    return result
+            except Exception as e:
+                logger.error(f"查询尝试 {attempt + 1} 失败: {e}")
+                if attempt < max_retries - 1:
+                    time.sleep(1)
+                else:
+                    raise

timescaledb/router.py

@@ -358,20 +358,20 @@ async def insert_scada_data(
 
 
 @router.get("/scada")
-async def get_scada_by_id_time_range(
-    device_id: str,
+async def get_scada_by_ids_time_range(
+    device_ids: List[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
+    return await ScadaRepository.get_scada_by_ids_time_range(
+        conn, device_ids, start_time, end_time
     )
 
 
-@router.get("/scada/{device_id}/field")
-async def get_scada_field_by_id_time_range(
-    device_id: str,
+@router.get("/scada/field")
+async def get_scada_field_by_ids_time_range(
+    device_ids: List[str],
     start_time: datetime,
     end_time: datetime,
     field: str,
@@ -379,7 +379,7 @@ async def get_scada_field_by_id_time_range(
 ):
     try:
         return await ScadaRepository.get_scada_field_by_id_time_range(
-            conn, device_id, start_time, end_time, field
+            conn, device_ids, start_time, end_time, field
         )
     except ValueError as e:
         raise HTTPException(status_code=400, detail=str(e))

timescaledb/schemas/scada.py

@@ -1,6 +1,6 @@
 from typing import List, Any
 from datetime import datetime
-from psycopg import AsyncConnection, sql
+from psycopg import AsyncConnection, Connection, sql
 
 
 class ScadaRepository:
@@ -25,36 +25,53 @@ class ScadaRepository:
                     )
 
     @staticmethod
-    async def get_scada_by_id_time_range(
-        conn: AsyncConnection, device_id: str, start_time: datetime, end_time: datetime
+    async def get_scada_by_ids_time_range(
+        conn: AsyncConnection,
+        device_ids: List[str],
+        start_time: datetime,
+        end_time: datetime,
     ) -> List[dict]:
         async with conn.cursor() as cur:
             await cur.execute(
-                "SELECT * FROM scada.scada_data WHERE device_id = %s AND time >= %s AND time <= %s",
-                (device_id, start_time, end_time),
+                "SELECT * FROM scada.scada_data WHERE device_id = ANY(%s) AND time >= %s AND time <= %s",
+                (device_ids, start_time, end_time),
             )
             return await cur.fetchall()
 
+    @staticmethod
+    def get_scada_by_ids_time_range_sync(
+        conn: Connection,
+        device_ids: List[str],
+        start_time: datetime,
+        end_time: datetime,
+    ) -> List[dict]:
+        with conn.cursor() as cur:
+            cur.execute(
+                "SELECT * FROM scada.scada_data WHERE device_id = ANY(%s) AND time >= %s AND time <= %s",
+                (device_ids, start_time, end_time),
+            )
+            return cur.fetchall()
+
     @staticmethod
     async def get_scada_field_by_id_time_range(
         conn: AsyncConnection,
-        device_id: str,
+        device_ids: List[str],
         start_time: datetime,
         end_time: datetime,
         field: str,
-    ) -> Any:
+    ) -> List[dict]:
         valid_fields = {"monitored_value", "cleaned_value"}
         if field not in valid_fields:
             raise ValueError(f"Invalid field: {field}")
 
         query = sql.SQL(
-            "SELECT {} FROM scada.scada_data WHERE time >= %s AND time <= %s AND device_id = %s"
+            "SELECT device_id, {} FROM scada.scada_data WHERE time >= %s AND time <= %s AND device_id = ANY(%s)"
         ).format(sql.Identifier(field))
 
         async with conn.cursor() as cur:
-            await cur.execute(query, (start_time, end_time, device_id))
-            row = await cur.fetchone()
-            return row[field] if row else None
+            await cur.execute(query, (start_time, end_time, device_ids))
+            rows = await cur.fetchall()
+            return [{"device_id": row["device_id"], field: row[field]} for row in rows]
 
     @staticmethod
     async def update_scada_field(