创建项目及AI生成基本代码

internal/protocol/websocket/client.py

@@ -0,0 +1,143 @@
+#!/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