diff --git a/internal/websocket/broadcast.go b/internal/websocket/broadcast.go index f9fcab3..57f4449 100644 --- a/internal/websocket/broadcast.go +++ b/internal/websocket/broadcast.go @@ -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)) } diff --git a/pkg/models/client.go b/pkg/models/client.go index 84979ee..9d5dbdb 100644 --- a/pkg/models/client.go +++ b/pkg/models/client.go @@ -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) -}