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()
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))
}
}

3
internal/service/computer.go
View file

@ -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))