Send response directly when the handler does not exists. (#3308)

Author: limingxinleo

src/CoreMiddleware.php

@@ -21,6 +21,8 @@
 
 class CoreMiddleware extends HttpCoreMiddleware
 {
+    const HANDLER_NAME = 'class';
+
     /**
      * Handle the response when found.
      */
@@ -42,6 +44,6 @@ protected function handleFound(Dispatched $dispatched, ServerRequestInterface $r
             $response = $response->withHeader(Security::SEC_WEBSOCKET_PROTOCOL, $wsProtocol);
         }
 
-        return $response->withAttribute('class', $controller);
+        return $response->withAttribute(self::HANDLER_NAME, $controller);
     }
 }

src/Exception/Handler/WebSocketExceptionHandler.php

@@ -14,6 +14,7 @@
 use Hyperf\Contract\StdoutLoggerInterface;
 use Hyperf\ExceptionHandler\ExceptionHandler;
 use Hyperf\ExceptionHandler\Formatter\FormatterInterface;
+use Hyperf\HttpMessage\Exception\HttpException;
 use Hyperf\HttpMessage\Stream\SwooleStream;
 use Psr\Http\Message\ResponseInterface;
 use Throwable;
@@ -39,6 +40,9 @@ public function __construct(StdoutLoggerInterface $logger, FormatterInterface $f
     public function handle(Throwable $throwable, ResponseInterface $response)
     {
         $this->logger->warning($this->formatter->format($throwable));
+        if ($throwable instanceof HttpException) {
+            $response = $response->withStatus($throwable->getStatusCode());
+        }
         $stream = new SwooleStream((string) $throwable->getMessage());
         return $response->withBody($stream);
     }

src/Server.php

@@ -163,10 +163,11 @@ public function onHandShake(SwooleRequest $request, SwooleResponse $response): v
             /** @var Response $psr7Response */
             $psr7Response = $this->dispatcher->dispatch($psr7Request, $middlewares, $this->coreMiddleware);
 
-            $class = $psr7Response->getAttribute('class');
+            $class = $psr7Response->getAttribute(CoreMiddleware::HANDLER_NAME);
 
             if (empty($class)) {
-                throw new WebSocketHandeShakeException('WebSocket hande shake failed, because the class does not exists.');
+                $this->logger->warning('WebSocket hande shake failed, because the class does not exists.');
+                return;
             }
 
             FdCollector::set($fd, $class);