fix(client): 增强WebSocket消息处理与心跳机制

- 为WebSocket消息读取设置超时，确保有足够时间接收消息
- 在心跳发送失败时主动关闭连接以触发重连机制，并记录警告日志

Author: orange-juzipi

internal/client/ws_message_handler.go

@@ -49,7 +49,11 @@ func (c *WSClient) handleWSMessages() error {
 			return errors.New("连接已关闭")
 		}
 
-		_, data, err := conn.Read(c.ctx)
+		// 设置读取超时（心跳间隔的 3 倍，确保有足够时间接收消息）
+		readCtx, readCancel := context.WithTimeout(c.ctx, heartbeatInterval*3)
+		_, data, err := conn.Read(readCtx)
+		readCancel()
+
 		if err != nil {
 			if errors.Is(err, context.Canceled) {
 				return nil

internal/client/ws_sender.go

@@ -107,6 +107,13 @@ func (c *WSClient) sendHeartbeat(ctx context.Context) {
 			}
 
 			if err := c.sendNotifyRequest(req); err != nil {
+				logger.Warn("发送心跳失败，主动关闭连接以触发重连", "error", err)
+				// 主动关闭连接，触发重连机制
+				c.connMu.Lock()
+				if c.conn != nil {
+					c.conn.Close(websocket.StatusAbnormalClosure, "heartbeat failed")
+				}
+				c.connMu.Unlock()
 				return
 			}
 		}