Changes

Author: DikoIbragimov

src/Application/Websocket/Application.php

@@ -100,7 +100,6 @@ public function onMessage(TcpConnection $tcpConnection, string $payload): void {
         try {
             $this->container->get(OnMessageInterface::class)->handle($client, $this->hydrator->create(Message::class, $payloadArray));
         } catch (Throwable $exception) {
-            print $exception->getMessage() . "\n";
             $this->logger->error($exception->getMessage(), [
                 'payload' => $payloadArray,
                 'exception' => $exception,
@@ -111,11 +110,15 @@ public function onMessage(TcpConnection $tcpConnection, string $payload): void {
     /**
      * @param TcpConnection $tcpConnection
      * @throws ContainerExceptionInterface
-     * @throws NotFoundException
      * @throws NotFoundExceptionInterface
      */
     public function onClose(TcpConnection $tcpConnection): void {
-        $client = $this->collection->get($tcpConnection->id);
+        try {
+            $client = $this->collection->get($tcpConnection->id);
+        } catch (NotFoundException $e) {
+            print_r("Client not found on close: " . $tcpConnection->id . "\n");
+            return;
+        }
         $this->collection->remove($client->id);
         $this->container->get(OnCloseInterface::class)->handle($client);
     }

src/Application/Websocket/Event/OnConnect.php

@@ -28,7 +28,6 @@ public function handle(Client $client): void {
         try {
             $this->registerService->register($client);
         } catch (Throwable $e) {
-            print_r($e->getMessage());
             $client->connection->close(['error' => 'Server error occurred during registration.']);
             return;
         }

src/Application/Websocket/Service/Brain/ReflectionService.php

@@ -28,6 +28,8 @@ public function handle(Client $client, Message $payload): void {
 
         $code = $this->brain->reflection($client, $payload->payload['query']);
 
+        print_r($code);
+
         if (!str_ends_with($code, ';')) {
             $code .= ';';
         }