CoreMiddleware.php 2.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  1. <?php
  2. declare(strict_types=1);
  3. /**
  4. * This file is part of Hyperf.
  5. *
  6. * @link https://www.hyperf.io
  7. * @document https://hyperf.wiki
  8. * @contact group@hyperf.io
  9. * @license https://github.com/hyperf/hyperf/blob/master/LICENSE
  10. */
  11. namespace Hyperf\JsonRpc;
  12. use Closure;
  13. use Hyperf\HttpServer\Router\Dispatched;
  14. use Hyperf\Rpc\Protocol;
  15. use InvalidArgumentException;
  16. use Psr\Container\ContainerInterface;
  17. use Psr\Http\Message\ServerRequestInterface;
  18. use Swow\Psr7\Message\ResponsePlusInterface;
  19. use Throwable;
  20. class CoreMiddleware extends \Hyperf\RpcServer\CoreMiddleware
  21. {
  22. protected ResponseBuilder $responseBuilder;
  23. public function __construct(ContainerInterface $container, Protocol $protocol, ResponseBuilder $builder, string $serverName)
  24. {
  25. parent::__construct($container, $protocol, $serverName);
  26. $this->responseBuilder = $builder;
  27. }
  28. protected function handleFound(Dispatched $dispatched, ServerRequestInterface $request): mixed
  29. {
  30. if ($dispatched->handler->callback instanceof Closure) {
  31. $callback = $dispatched->handler->callback;
  32. $response = $callback();
  33. } else {
  34. [$controller, $action] = $this->prepareHandler($dispatched->handler->callback);
  35. $controllerInstance = $this->container->get($controller);
  36. if (! method_exists($controller, $action)) {
  37. // Route found, but the handler does not exist.
  38. return $this->responseBuilder->buildErrorResponse($request, ResponseBuilder::INTERNAL_ERROR);
  39. }
  40. try {
  41. $parameters = $this->parseMethodParameters($controller, $action, $request->getParsedBody());
  42. } catch (InvalidArgumentException) {
  43. return $this->responseBuilder->buildErrorResponse($request, ResponseBuilder::INVALID_PARAMS);
  44. }
  45. try {
  46. $response = $controllerInstance->{$action}(...$parameters);
  47. } catch (Throwable $exception) {
  48. $response = $this->responseBuilder->buildErrorResponse($request, ResponseBuilder::SERVER_ERROR, $exception);
  49. $this->responseBuilder->persistToContext($response);
  50. throw $exception;
  51. }
  52. }
  53. return $response;
  54. }
  55. protected function handleNotFound(ServerRequestInterface $request): mixed
  56. {
  57. return $this->responseBuilder->buildErrorResponse($request, ResponseBuilder::METHOD_NOT_FOUND);
  58. }
  59. protected function handleMethodNotAllowed(array $methods, ServerRequestInterface $request): mixed
  60. {
  61. return $this->handleNotFound($request);
  62. }
  63. protected function transferToResponse($response, ServerRequestInterface $request): ResponsePlusInterface
  64. {
  65. return $this->responseBuilder->buildResponse($request, $response);
  66. }
  67. }