58 lines
2.5 KiB
Python
58 lines
2.5 KiB
Python
import asyncio
|
|
import websockets
|
|
|
|
|
|
async def connect_flow(client):
|
|
"""连接 WebSocket 服务器并自动重连。"""
|
|
while client.running:
|
|
try:
|
|
client.logger.info(f"[{client.get_time()}] 正在连接轻简API {client.uri}...")
|
|
async with websockets.connect(client.uri) as websocket:
|
|
client.websocket = websocket
|
|
from utils.health_check import set_qingjian_connected
|
|
set_qingjian_connected(True)
|
|
client.logger.info(f"[{client.get_time()}] 连接成功!")
|
|
if client.enable_agent:
|
|
client.logger.info(f"[{client.get_time()}] AI Agent 已启用，将自动处理消息")
|
|
client.logger.info(f"[{client.get_time()}] 等待接收消息...")
|
|
|
|
await client.receive_messages()
|
|
|
|
except ConnectionRefusedError:
|
|
from utils.health_check import set_qingjian_connected
|
|
set_qingjian_connected(False)
|
|
client.logger.info(f"[{client.get_time()}] 连接被拒绝，请检查轻简软件是否已启动")
|
|
except websockets.exceptions.InvalidURI:
|
|
from utils.health_check import set_qingjian_connected
|
|
set_qingjian_connected(False)
|
|
client.logger.info(f"[{client.get_time()}] URI格式错误")
|
|
except Exception as e:
|
|
from utils.health_check import set_qingjian_connected
|
|
set_qingjian_connected(False)
|
|
client.logger.info(f"[{client.get_time()}] 连接错误: {e}")
|
|
|
|
if client.running:
|
|
client.logger.info(f"[{client.get_time()}] 5秒后尝试重连...")
|
|
await asyncio.sleep(5)
|
|
|
|
|
|
async def receive_messages_flow(client):
|
|
"""持续接收消息。"""
|
|
try:
|
|
async for message in client.websocket:
|
|
await client.handle_message(message)
|
|
except websockets.exceptions.ConnectionClosed:
|
|
from utils.health_check import set_qingjian_connected
|
|
set_qingjian_connected(False)
|
|
client.logger.info(f"[{client.get_time()}] 连接已关闭")
|
|
except Exception as e:
|
|
from utils.health_check import set_qingjian_connected
|
|
set_qingjian_connected(False)
|
|
client.logger.info(f"[{client.get_time()}] 接收消息错误: {e}")
|
|
|
|
|
|
async def handle_message_flow(client, message, *, shop_type_resolver):
|
|
from core.websocket_inbound_flow import handle_incoming_message
|
|
|
|
await handle_incoming_message(client, message, shop_type_resolver=shop_type_resolver)
|