This commit is contained in:
as2252258@163.com
2021-09-26 00:08:33 +08:00
parent 9b240f7b45
commit 5677cbe254
2 changed files with 31 additions and 20 deletions
+22 -14
View File
@@ -2,6 +2,7 @@
namespace Http\Handler\Abstracts;
use Annotation\Inject;
use Http\Handler\Handler as CHl;
use Http\Message\ServerRequest;
use Kiri\Core\Help;
@@ -19,20 +20,27 @@ abstract class Handler implements RequestHandlerInterface
{
#[Inject(AspectProxy::class)]
protected AspectProxy $aspectProxy;
public CHl $handler;
public ?Iterator $middlewares = null;
protected int $offset = 0;
/**
* @param \Http\Handler\Handler $handler
*/
public function setHandler(CHl $handler): void
{
$this->handler = $handler;
}
/**
* @param CHl $handler
* @param null|Iterator $middlewares
*/
public function __construct(public CHl $handler, public ?Iterator $middlewares)
{
$this->aspectProxy = Kiri::getDi()->get(AspectProxy::class);
}
/**
* @param \Swoole\Coroutine\Iterator|null $middlewares
*/
public function setMiddlewares(?Iterator $middlewares): void
{
$this->middlewares = $middlewares;
}
/**
@@ -42,16 +50,16 @@ abstract class Handler implements RequestHandlerInterface
*/
protected function execute(ServerRequestInterface $request): ResponseInterface
{
if ($this->middlewares->count() < 1) {
if (empty($this->middlewares) || !$this->middlewares->valid()) {
return $this->dispatcher($request);
}
$middleware = $this->middlewares->offsetGet($this->offset);
$middleware = $this->middlewares->current();
if (!($middleware instanceof MiddlewareInterface)) {
throw new \Exception('get_implements_class($middleware) not found method process.');
}
$this->offset++;
$this->middlewares->next();
return $middleware->process($request, $this);
}
@@ -62,7 +70,7 @@ abstract class Handler implements RequestHandlerInterface
* @return mixed
* @throws \Exception
*/
protected function dispatcher(ServerRequestInterface $request): mixed
public function dispatcher(ServerRequestInterface $request): mixed
{
$response = $this->aspectProxy->proxy($this->handler);
if (!($response instanceof ResponseInterface)) {
+9 -6
View File
@@ -47,6 +47,9 @@ class Http implements OnCloseInterface, OnConnectInterface, OnRequestInterface
public Router $router;
public Dispatcher $dispatcher;
/**
* @var ExceptionHandlerInterface
*/
@@ -64,6 +67,7 @@ class Http implements OnCloseInterface, OnConnectInterface, OnRequestInterface
}
$this->exceptionHandler = Kiri::getDi()->get($exceptionHandler);
$this->responseEmitter = Kiri::getDi()->get(ResponseEmitter::class);
$this->dispatcher = Kiri::getDi()->get(Dispatcher::class);
}
@@ -113,12 +117,11 @@ class Http implements OnCloseInterface, OnConnectInterface, OnRequestInterface
protected function handler(Handler $handler, $PsrRequest): \Psr\Http\Message\ResponseInterface
{
$middlewares = MiddlewareManager::get($handler->callback);
if ($middlewares instanceof Iterator) {
$dispatcher = new Dispatcher($handler, $middlewares);
} else {
$dispatcher = new Dispatcher($handler, new Iterator());
}
return $dispatcher->handle($PsrRequest);
$this->dispatcher->setHandler($handler);
if (!empty($middlewares)) {
$this->dispatcher->setMiddlewares($middlewares);
}
return $this->dispatcher->handle($PsrRequest);
}