feat: https://github.com/komari-monitor/komari/issues/108 实现应用层的心跳机制

git忽略__debug

.gitignore

@@ -1,3 +1,4 @@
+__debug*
 agent.json
 .vscode/
 komari-agent.exe

server/websocket.go

@@ -33,11 +33,15 @@ func EstablishWebSocketConnection() {
 		interval = flags.Interval - 1
 	}
 
-	ticker := time.NewTicker(time.Duration(interval * float64(time.Second)))
+	dataTicker := time.NewTicker(time.Duration(interval * float64(time.Second)))
-	defer ticker.Stop()
+	defer dataTicker.Stop()
 
-	for range ticker.C {
+	heartbeatTicker := time.NewTicker(30 * time.Second)
-		// If no connection, attempt to connect
+	defer heartbeatTicker.Stop()
+
+	for {
+		select {
+		case <-dataTicker.C:
 			if conn == nil {
 				log.Println("Attempting to connect to WebSocket...")
 				retry := 0
@@ -71,6 +75,16 @@ func EstablishWebSocketConnection() {
 				conn = nil // Mark connection as dead
 				continue
 			}
+		case <-heartbeatTicker.C:
+			if conn != nil {
+				err := conn.WriteMessage(websocket.PingMessage, nil)
+				if err != nil {
+					log.Println("Failed to send heartbeat:", err)
+					conn.Close()
+					conn = nil // Mark connection as dead
+				}
+			}
+		}
 	}
 }