From e7e06cf64632455784cfb9db27e80af31b9bfa5a Mon Sep 17 00:00:00 2001 From: trafficlunar Date: Fri, 29 Nov 2024 19:18:06 +0000 Subject: [PATCH] fix: computer online not changing when websocket disconnected --- internal/handler/computer.go | 6 +++--- internal/service/computer.go | 3 +-- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/internal/handler/computer.go b/internal/handler/computer.go index 313059c..cf130c5 100644 --- a/internal/handler/computer.go +++ b/internal/handler/computer.go @@ -26,14 +26,14 @@ func HandleComputerWebSocket(w http.ResponseWriter, r *http.Request) { defer conn.Close() slog.Info("Websocket connection established!") - online := true + service.ComputerData.Online = true // Read messages for { _, message, err := conn.ReadMessage() if err != nil { slog.Error("WebSocket connection closed by client", slog.Any("error", err)) - online = false + service.ComputerData.Online = false break } @@ -43,7 +43,7 @@ func HandleComputerWebSocket(w http.ResponseWriter, r *http.Request) { continue } - service.AddComputerData(online, clientMessage) + service.AddComputerData(clientMessage) slog.Info("Recieved message", slog.Any("message", clientMessage)) } } diff --git a/internal/service/computer.go b/internal/service/computer.go index 84e910d..d6e3462 100644 --- a/internal/service/computer.go +++ b/internal/service/computer.go @@ -6,8 +6,7 @@ import ( var ComputerData model.ComputerGraphData -func AddComputerData(online bool, clientMessage model.ComputerWebSocketMessage) { - ComputerData.Online = online +func AddComputerData(clientMessage model.ComputerWebSocketMessage) { ComputerData.Cpu = append(ComputerData.Cpu, int(clientMessage.Cpu)) ComputerData.Ram = append(ComputerData.Ram, int(clientMessage.Ram))