comment updates

internal/websocket/handle_connection.go

@@ -36,8 +36,7 @@ func HandleConnection(w http.ResponseWriter, r *http.Request) {
 		LastPong: time.Now(),
 		RealIP:   r.RemoteAddr,
 	}
-
-	client.Conn.SetPongHandler(func(appData string) error {
+	client.Conn.SetPongHandler(func(s string) error {
 		client.LastPong = time.Now()
 		return nil
 	})

pkg/models/client.go

@@ -39,7 +39,7 @@ func (c *WebsocketClient) SendMessage(messageType int, data []byte) error {
 	return nil
 }
 
-// Check if websockets last Pong is recent
+// Checks if the websockets last Pong is recent
 func (c *WebsocketClient) IsStale() bool {
 	return time.Since(c.LastPong) > (TIMEOUT * time.Second)
 }