新增同步存储方法；新增run_server.py文件；修改默认的数据库连接方式；

timescaledb/internal_queries.py

@@ -1,28 +1,49 @@
 from typing import List
+
+from fastapi.logger import logger
 from timescaledb.schemas.scheme import SchemeRepository
 from timescaledb.schemas.realtime import RealtimeRepository
-from timescaledb.database import get_database_instance
+import timescaledb.timescaledb_info as timescaledb_info
+import psycopg
+import time
 
 # 内部使用存储类
 
 
 class InternalStorage:
     @staticmethod
-    async def store_realtime_simulation(
+    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,
     ):
         """存储实时模拟结果"""
-        instance = await get_database_instance(db_name)
-        async with instance.get_connection() as conn:
-            await RealtimeRepository.store_realtime_simulation_result(
-                conn, node_result_list, link_result_list, result_start_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:
+                    starttime = time.time()
+                    RealtimeRepository.store_realtime_simulation_result_sync(
+                        conn, node_result_list, link_result_list, result_start_time
+                    )
+                    endtime = time.time()
+                    logger.info(f"存储实时模拟结果耗时: {endtime - starttime} 秒")
+                break  # 成功
+            except Exception as e:
+                logger.error(f"存储尝试 {attempt + 1} 失败: {e}")
+                if attempt < max_retries - 1:
+                    time.sleep(1)  # 重试前等待
+                else:
+                    raise  # 达到最大重试次数后抛出异常
 
     @staticmethod
-    async def store_scheme_simulation(
+    def store_scheme_simulation(
         scheme_type: str,
         scheme_name: str,
         node_result_list: List[dict],
@@ -30,16 +51,30 @@ class InternalStorage:
         result_start_time: str,
         num_periods: int = 1,
         db_name: str = None,
+        max_retries: int = 3,
     ):
         """存储方案模拟结果"""
-        instance = await get_database_instance(db_name)
-        async with instance.get_connection() as conn:
-            await SchemeRepository.store_scheme_simulation_result(
-                conn,
-                scheme_type,
-                scheme_name,
-                node_result_list,
-                link_result_list,
-                result_start_time,
-                num_periods,
-            )
+        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  # 达到最大重试次数后抛出异常