diff --git a/internal/websocket/staleClients.go b/internal/websocket/staleClients.go index 6aed573..5bb5047 100644 --- a/internal/websocket/staleClients.go +++ b/internal/websocket/staleClients.go @@ -8,21 +8,6 @@ import ( "github.com/gorilla/websocket" ) -func sendPing(client *models.WebsocketClient) bool { - client.Conn.SetWriteDeadline(time.Now().Add(SEND_TIMEOUT * time.Second)) - err := client.Conn.WriteMessage(websocket.PingMessage, nil) - - if err != nil { - return false - } - - return true -} - -func isStale(client *models.WebsocketClient) bool { - return time.Since(client.LastPong) > (STALE_CLIENT_TIMEOUT * time.Second) -} - // Check and remove stale clients func RemoveStaleClients() { ticker := time.NewTicker(STALE_CHECK_INTERVALL * time.Second) @@ -40,3 +25,18 @@ func RemoveStaleClients() { mu.Unlock() } } + +func sendPing(client *models.WebsocketClient) bool { + client.Conn.SetWriteDeadline(time.Now().Add(SEND_TIMEOUT * time.Second)) + err := client.Conn.WriteMessage(websocket.PingMessage, nil) + + if err != nil { + return false + } + + return true +} + +func isStale(client *models.WebsocketClient) bool { + return time.Since(client.LastPong) > (STALE_CLIENT_TIMEOUT * time.Second) +}