Update.

Author: xOS

cmd/dashboard/controller/common_page.go

@@ -416,22 +416,21 @@ func (cp *commonPage) network(c *gin.Context) {
 		servers = make([]*model.Server, 0)
 	}
 
+	// 性能优化：使用map进行O(1)查找，避免双重循环
+	monitorServerMap := make(map[uint64]bool)
+	for _, serverID := range serverIdsWithMonitor {
+		monitorServerMap[serverID] = true
+	}
+
 	if authorized {
 		// 有权限的用户可以访问所有服务器
 		if singleton.SortedServerList != nil {
 			for _, server := range singleton.SortedServerList {
 				// 确保服务器不为nil
 				if server != nil {
 					// 如果serverIdsWithMonitor为空或包含当前服务器ID，则添加
-					if len(serverIdsWithMonitor) == 0 {
+					if len(serverIdsWithMonitor) == 0 || monitorServerMap[server.ID] {
 						servers = append(servers, server)
-					} else {
-						for _, monitorServerID := range serverIdsWithMonitor {
-							if server.ID == monitorServerID {
-								servers = append(servers, server)
-								break
-							}
-						}
 					}
 				}
 			}
@@ -445,15 +444,8 @@ func (cp *commonPage) network(c *gin.Context) {
 				// 确保服务器不为nil
 				if server != nil {
 					// 如果serverIdsWithMonitor为空或包含当前服务器ID，则添加
-					if len(serverIdsWithMonitor) == 0 {
+					if len(serverIdsWithMonitor) == 0 || monitorServerMap[server.ID] {
 						servers = append(servers, server)
-					} else {
-						for _, monitorServerID := range serverIdsWithMonitor {
-							if server.ID == monitorServerID {
-								servers = append(servers, server)
-								break
-							}
-						}
 					}
 				}
 			}

cmd/dashboard/main.go

@@ -197,36 +197,49 @@ func main() {
 		// 等待所有任务完成
 		done := make(chan struct{})
 		go func() {
+			defer close(done)
 			log.Println("程序正在优雅关闭，保存所有数据...")
 
-			// 保存流量数据
-			singleton.RecordTransferHourlyUsage()
-			singleton.SaveAllTrafficToDB()
-
-			// 保存所有数据到数据库（BadgerDB模式下特别重要）
-			singleton.SaveAllDataToDB()
-
-			// 清理Goroutine池，防止内存泄漏
-			singleton.CleanupGoroutinePools()
+			// 1. 首先停止所有定时任务，防止新的goroutine产生
+			if singleton.Cron != nil {
+				singleton.Cron.Stop()
+				log.Println("定时任务已停止")
+			}
 
-			// 关闭所有WebSocket连接
-			singleton.ServerLock.RLock()
+			// 2. 关闭所有WebSocket连接，防止新的连接
+			singleton.ServerLock.Lock()
 			for _, server := range singleton.ServerList {
-				if server != nil && server.TaskClose != nil {
-					select {
-					case server.TaskClose <- fmt.Errorf("server shutting down"):
-					default:
+				if server != nil && server.TaskCloseLock != nil {
+					server.TaskCloseLock.Lock()
+					if server.TaskClose != nil {
+						select {
+						case server.TaskClose <- fmt.Errorf("server shutting down"):
+						default:
+						}
+						// 不要在这里close，让RequestTask自己处理
+						server.TaskClose = nil
 					}
-					close(server.TaskClose)
+					server.TaskStream = nil
+					server.TaskCloseLock.Unlock()
 				}
 			}
-			singleton.ServerLock.RUnlock()
+			singleton.ServerLock.Unlock()
+			log.Println("所有服务器连接已关闭")
+
+			// 3. 清理Goroutine池，防止内存泄漏
+			singleton.CleanupGoroutinePools()
+
+			// 4. 保存流量数据
+			singleton.RecordTransferHourlyUsage()
+			singleton.SaveAllTrafficToDB()
+
+			// 5. 保存所有数据到数据库（BadgerDB模式下特别重要）
+			singleton.SaveAllDataToDB()
 
-			// 关闭HTTP服务器
+			// 6. 关闭HTTP服务器
 			srv.Shutdown(c)
 
 			log.Println("所有数据已保存，程序关闭完成")
-			close(done)
 		}()
 
 		// 设置超时