trafficlunar/api
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: computer online not changing when websocket disconnected

Browse source
This commit is contained in:
trafficlunar 2024-11-29 19:18:06 +00:00
parent 58adca9cd0
commit e7e06cf646
2 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
internal/handler/computer.go
View file

@ -26,14 +26,14 @@ func HandleComputerWebSocket(w http.ResponseWriter, r *http.Request) {
defer conn.Close() defer conn.Close()
slog.Info("Websocket connection established!") slog.Info("Websocket connection established!")
online := true service.ComputerData.Online = true
// Read messages // Read messages
for { for {
_, message, err := conn.ReadMessage() _, message, err := conn.ReadMessage()
if err != nil { if err != nil {
slog.Error("WebSocket connection closed by client", slog.Any("error", err)) slog.Error("WebSocket connection closed by client", slog.Any("error", err))
online = false service.ComputerData.Online = false
break break
} }
@ -43,7 +43,7 @@ func HandleComputerWebSocket(w http.ResponseWriter, r *http.Request) {
continue continue
} }
service.AddComputerData(online, clientMessage) service.AddComputerData(clientMessage)
slog.Info("Recieved message", slog.Any("message", clientMessage)) slog.Info("Recieved message", slog.Any("message", clientMessage))
} }
} }

3
internal/service/computer.go
View file

@ -6,8 +6,7 @@ import (
var ComputerData model.ComputerGraphData var ComputerData model.ComputerGraphData
func AddComputerData(online bool, clientMessage model.ComputerWebSocketMessage) { func AddComputerData(clientMessage model.ComputerWebSocketMessage) {
ComputerData.Online = online
ComputerData.Cpu = append(ComputerData.Cpu, int(clientMessage.Cpu)) ComputerData.Cpu = append(ComputerData.Cpu, int(clientMessage.Cpu))
ComputerData.Ram = append(ComputerData.Ram, int(clientMessage.Ram)) ComputerData.Ram = append(ComputerData.Ram, int(clientMessage.Ram))