add WebSocketHandler to WebSocket struct

server/internal/websocket/socket.go

@@ -10,7 +10,7 @@ import (
 
 type WebSocket struct {
 	id         string
-	ws         WebSocketHandler
+	ws         *WebSocketHandler
 	connection *websocket.Conn
 	mu         sync.Mutex
 }

server/internal/websocket/websocket.go

@@ -142,6 +142,7 @@ func (ws *WebSocketHandler) Upgrade(w http.ResponseWriter, r *http.Request) erro
 
 	socket := &WebSocket{
 		id:         id,
+		ws:         ws,
 		connection: connection,
 	}