Update Websocket.php

localzet · Aug 21, 2024 · 1bb56fe · 1bb56fe
1 parent 26f5591
commit 1bb56fe
Showing 1 changed file with 23 additions and 1 deletion.
diff --git a/src/Server/Protocols/Websocket.php b/src/Server/Protocols/Websocket.php
@@ -66,6 +66,27 @@ class Websocket
      */
     public const BINARY_TYPE_ARRAYBUFFER = "\x82";
 
+    /**
+     * Имя класса Request.
+     *
+     * @var string
+     */
+    protected static string $requestClass = Request::class;
+
+    /**
+     * Получить или установить имя класса Request для рукопожатия.
+     *
+     * @param string|null $className
+     * @return string
+     */
+    public static function requestClass(string $className = null): string
+    {
+        if ($className !== null) {
+            static::$requestClass = $className;
+        }
+        return static::$requestClass;
+    }
+
     /**
      * Проверка целостности пакета.
      *
@@ -330,7 +351,8 @@ public static function dealHandshake(string $buffer, TcpConnection $connection):
             $onWebSocketConnect = $connection->onWebSocketConnect ?? $connection->server->onWebSocketConnect ?? false;
             if ($onWebSocketConnect) {
                 try {
-                    $request = new Request($buffer);
+                    /** @var Request $request */
+                    $request = new static::$requestClass($buffer);
                     $request->connection = $connection;
                     $connection->request = $request;
                     $onWebSocketConnect($connection, $request);