add timeout consts in new file

internal/websocket/broadcast.go

@@ -12,6 +12,7 @@ import (
 // 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)

internal/websocket/staleClients.go

@@ -8,7 +8,7 @@ import (
 )
 
 func sendPing(client *models.WebsocketClient) bool {
-	client.Conn.SetWriteDeadline(time.Now().Add(10 * time.Second))
+	client.Conn.SetWriteDeadline(time.Now().Add(SEND_TIMEOUT * time.Second))
 	err := client.Conn.WriteMessage(websocket.PingMessage, nil)
 
 	if err != nil {
@@ -19,12 +19,12 @@ func sendPing(client *models.WebsocketClient) bool {
 }
 
 func isStale(client *models.WebsocketClient) bool {
-	return time.Since(client.LastPong) > (90 * time.Second)
+	return time.Since(client.LastPong) > (STALE_CLIENT_TIMEOUT * time.Second)
 }
 
 // Check and remove stale clients
 func RemoveStaleClients() {
-	ticker := time.NewTicker(30 * time.Second)
+	ticker := time.NewTicker(CHECK_INTERVALL * time.Second)
 	defer ticker.Stop()
 
 	for range ticker.C {

internal/websocket/timeouts.go

@@ -0,0 +1,5 @@
+package websocket
+
+const SEND_TIMEOUT = 10
+const STALE_CLIENT_TIMEOUT = 90
+const CHECK_INTERVALL = 30

pkg/models/client.go

@@ -6,8 +6,6 @@ import (
 	"github.com/gorilla/websocket"
 )
 
-const TIMEOUT = 10
-
 // Represents a command from the client (start/stop)
 type ClientCommand struct {
 	Command  string         `json:"command"`  // Command send to the server