143 lines
4.6 KiB
Python
143 lines
4.6 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
WebSocket客户端模块
|
||
用于与平台建立WebSocket连接并处理通信
|
||
"""
|
||
|
||
import asyncio
|
||
import json
|
||
import logging
|
||
import websockets
|
||
from datetime import datetime
|
||
from urllib.parse import urlencode
|
||
|
||
from config import config
|
||
|
||
# 配置日志
|
||
logging.basicConfig(level=logging.INFO)
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class WebSocketClient:
|
||
"""WebSocket客户端类"""
|
||
|
||
def __init__(self, relay_service):
|
||
"""
|
||
初始化WebSocket客户端
|
||
|
||
Args:
|
||
relay_service: 中继器服务实例
|
||
"""
|
||
self.relay_service = relay_service
|
||
self.device_id = config._config.get('relay', {}).get('device_id', '1')
|
||
self.host = config._config.get('websocket', {}).get('host', 'localhost')
|
||
self.port = config._config.get('websocket', {}).get('port', 8086)
|
||
self.timeout = config._config.get('websocket', {}).get('timeout', 5)
|
||
self.websocket = None
|
||
self.connected = False
|
||
self.running = False
|
||
|
||
logger.info(f"初始化WebSocket客户端: device_id={self.device_id}, host={self.host}, port={self.port}")
|
||
|
||
async def connect(self):
|
||
"""建立WebSocket连接"""
|
||
try:
|
||
# 构建连接URL
|
||
params = urlencode({'device_id': self.device_id})
|
||
uri = f"ws://{self.host}:{self.port}/ws/device?{params}"
|
||
|
||
logger.info(f"正在连接到平台: {uri}")
|
||
|
||
# 建立WebSocket连接（移除了timeout参数以兼容新版本websockets库）
|
||
self.websocket = await websockets.connect(uri)
|
||
self.connected = True
|
||
self.running = True
|
||
|
||
logger.info("成功连接到平台")
|
||
return True
|
||
|
||
except Exception as e:
|
||
logger.error(f"连接平台失败: {e}")
|
||
self.connected = False
|
||
return False
|
||
|
||
async def disconnect(self):
|
||
"""断开WebSocket连接"""
|
||
logger.info("断开WebSocket连接")
|
||
self.running = False
|
||
|
||
if self.websocket:
|
||
await self.websocket.close()
|
||
self.websocket = None
|
||
|
||
self.connected = False
|
||
logger.info("WebSocket连接已断开")
|
||
|
||
async def send_message(self, message):
|
||
"""
|
||
发送消息到平台
|
||
|
||
Args:
|
||
message (dict): 要发送的消息
|
||
"""
|
||
if not self.connected or not self.websocket:
|
||
logger.warning("WebSocket未连接，无法发送消息")
|
||
return False
|
||
|
||
try:
|
||
message_str = json.dumps(message, ensure_ascii=False)
|
||
await self.websocket.send(message_str)
|
||
logger.debug(f"发送消息到平台: {message_str}")
|
||
return True
|
||
except Exception as e:
|
||
logger.error(f"发送消息失败: {e}")
|
||
return False
|
||
|
||
async def listen(self):
|
||
"""监听平台消息"""
|
||
if not self.connected or not self.websocket:
|
||
logger.error("WebSocket未连接，无法监听消息")
|
||
return
|
||
|
||
logger.info("开始监听平台消息")
|
||
|
||
try:
|
||
async for message in self.websocket:
|
||
try:
|
||
# 解析收到的消息
|
||
message_data = json.loads(message)
|
||
logger.info(f"收到平台消息: {message_data}")
|
||
|
||
# 将消息传递给中继器服务处理
|
||
self.relay_service.handle_platform_command(message)
|
||
|
||
except json.JSONDecodeError as e:
|
||
logger.error(f"解析平台消息失败: {e}")
|
||
except Exception as e:
|
||
logger.error(f"处理平台消息时发生错误: {e}")
|
||
|
||
except websockets.exceptions.ConnectionClosed:
|
||
logger.warning("WebSocket连接已关闭")
|
||
self.connected = False
|
||
except Exception as e:
|
||
logger.error(f"监听平台消息时发生错误: {e}")
|
||
self.connected = False
|
||
|
||
async def send_heartbeat(self):
|
||
"""发送心跳消息"""
|
||
if not self.connected:
|
||
return False
|
||
|
||
heartbeat_msg = {
|
||
"type": "heartbeat",
|
||
"device_id": self.device_id,
|
||
"timestamp": datetime.utcnow().isoformat() + 'Z'
|
||
}
|
||
|
||
return await self.send_message(heartbeat_msg)
|
||
|
||
def is_connected(self):
|
||
"""检查是否已连接"""
|
||
return self.connected |