Update.

Author: xOS

cmd/dashboard/controller/member_api.go

@@ -131,14 +131,17 @@ func (ma *memberAPI) issueNewToken(c *gin.Context) {
 	if singleton.Conf.DatabaseType == "badger" {
 		// 为新API令牌生成ID
 		token.ID = uint64(time.Now().UnixNano())
-		err = db.DB.SaveModel("api_token", token.ID, token)
+		// 使用ApiTokenOps保存
+		apiTokenOps := db.NewApiTokenOps(db.DB)
+		err = apiTokenOps.SaveApiToken(token)
 		if err != nil {
 			c.JSON(http.StatusOK, model.Response{
 				Code:    http.StatusBadRequest,
 				Message: fmt.Sprintf("保存API令牌失败：%s", err),
 			})
 			return
 		}
+		log.Printf("API Token已保存到BadgerDB: %s (Note: %s)", token.Token, token.Note)
 	} else if singleton.DB != nil {
 		err = singleton.DB.Create(token).Error
 		if err != nil {
@@ -618,7 +621,12 @@ func (ma *memberAPI) addOrEditServer(c *gin.Context) {
 
 				// BadgerDB 模式下使用 BadgerDB 操作
 				if singleton.Conf.DatabaseType == "badger" {
-					err = db.DB.SaveModel("server", s.ID, &s)
+					// 使用ServerOps来确保Secret字段正确保存
+					serverOps := db.NewServerOps(db.DB)
+					err = serverOps.SaveServer(&s)
+					if err == nil {
+						log.Printf("服务器 %d 的Secret已保存到BadgerDB: %s", s.ID, s.Secret)
+					}
 				} else if singleton.DB != nil {
 					err = singleton.DB.Save(&s).Error
 				} else {
@@ -1000,13 +1008,16 @@ func (ma *memberAPI) batchUpdateServerGroup(c *gin.Context) {
 			singleton.ServerLock.RUnlock()
 
 			server.Tag = req.Group
-			if err := db.DB.SaveModel("server", serverID, &server); err != nil {
+			// 使用ServerOps保存到数据库，确保Secret字段正确保存
+			serverOps := db.NewServerOps(db.DB)
+			if err := serverOps.SaveServer(&server); err != nil {
 				c.JSON(http.StatusOK, model.Response{
 					Code:    http.StatusBadRequest,
 					Message: fmt.Sprintf("更新服务器 %d 失败: %v", serverID, err),
 				})
 				return
 			}
+			log.Printf("批量更新服务器 %d 的Secret已保存: %s", serverID, server.Secret)
 		}
 	} else if singleton.DB != nil {
 		if err := singleton.DB.Model(&model.Server{}).Where("id in (?)", req.Servers).Update("tag", req.Group).Error; err != nil {

service/rpc/server.go

@@ -175,9 +175,11 @@ func (s *ServerHandler) RequestTask(h *pb.Host, stream pb.ServerService_RequestT
 			// 连接断开时清理资源，增强错误处理
 			if ctxErr := ctx.Err(); ctxErr != nil {
 				// 检查是否为网络连接错误或超时错误
-				if isConnectionError(ctxErr) || ctxErr == context.DeadlineExceeded {
-					// 静默处理网络连接错误和正常超时，避免日志干扰
-					log.Printf("RequestTask连接正常断开: 客户端ID %d", clientID)
+				if isConnectionError(ctxErr) || ctxErr == context.DeadlineExceeded || ctxErr == context.Canceled {
+					// 静默处理网络连接错误、正常超时和取消，避免日志干扰
+					if singleton.Conf.Debug {
+						log.Printf("RequestTask连接正常断开: 客户端ID %d, 原因: %v", clientID, ctxErr)
+					}
 				} else {
 					log.Printf("RequestTask连接上下文错误: %v", ctxErr)
 				}
@@ -236,15 +238,20 @@ func (s *ServerHandler) RequestTask(h *pb.Host, stream pb.ServerService_RequestT
 	case <-ctx.Done():
 		// 超时或连接取消，增强错误分类
 		if ctx.Err() == context.DeadlineExceeded {
-			log.Printf("客户端 %d RequestTask连接超时 (20分钟)", clientID)
-			return fmt.Errorf("request task timeout after 20 minutes")
+			if singleton.Conf.Debug {
+				log.Printf("客户端 %d RequestTask连接超时 (20分钟)", clientID)
+			}
+			return nil // 将超时视为正常断开，不返回错误
 		}
-		// 检查是否为网络连接错误
-		if isConnectionError(ctx.Err()) {
-			log.Printf("客户端 %d 网络连接异常: %v", clientID, ctx.Err())
-			return nil // 将网络连接错误视为正常断开
+		// 检查是否为网络连接错误或取消
+		if isConnectionError(ctx.Err()) || ctx.Err() == context.Canceled {
+			if singleton.Conf.Debug {
+				log.Printf("客户端 %d 连接正常断开: %v", clientID, ctx.Err())
+			}
+			return nil // 将网络连接错误和取消视为正常断开
 		}
-		return ctx.Err()
+		log.Printf("客户端 %d RequestTask异常断开: %v", clientID, ctx.Err())
+		return nil // 即使异常也不返回错误，避免程序重启
 	}
 }
 

service/singleton/servicesentinel.go

@@ -530,6 +530,9 @@ func (ss *ServiceSentinel) handleServiceReport(r ReportData) {
 						err := monitorOps.SaveMonitorHistory(h)
 						if err != nil {
 							log.Printf("NG>> BadgerDB TCP/ICMP监控数据保存失败 (MonitorID: %d): %v", h.MonitorID, err)
+						} else {
+							log.Printf("BadgerDB TCP/ICMP监控数据已保存 (MonitorID: %d, ServerID: %d, Delay: %.2f)",
+								h.MonitorID, h.ServerID, h.AvgDelay)
 						}
 					}
 				}(history)