重构现代化 FastAPI 后端项目框架

app/infra/db/timescaledb/internal_queries.py

@@ -0,0 +1,122 @@
+from typing import List
+
+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
+
+
+class InternalStorage:
+    @staticmethod
+    def store_realtime_simulation(
+        node_result_list: List[dict],
+        link_result_list: List[dict],
+        result_start_time: str,
+        db_name: str = None,
+        max_retries: int = 3,
+    ):
+        """存储实时模拟结果"""
+        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:
+                    RealtimeRepository.store_realtime_simulation_result_sync(
+                        conn, node_result_list, link_result_list, result_start_time
+                    )
+                break  # 成功
+            except Exception as e:
+                logger.error(f"存储尝试 {attempt + 1} 失败: {e}")
+                if attempt < max_retries - 1:
+                    time.sleep(1)  # 重试前等待
+                else:
+                    raise  # 达到最大重试次数后抛出异常
+
+    @staticmethod
+    def store_scheme_simulation(
+        scheme_type: str,
+        scheme_name: str,
+        node_result_list: List[dict],
+        link_result_list: List[dict],
+        result_start_time: str,
+        num_periods: int = 1,
+        db_name: str = None,
+        max_retries: int = 3,
+    ):
+        """存储方案模拟结果"""
+        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:
+                    SchemeRepository.store_scheme_simulation_result_sync(
+                        conn,
+                        scheme_type,
+                        scheme_name,
+                        node_result_list,
+                        link_result_list,
+                        result_start_time,
+                        num_periods,
+                    )
+                break  # 成功
+            except Exception as e:
+                logger.error(f"存储尝试 {attempt + 1} 失败: {e}")
+                if attempt < max_retries - 1:
+                    time.sleep(1)  # 重试前等待
+                else:
+                    raise  # 达到最大重试次数后抛出异常
+
+
+class InternalQueries:
+    @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