40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
|
from cloud import CloudConnector
|
|||
|
|
from local import LocalConnector
|
|||
|
|
from ..FFAICilent.config.configloder import load_config
|
|||
|
|
|
|||
|
|
class ConnectionManager:
|
|||
|
|
def __init__(self):
|
|||
|
|
self.config = load_config()
|
|||
|
|
self.mode = self.config['Deployment']['default_mode']
|
|||
|
|
self._init_connectors()
|
|||
|
|
|
|||
|
|
def _init_connectors(self):
|
|||
|
|
self.cloud = CloudConnector(self.config)
|
|||
|
|
self.local = LocalConnector(self.config)
|
|||
|
|
|
|||
|
|
def execute(self, command):
|
|||
|
|
# 优先使用默认模式
|
|||
|
|
if self.mode == 'cloud':
|
|||
|
|
try:
|
|||
|
|
return self.cloud.execute(command)
|
|||
|
|
except ConnectionError as e:
|
|||
|
|
if self.config.getboolean('Deployment', 'local_fallback'):
|
|||
|
|
return self._fallback_to_local(command)
|
|||
|
|
raise
|
|||
|
|
else:
|
|||
|
|
return self.local.execute(command)
|
|||
|
|
|
|||
|
|
def _fallback_to_local(self, command):
|
|||
|
|
"""自动降级到本地模式"""
|
|||
|
|
from logger import log
|
|||
|
|
log.warning("Falling back to local mode")
|
|||
|
|
self.mode = 'local'
|
|||
|
|
|
|||
|
|
# 转换云指令到本地查询
|
|||
|
|
local_query = self._adapt_command(command)
|
|||
|
|
return self.local.execute(local_query)
|
|||
|
|
|
|||
|
|
def _adapt_command(self, cloud_cmd):
|
|||
|
|
"""将云API指令转换为本地查询（示例）"""
|
|||
|
|
# 这里添加你的业务逻辑转换规则
|
|||
|
|
return f"SELECT * FROM cache WHERE key='{cloud_cmd}'"
|