40 lines
1.2 KiB
Go
40 lines
1.2 KiB
Go
package websocket
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/gorilla/websocket"
|
|
"time"
|
|
|
|
"git.smsvc.net/pomodoro/GoTomato/internal/helper"
|
|
"git.smsvc.net/pomodoro/GoTomato/internal/shared"
|
|
)
|
|
|
|
// Sends continous messages to all connected WebSocket clients
|
|
func SendPermanentBroadCastMessage() {
|
|
tick := time.NewTicker(time.Second)
|
|
for {
|
|
// Marshal the message into JSON format
|
|
jsonMessage, err := json.Marshal(shared.State)
|
|
if err != nil {
|
|
helper.Logger.Error("Error marshalling message:", "msg", err)
|
|
return
|
|
}
|
|
|
|
// Iterate over all connected clients and broadcast the message
|
|
for _, client := range Clients {
|
|
err := client.SendMessage(websocket.TextMessage, jsonMessage)
|
|
if err != nil {
|
|
helper.Logger.Info("Error broadcasting to client:", "msg", err, "host", client.RealIP, "clients", len(Clients))
|
|
}
|
|
}
|
|
for _, client := range Clients {
|
|
if time.Now().Sub(client.LastPong) > (10 * time.Second) {
|
|
// Remove unresponsive client
|
|
client.Conn.Close()
|
|
delete(Clients, client.Conn.LocalAddr())
|
|
helper.Logger.Info("Client timed out", "lastPong", client.LastPong, "host", client.RealIP, "clients", len(Clients))
|
|
}
|
|
}
|
|
<-tick.C
|
|
}
|
|
}
|