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

refactor: drop SendMessage() and merge into SendPermanentBroadCastMessage()

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

4
internal/websocket/broadcast.go
View file

@ -24,7 +24,9 @@ func SendPermanentBroadCastMessage() {
mu.Lock() mu.Lock()
for _, client := range Clients { for _, client := range Clients {
// Send message to client // 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 { if err != nil {
helper.Logger.Info("Error broadcasting to client:", "msg", err, "host", client.RealIP, "clients", len(Clients)) helper.Logger.Info("Error broadcasting to client:", "msg", err, "host", client.RealIP, "clients", len(Clients))
} }

6
pkg/models/client.go
View file

@ -21,9 +21,3 @@ type WebsocketClient struct {
LastPong time.Time LastPong time.Time
RealIP string 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)
}