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
|