优化漏损识别器，支持多进程评估

app/infra/db/timescaledb/internal_queries.py

@@ -120,3 +120,53 @@ class InternalQueries:
                     time.sleep(1)
                 else:
                     raise
+
+    @staticmethod
+    def query_scada_by_ids_timerange(
+        device_ids: List[str],
+        start_time: str | datetime,
+        end_time: str | datetime,
+        db_name: str = None,
+        max_retries: int = 3,
+    ) -> dict[str, list[dict]]:
+        """查询指定时间窗的 SCADA 数据，返回 {device_id: [{time, value}, ...]}。"""
+        start_dt = (
+            datetime.fromisoformat(start_time)
+            if isinstance(start_time, str)
+            else start_time
+        )
+        end_dt = (
+            datetime.fromisoformat(end_time) if isinstance(end_time, str) else end_time
+        )
+
+        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_dt, end_dt
+                    )
+                    result: dict[str, list[dict]] = {
+                        device_id: [] for device_id in device_ids
+                    }
+                    for row in rows:
+                        device_id = row["device_id"]
+                        value = row.get("cleaned_value")
+                        if value is None:
+                            value = row.get("monitored_value")
+                        result.setdefault(device_id, []).append(
+                            {"time": row["time"].isoformat(), "value": value}
+                        )
+                    for device_id in result:
+                        result[device_id].sort(key=lambda item: item["time"])
+                    return result
+            except Exception as e:
+                logger.error(f"查询尝试 {attempt + 1} 失败: {e}")
+                if attempt < max_retries - 1:
+                    time.sleep(1)
+                else:
+                    raise

app/infra/db/timescaledb/schemas/scada.py

@@ -2,6 +2,7 @@ from typing import List, Any
 from datetime import datetime
 from collections import defaultdict
 from psycopg import AsyncConnection, Connection, sql
+from psycopg.rows import dict_row
 
 
 class ScadaRepository:
@@ -46,7 +47,7 @@ class ScadaRepository:
         start_time: datetime,
         end_time: datetime,
     ) -> List[dict]:
-        with conn.cursor() as cur:
+        with conn.cursor(row_factory=dict_row) 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),