Update.

Author: xOS

.gitignore

@@ -25,3 +25,4 @@ server
 scripts/monitor_goroutines.sh
 FIXES_APPLIED.md
 debug_monitor.sh
+test_websocket.sh

cmd/dashboard/controller/common_page.go

@@ -2,6 +2,7 @@ package controller
 
 import (
 	"context"
+	"encoding/json"
 	"errors"
 	"fmt"
 	"log"
@@ -1141,6 +1142,9 @@ func (cp *commonPage) ws(c *gin.Context) {
 		CheckOrigin: func(r *http.Request) bool {
 			return true
 		},
+		// 添加缓冲区大小配置
+		ReadBufferSize:  1024,
+		WriteBufferSize: 1024,
 	}
 	conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
 	if err != nil {
@@ -1151,8 +1155,19 @@ func (cp *commonPage) ws(c *gin.Context) {
 	// 使用正确的构造函数
 	safeConn := websocketx.NewConn(conn)
 
+	// 设置连接参数
+	conn.SetReadDeadline(time.Now().Add(60 * time.Second))
+	conn.SetWriteDeadline(time.Now().Add(10 * time.Second))
+
+	// 设置ping/pong处理
+	conn.SetPongHandler(func(string) error {
+		conn.SetReadDeadline(time.Now().Add(60 * time.Second))
+		return nil
+	})
+
 	// 使用一个channel来通知写入goroutine退出
 	done := make(chan struct{})
+	lastActivity := time.Now()
 
 	// Read goroutine
 	go func() {
@@ -1161,35 +1176,76 @@ func (cp *commonPage) ws(c *gin.Context) {
 			safeConn.Close()
 		}()
 		for {
-			// 我们需要从连接中读取，以检测客户端是否已断开连接。
-			// 我们不需要处理任何传入的消息。
-			if _, _, err := safeConn.ReadMessage(); err != nil {
+			// 设置读取超时
+			conn.SetReadDeadline(time.Now().Add(60 * time.Second))
+
+			messageType, message, err := safeConn.ReadMessage()
+			if err != nil {
 				if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
 					log.Printf("NG-ERROR: websocket read error: %v", err)
 				}
 				break // 退出循环
 			}
+
+			// 更新活动时间
+			lastActivity = time.Now()
+
+			// 处理客户端ping消息
+			if messageType == websocket.TextMessage {
+				var msg map[string]interface{}
+				if err := json.Unmarshal(message, &msg); err == nil {
+					if msgType, ok := msg["type"].(string); ok && msgType == "ping" {
+						// 发送pong响应
+						pongMsg := map[string]interface{}{"type": "pong", "timestamp": time.Now().Unix()}
+						if pongData, err := json.Marshal(pongMsg); err == nil {
+							safeConn.WriteMessage(websocket.TextMessage, pongData)
+						}
+					}
+				}
+			}
 		}
 	}()
 
 	// Write goroutine
 	go func() {
-		ticker := time.NewTicker(time.Second * 1) // 从5秒改为1秒，提高更新频率
-		defer ticker.Stop()
+		ticker := time.NewTicker(time.Second * 1)      // 1秒更新间隔，提供更好的实时性
+		pingTicker := time.NewTicker(time.Second * 30) // 30秒心跳间隔
+		defer func() {
+			ticker.Stop()
+			pingTicker.Stop()
+		}()
+
 		for {
 			select {
 			case <-done: // 从读取goroutine接收到退出信号
 				return
+			case <-pingTicker.C:
+				// 发送ping消息保持连接活跃
+				conn.SetWriteDeadline(time.Now().Add(10 * time.Second))
+				if err := safeConn.WriteMessage(websocket.PingMessage, []byte{}); err != nil {
+					log.Printf("发送ping消息失败: %v", err)
+					return
+				}
+
+				// 检查连接是否长时间无活动
+				if time.Since(lastActivity) > 90*time.Second {
+					log.Printf("WebSocket连接长时间无活动，主动关闭")
+					return
+				}
+
 			case <-ticker.C:
 				stat, err := cp.getServerStat(c, false)
 				if err != nil {
 					log.Printf("NG-ERROR: failed to get server stat for websocket: %v", err)
 					// 不要退出，让 done channel 处理终止
 					continue
 				}
+
+				// 设置写入超时
+				conn.SetWriteDeadline(time.Now().Add(10 * time.Second))
 				if err = safeConn.WriteMessage(websocket.TextMessage, stat); err != nil {
 					// 写入失败，可能是因为连接已关闭。
-					// 读取goroutine将处理清理工作。我们可以在这里退出。
+					log.Printf("WebSocket写入失败: %v", err)
 					return
 				}
 			}