81 lines
2.9 KiB
Python
81 lines
2.9 KiB
Python
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
|
|
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:
|
|
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
|
|
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 # 达到最大重试次数后抛出异常
|