This commit is contained in:
2021-09-24 18:14:38 +08:00
parent e9bbff5f20
commit 0f988a4c3b
3 changed files with 17 additions and 16 deletions
+10 -13
View File
@@ -12,23 +12,21 @@ use Psr\Http\Message\ResponseInterface;
use Psr\Http\Message\ServerRequestInterface; use Psr\Http\Message\ServerRequestInterface;
use Psr\Http\Server\MiddlewareInterface; use Psr\Http\Server\MiddlewareInterface;
use Psr\Http\Server\RequestHandlerInterface; use Psr\Http\Server\RequestHandlerInterface;
use Swoole\Coroutine\Iterator;
abstract class Handler implements RequestHandlerInterface abstract class Handler implements RequestHandlerInterface
{ {
private int $offset = 0;
protected AspectProxy $aspectProxy; protected AspectProxy $aspectProxy;
/** /**
* @param \Http\Handler\Handler $handler * @param CHl $handler
* @param array|null $middlewares * @param Iterator $middlewares
*/ */
public function __construct(public CHl $handler, public ?array $middlewares) public function __construct(public CHl $handler, public Iterator $middlewares)
{ {
$this->aspectProxy = Kiri::getDi()->get(AspectProxy::class); $this->aspectProxy = Kiri::getDi()->get(AspectProxy::class);
} }
@@ -41,16 +39,15 @@ abstract class Handler implements RequestHandlerInterface
*/ */
protected function execute(ServerRequestInterface $request): ResponseInterface protected function execute(ServerRequestInterface $request): ResponseInterface
{ {
if (empty($this->middlewares) || !isset($this->middlewares[$this->offset])) { $middleware = $this->middlewares->current();
if (empty($middleware)) {
return $this->dispatcher($request); return $this->dispatcher($request);
} }
if (!($middleware instanceof MiddlewareInterface)) {
throw new \Exception('get_implements_class($middleware) not found method process.');
}
$middleware = $this->middlewares[$this->offset]; $this->middlewares->next();
if (!($middleware instanceof MiddlewareInterface)) {
throw new \Exception('get_implements_class($middleware) not found method process.');
}
++$this->offset;
return $middleware->process($request, $this); return $middleware->process($request, $this);
} }
+3 -2
View File
@@ -38,11 +38,12 @@ class MiddlewareManager extends BaseObject
if (is_string($middlewares)) { if (is_string($middlewares)) {
$middlewares = [$middlewares]; $middlewares = [$middlewares];
} }
$source = &static::$_middlewares[$class][$method];
foreach ($middlewares as $middleware) { foreach ($middlewares as $middleware) {
if (isset(static::$_middlewares[$class][$method][$middleware])) { if (isset($source[$middleware])) {
continue; continue;
} }
static::$_middlewares[$class][$method][$middleware] = di($middleware); $source[$middleware] = di($middleware);
} }
return true; return true;
} }
+4 -1
View File
@@ -16,6 +16,7 @@ use Http\Message\Stream;
use Kiri\Abstracts\Config; use Kiri\Abstracts\Config;
use Kiri\Exception\ConfigException; use Kiri\Exception\ConfigException;
use Kiri\Kiri; use Kiri\Kiri;
use Kiri\Pool\Helper\SplQueue;
use Psr\Http\Message\ServerRequestInterface; use Psr\Http\Message\ServerRequestInterface;
use Server\Abstracts\Utility\EventDispatchHelper; use Server\Abstracts\Utility\EventDispatchHelper;
use Server\Abstracts\Utility\ResponseHelper; use Server\Abstracts\Utility\ResponseHelper;
@@ -28,6 +29,7 @@ use Server\ExceptionHandlerInterface;
use Server\SInterface\OnCloseInterface; use Server\SInterface\OnCloseInterface;
use Server\SInterface\OnConnectInterface; use Server\SInterface\OnConnectInterface;
use Server\SInterface\OnRequestInterface; use Server\SInterface\OnRequestInterface;
use Swoole\Coroutine\Iterator;
use Swoole\Http\Request; use Swoole\Http\Request;
use Swoole\Http\Response; use Swoole\Http\Response;
use Swoole\Server; use Swoole\Server;
@@ -113,7 +115,8 @@ class Http implements OnCloseInterface, OnConnectInterface, OnRequestInterface
{ {
$middlewares = MiddlewareManager::get($handler->callback); $middlewares = MiddlewareManager::get($handler->callback);
$dispatcher = new Dispatcher($handler, $middlewares);
$dispatcher = new Dispatcher($handler, new Iterator($middlewares));
return $dispatcher->handle($PsrRequest); return $dispatcher->handle($PsrRequest);
} }