Update.

Author: xOS

cmd/dashboard/controller/common_page.go

@@ -1038,20 +1038,22 @@ func (cp *commonPage) home(c *gin.Context) {
 var upgrader = websocket.Upgrader{
 	ReadBufferSize:  32768,
 	WriteBufferSize: 32768,
+	CheckOrigin: func(r *http.Request) bool {
+		// 允许所有来源的WebSocket连接，避免跨域问题
+		return true
+	},
+	Error: func(w http.ResponseWriter, r *http.Request, status int, reason error) {
+		// 自定义错误处理，避免写入响应头冲突
+		log.Printf("WebSocket升级错误 [%d]: %v", status, reason)
+	},
 }
 
 func (cp *commonPage) ws(c *gin.Context) {
 	conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
 	if err != nil {
-		mygin.ShowErrorPage(c, mygin.ErrInfo{
-			Code: http.StatusInternalServerError,
-			Title: singleton.Localizer.MustLocalize(&i18n.LocalizeConfig{
-				MessageID: "NetworkError",
-			}),
-			Msg:  "Websocket协议切换失败",
-			Link: "/",
-			Btn:  "返回首页",
-		}, true)
+		// WebSocket升级失败时，不能再写入HTTP响应，因为头部已经被写入
+		// 只记录错误日志，让客户端处理重连
+		log.Printf("WebSocket升级失败: %v", err)
 		return
 	}
 	defer func() {
@@ -1153,15 +1155,8 @@ func (cp *commonPage) terminal(c *gin.Context) {
 
 	wsConn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
 	if err != nil {
-		mygin.ShowErrorPage(c, mygin.ErrInfo{
-			Code: http.StatusInternalServerError,
-			Title: singleton.Localizer.MustLocalize(&i18n.LocalizeConfig{
-				MessageID: "NetworkError",
-			}),
-			Msg:  "Websocket协议切换失败",
-			Link: "/",
-			Btn:  "返回首页",
-		}, true)
+		// WebSocket升级失败时，不能再写入HTTP响应
+		log.Printf("Terminal WebSocket升级失败: %v", err)
 		return
 	}
 	defer wsConn.Close()
@@ -1307,15 +1302,8 @@ func (cp *commonPage) fm(c *gin.Context) {
 
 	wsConn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
 	if err != nil {
-		mygin.ShowErrorPage(c, mygin.ErrInfo{
-			Code: http.StatusInternalServerError,
-			Title: singleton.Localizer.MustLocalize(&i18n.LocalizeConfig{
-				MessageID: "NetworkError",
-			}),
-			Msg:  "Websocket协议切换失败",
-			Link: "/",
-			Btn:  "返回首页",
-		}, true)
+		// WebSocket升级失败时，不能再写入HTTP响应
+		log.Printf("FM WebSocket升级失败: %v", err)
 		return
 	}
 	defer wsConn.Close()

cmd/dashboard/rpc/rpc.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"log"
 	"net"
+	"strings"
 	"time"
 
 	"google.golang.org/grpc"
@@ -15,6 +16,17 @@ import (
 	"github.com/xos/serverstatus/service/singleton"
 )
 
+// isContextCanceledError 检查是否为context canceled错误
+func isContextCanceledError(err error) bool {
+	if err == nil {
+		return false
+	}
+	errStr := err.Error()
+	return strings.Contains(errStr, "context canceled") ||
+		strings.Contains(errStr, "context deadline exceeded") ||
+		strings.Contains(errStr, "Canceled desc = context canceled")
+}
+
 func ServeRPC(port uint) {
 	// 配置 gRPC 服务器选项，防止 goroutine 泄漏和连接问题
 	opts := []grpc.ServerOption{
@@ -105,21 +117,30 @@ func DispatchTask(serviceSentinelDispatchBus <-chan model.Monitor) {
 			}
 			if task.Cover == model.MonitorCoverIgnoreAll && skipServers[currentServer.ID] {
 				if err := currentServer.TaskStream.Send(task.PB()); err != nil {
-					log.Printf("DispatchTask: 发送任务到服务器 %d 失败: %v", currentServer.ID, err)
+					// 只在非context canceled错误时记录日志
+					if !isContextCanceledError(err) {
+						log.Printf("DispatchTask: 发送任务到服务器 %d 失败: %v", currentServer.ID, err)
+					}
 				}
 				workedServerIndex++
 				continue
 			}
 			if task.Cover == model.MonitorCoverAll && !skipServers[currentServer.ID] {
 				if err := currentServer.TaskStream.Send(task.PB()); err != nil {
-					log.Printf("DispatchTask: 发送任务到服务器 %d 失败: %v", currentServer.ID, err)
+					// 只在非context canceled错误时记录日志
+					if !isContextCanceledError(err) {
+						log.Printf("DispatchTask: 发送任务到服务器 %d 失败: %v", currentServer.ID, err)
+					}
 				}
 				workedServerIndex++
 				continue
 			}
 			// 找到合适机器执行任务，跳出循环
 			if err := currentServer.TaskStream.Send(task.PB()); err != nil {
-				log.Printf("DispatchTask: 发送任务到服务器 %d 失败: %v", currentServer.ID, err)
+				// 只在非context canceled错误时记录日志
+				if !isContextCanceledError(err) {
+					log.Printf("DispatchTask: 发送任务到服务器 %d 失败: %v", currentServer.ID, err)
+				}
 			}
 			workedServerIndex++
 			break

service/singleton/singleton.go

@@ -2155,16 +2155,27 @@ func StartDBInsertWorker() {
 // executeDBWriteRequest 执行单个数据库写入请求
 func executeDBWriteRequest(req DBWriteRequest) error {
 	return ExecuteWithRetry(func() error {
-		// 紧急修复：检查DB是否为nil，防止panic
-		if DB == nil {
-			return fmt.Errorf("数据库连接未初始化")
-		}
+		// 根据数据库类型选择不同的处理方式
+		if Conf.DatabaseType == "badger" {
+			// BadgerDB模式：使用BadgerDB操作
+			if db.DB == nil {
+				return fmt.Errorf("BadgerDB连接未初始化")
+			}
 
-		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
-		defer cancel()
+			// 使用BadgerDB的SafeUpdateServerStatus
+			return SafeUpdateServerStatus(req.ServerID, req.Updates)
+		} else {
+			// SQLite模式：使用GORM操作
+			if DB == nil {
+				return fmt.Errorf("SQLite数据库连接未初始化")
+			}
 
-		// 直接更新，避免不必要的事务开销
-		return DB.WithContext(ctx).Table(req.TableName).Where("id = ?", req.ServerID).Updates(req.Updates).Error
+			ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+			defer cancel()
+
+			// 直接更新，避免不必要的事务开销
+			return DB.WithContext(ctx).Table(req.TableName).Where("id = ?", req.ServerID).Updates(req.Updates).Error
+		}
 	})
 }