This commit is contained in:
2021-09-16 14:19:05 +08:00
parent b94efaa63a
commit ad6aa3e7b5
3 changed files with 57 additions and 103 deletions
+54 -102
View File
@@ -5,7 +5,6 @@ namespace Http\Route;
use Closure; use Closure;
use Http\IInterface\MiddlewareInterface;
use Kiri\Abstracts\BaseObject; use Kiri\Abstracts\BaseObject;
@@ -16,112 +15,65 @@ use Kiri\Abstracts\BaseObject;
class MiddlewareManager extends BaseObject class MiddlewareManager extends BaseObject
{ {
private static array $_middlewares = []; private static array $_middlewares = [];
/** /**
* @param $class * @param $class
* @param $method * @param $method
* @param array|string $middlewares * @param array|string $middlewares
*/ * @return bool
public static function addMiddlewares($class, $method, array|string $middlewares) */
{ public static function add($class, $method, array|string $middlewares): bool
if (is_object($class)) { {
$class = $class::class; if (is_object($class)) {
} $class = $class::class;
if (!isset(static::$_middlewares[$class . '::' . $method])) { }
static::$_middlewares[$class . '::' . $method] = []; if (!isset(static::$_middlewares[$class . '::' . $method])) {
} static::$_middlewares[$class . '::' . $method] = [];
}
if (is_string($middlewares) && !in_array($middlewares, static::$_middlewares[$class . '::' . $method])) { if (is_string($middlewares) && !in_array($middlewares, static::$_middlewares[$class . '::' . $method])) {
static::$_middlewares[$class . '::' . $method][] = $middlewares; static::$_middlewares[$class . '::' . $method][] = $middlewares;
return; } else {
} foreach ($middlewares as $middleware) {
foreach ($middlewares as $middleware) { if (in_array($middleware, static::$_middlewares[$class . '::' . $method])) {
if (in_array($middleware, static::$_middlewares[$class . '::' . $method])) { continue;
continue; }
} static::$_middlewares[$class . '::' . $method][] = $middleware;
static::$_middlewares[$class . '::' . $method][] = $middleware; }
} }
} return true;
}
/** /**
* @param $handler * @param $handler
* @return mixed|null * @return mixed
*/ */
public static function get($handler) public static function get($handler): mixed
{ {
if ($handler instanceof Closure) { if ($handler instanceof Closure) {
return null; return null;
} }
[$class, $method] = [$handler[0]::class, $handler[1]]; [$class, $method] = [$handler[0]::class, $handler[1]];
if (!static::hasMiddleware($class, $method)) { if (!static::hasMiddleware($class, $method)) {
return null; return null;
} }
return static::$_middlewares[$class . '::' . $method]; return static::$_middlewares[$class . '::' . $method];
} }
/** /**
* @param $class * @param $class
* @param $method * @param $method
* @return bool * @return bool
*/ */
public static function hasMiddleware($class, $method): bool public static function hasMiddleware($class, $method): bool
{ {
if (is_object($class)) { if (is_object($class)) {
$class = $class::class; $class = $class::class;
} }
return isset(static::$_middlewares[$class . '::' . $method]); return isset(static::$_middlewares[$class . '::' . $method]);
} }
/**
* @param $class
* @param $method
* @param $caller
* @return mixed
*/
public static function callerMiddlewares($class, $method, $caller): mixed
{
if (is_object($class)) {
$class = $class::class;
}
$middlewares = static::$_middlewares[$class . '::' . $method] ?? [];
if (empty($middlewares)) {
return $caller;
}
return array_reduce(array_reverse($middlewares), function ($stack, $pipe) {
return function ($passable) use ($stack, $pipe) {
if ($pipe instanceof MiddlewareInterface) {
$pipe = [$pipe, 'onHandler'];
}
return call_user_func($pipe, $passable, $stack);
};
}, $caller);
}
/**
* @param $middlewares
* @param Closure $caller
* @return Closure
*/
public static function closureMiddlewares($middlewares, Closure $caller): Closure
{
if (empty($middlewares)) {
return $caller;
}
return array_reduce(array_reverse($middlewares), function ($stack, $pipe) {
return function ($passable) use ($stack, $pipe) {
if ($pipe instanceof MiddlewareInterface) {
$pipe = [$pipe, 'onHandler'];
}
return call_user_func($pipe, $passable, $stack);
};
}, $caller);
}
} }
+2
View File
@@ -52,6 +52,8 @@ class Http extends \Server\Abstracts\Http implements OnClose, OnConnect
} catch (Error | \Throwable $exception) { } catch (Error | \Throwable $exception) {
$psr7Response = $this->exceptionHandler->emit($exception, $this->response); $psr7Response = $this->exceptionHandler->emit($exception, $this->response);
} finally { } finally {
$request_time_float = $request->server['request_time_float'] - $request->server['request_time'];
$psr7Response->withHeader('Run-Time', microtime(true) - $request_time_float);
$this->responseEmitter->sender($response, $psr7Response); $this->responseEmitter->sender($response, $psr7Response);
$this->eventDispatch->dispatch(new OnAfterRequest()); $this->eventDispatch->dispatch(new OnAfterRequest());
} }
+1 -1
View File
@@ -46,7 +46,7 @@ use Http\IInterface\MiddlewareInterface;
*/ */
public function execute(mixed $class, mixed $method = null): mixed public function execute(mixed $class, mixed $method = null): mixed
{ {
MiddlewareManager::addMiddlewares($class, $method, $this->middleware); MiddlewareManager::add($class, $method, $this->middleware);
return parent::execute($class, $method); return parent::execute($class, $method);
} }