Fixed bug that WsContext does not work in onOpen. (#4614)

Author: limingxinleo

src/Server.php

@@ -150,13 +150,13 @@ public function onHandShake($request, $response): void
 
                 // TODO: Support SWOW
                 $response instanceof SwooleResponse && $this->getSender()->setResponse($fd, $response);
-                $this->deferOnOpen($request, $class, $response);
+                $this->deferOnOpen($request, $class, $response, $fd);
 
                 $upgrade->on(WebSocket::ON_MESSAGE, $this->getOnMessageCallback());
                 $upgrade->on(WebSocket::ON_CLOSE, $this->getOnCloseCallback());
                 $upgrade->start();
             } else {
-                $this->deferOnOpen($request, $class, $server);
+                $this->deferOnOpen($request, $class, $server, $fd);
             }
         } catch (Throwable $throwable) {
             // Delegate the exception to exception handler.
@@ -234,10 +234,11 @@ protected function getFd($response): int
     /**
      * @param mixed $request
      */
-    protected function deferOnOpen($request, string $class, SwooleResponse|WebSocketServer $server)
+    protected function deferOnOpen($request, string $class, SwooleResponse|WebSocketServer $server, int $fd)
     {
         $instance = $this->container->get($class);
-        wait(static function () use ($request, $instance, $server) {
+        wait(static function () use ($request, $instance, $server, $fd) {
+            Context::set(WsContext::FD, $fd);
             if ($instance instanceof OnOpenInterface) {
                 $instance->onOpen($server, $request);
             }

tests/ServerTest.php

@@ -54,7 +54,7 @@ public function testDeferOnOpenInCoroutineStyleServer()
         );
 
         $server = new ClassInvoker($server);
-        $server->deferOnOpen(new SwooleRequest(), WebSocketStub::class, new SwooleResponse());
+        $server->deferOnOpen(new SwooleRequest(), WebSocketStub::class, new SwooleResponse(), 1);
         $this->assertNotEquals(Coroutine::id(), WebSocketStub::$coroutineId);
         $this->assertFalse(\Swoole\Coroutine::exists(WebSocketStub::$coroutineId));
     }