This commit is contained in:
2020-12-15 18:56:23 +08:00
parent dc596b642c
commit d0750a95a7
+6 -4
View File
@@ -395,9 +395,6 @@ class Node extends Application
*/ */
public function addMiddleware(Closure|string|array $class) public function addMiddleware(Closure|string|array $class)
{ {
if (!is_callable($class, true)) {
return;
}
if (is_string($class)) { if (is_string($class)) {
$class = Snowflake::createObject($class); $class = Snowflake::createObject($class);
if (!($class instanceof \HttpServer\IInterface\Middleware)) { if (!($class instanceof \HttpServer\IInterface\Middleware)) {
@@ -405,7 +402,12 @@ class Node extends Application
} }
$class = [$class, 'onHandler']; $class = [$class, 'onHandler'];
} }
$this->middleware[] = $class; if (!is_array($class) || is_object($class[0])) {
$class = [$class];
}
foreach ($class as $closure) {
$this->middleware[] = $closure;
}
$this->restructure(); $this->restructure();
} }