pomodoro/GoTomato
3
0
Fork 0
Code Issues Pull requests Releases 15 Activity

refactor: merge SendMessage() into SendPermanentBroadCastMessage()

Browse source
This commit is contained in:
Sebastian Mark 2024-11-21 08:43:02 +01:00
parent c31b994d58
commit f83cf5bcc9
2 changed files with 2 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
internal/websocket/broadcast.go
View file

@ -25,7 +25,8 @@ func SendPermanentBroadCastMessage() {
mu.Lock()
for _, client := range Clients {
// Send message to client
err := client.SendMessage(websocket.TextMessage, jsonMessage)
client.Conn.SetWriteDeadline(time.Now().Add(SEND_TIMEOUT * time.Second))
err := client.Conn.WriteMessage(websocket.TextMessage, jsonMessage)
if err != nil {
helper.Logger.Error("Error broadcasting to client:", "msg", err, "host", client.RealIP, "clients", len(Clients))
}

6
pkg/models/client.go
View file

@ -19,9 +19,3 @@ type WebsocketClient struct {
LastPong time.Time
RealIP string
}
// Sends a message to the websocket.
func (c *WebsocketClient) SendMessage(messageType int, data []byte) error {
c.Conn.SetWriteDeadline(time.Now().Add(TIMEOUT * time.Second))
return c.Conn.WriteMessage(messageType, data)
}