This commit is contained in:
2021-07-27 14:15:08 +08:00
parent dcb588e02d
commit 8ee27b9b14
2 changed files with 245 additions and 247 deletions
+243 -246
View File
@@ -9,7 +9,6 @@ use Annotation\Route\RpcProducer;
use Closure; use Closure;
use Exception; use Exception;
use HttpServer\Abstracts\HttpService; use HttpServer\Abstracts\HttpService;
use HttpServer\Controller;
use HttpServer\Http\Context; use HttpServer\Http\Context;
use HttpServer\Http\Request; use HttpServer\Http\Request;
use JetBrains\PhpStorm\Pure; use JetBrains\PhpStorm\Pure;
@@ -26,291 +25,289 @@ use Snowflake\Snowflake;
class Node extends HttpService class Node extends HttpService
{ {
public string $path = ''; public string $path = '';
public int $index = 0; public int $index = 0;
public string $method = ''; public string $method = '';
/** @var Node[] $childes */ /** @var Node[] $childes */
public array $childes = []; public array $childes = [];
public array $group = []; public array $group = [];
private string $_error = ''; private string $_error = '';
private string $_dataType = ''; private string $_dataType = '';
/** @var ?Closure|?array */ /** @var ?Closure|?array */
public Closure|array|null $handler; public Closure|array|null $handler;
public string $htmlSuffix = '.html'; public string $htmlSuffix = '.html';
public bool $enableHtmlSuffix = false; public bool $enableHtmlSuffix = false;
public array $namespace = []; public array $namespace = [];
public array $middleware = []; public array $middleware = [];
/** @var array|Closure */ /** @var array|Closure */
public Closure|array $callback = []; public Closure|array $callback = [];
private string $_alias = ''; private string $_alias = '';
/** /**
* @param string $dataType * @param string $dataType
*/ */
public function setDataType(string $dataType) public function setDataType(string $dataType)
{ {
$this->_dataType = $dataType; $this->_dataType = $dataType;
} }
/** /**
* @param string $data * @param string $data
* @return mixed * @return mixed
*/ */
public function unpack(string $data): mixed public function unpack(string $data): mixed
{ {
if ($this->_dataType == RpcProducer::PROTOCOL_JSON) { if ($this->_dataType == RpcProducer::PROTOCOL_JSON) {
return json_decode($data, true); return json_decode($data, true);
} }
if ($this->_dataType == RpcProducer::PROTOCOL_SERIALIZE) { if ($this->_dataType == RpcProducer::PROTOCOL_SERIALIZE) {
return unserialize($data); return unserialize($data);
} }
return $data; return $data;
} }
/** /**
* @param $handler * @param $handler
* @return Node * @return Node
* @throws * @throws
*/ */
public function bindHandler($handler): static public function bindHandler($handler): static
{ {
if (is_string($handler) && str_contains($handler, '@')) { if (is_string($handler) && str_contains($handler, '@')) {
list($controller, $action) = explode('@', $handler); list($controller, $action) = explode('@', $handler);
if (!class_exists($controller) && !empty($this->namespace)) { if (!class_exists($controller) && !empty($this->namespace)) {
$controller = implode('\\', $this->namespace) . '\\' . $controller; $controller = implode('\\', $this->namespace) . '\\' . $controller;
} }
$this->handler = [Snowflake::getDi()->get($controller), $action]; $this->handler = [Snowflake::getDi()->get($controller), $action];
} else if ($handler != null && !is_callable($handler, true)) { } else if ($handler != null && !is_callable($handler, true)) {
$this->_error = 'Controller is con\'t exec.'; $this->_error = 'Controller is con\'t exec.';
} else { } else {
$this->handler = $handler; $this->handler = $handler;
} }
if (!empty($this->handler) && is_array($this->handler)) { if (!empty($this->handler) && is_array($this->handler)) {
$this->callback = di(MiddlewareManager::class)->callerMiddlewares( $this->callback = di(MiddlewareManager::class)->callerMiddlewares(
$this->handler[0], $this->handler[1], $this->createDispatch() $this->handler[0], $this->handler[1], $this->createDispatch()
); );
} }
return $this; return $this;
} }
/** /**
* @throws ReflectionException * @throws ReflectionException
* @throws NotFindClassException * @throws NotFindClassException
* @throws Exception * @throws Exception
*/ */
private function createDispatch() private function createDispatch()
{ {
$application = $this; $application = $this;
/** @var Aop $aop */ /** @var Aop $aop */
$aop = Snowflake::app()->get('aop'); $aop = Snowflake::app()->get('aop');
if ($this->handler instanceof Closure || !$aop->hasAop($this->handler)) { if ($this->handler instanceof Closure || !$aop->hasAop($this->handler)) {
return static function () use ($application) { return (static function ($application) {
$dispatchParam = Context::getContext('dispatch-param', [\request()]); $dispatchParam = Context::getContext('dispatch-param', [\request()]);
if (is_array($application->handler)){ if (is_array($application->handler)) {
Snowflake::injectProperty($application->handler[0]); Snowflake::injectProperty($application->handler[0]);
} }
return call_user_func($application->handler, ...$dispatchParam); return call_user_func($application->handler, ...$dispatchParam);
}; })($application);
} }
$reflect = $aop->getAop($this->handler); $reflect = $aop->getAop($this->handler);
$callback = [$reflect->getMethod('invoke'), 'invokeArgs']; return (static function ($callback, $application, $reflect) {
return static function () use ($callback, $application, $reflect) { $dispatchParam = Context::getContext('dispatch-param', [\request()]);
$dispatchParam = Context::getContext('dispatch-param', [\request()]); $asp = $reflect->newInstance($application->handler);
if (is_array($application->handler)) {
$asp = $reflect->newInstance($application->handler); Snowflake::injectProperty($application->handler[0]);
if (is_array($application->handler)){ }
Snowflake::injectProperty($application->handler[0]); call_user_func($callback, $asp, $dispatchParam);
} })([$reflect->getMethod('invoke'), 'invokeArgs'], $application, $reflect);
call_user_func($callback, $asp, $dispatchParam); }
};
}
/** /**
* @return array * @return array
*/ */
#[Pure] protected function annotation(): array #[Pure] protected function annotation(): array
{ {
return $this->getMiddleWares(); return $this->getMiddleWares();
} }
/** /**
* @param Request $request * @param Request $request
* @return bool * @return bool
*/ */
public function methodAllow(Request $request): bool public function methodAllow(Request $request): bool
{ {
if ($this->method == $request->getMethod()) { if ($this->method == $request->getMethod()) {
return true; return true;
} }
return $this->method == 'any'; return $this->method == 'any';
} }
/** /**
* @return bool * @return bool
* @throws Exception * @throws Exception
*/ */
public function checkSuffix(): bool public function checkSuffix(): bool
{ {
if ($this->enableHtmlSuffix) { if ($this->enableHtmlSuffix) {
$url = request()->getUri(); $url = request()->getUri();
$nowLength = strlen($this->htmlSuffix); $nowLength = strlen($this->htmlSuffix);
if (strpos($url, $this->htmlSuffix) !== strlen($url) - $nowLength) { if (strpos($url, $this->htmlSuffix) !== strlen($url) - $nowLength) {
return false; return false;
} }
} }
return true; return true;
} }
/** /**
* @param Closure|array|string $handler * @param Closure|array|string $handler
* @throws Exception * @throws Exception
*/ */
public function addLimits(Closure|string|array $handler) public function addLimits(Closure|string|array $handler)
{ {
if (!is_array($handler) || is_object($handler[0])) { if (!is_array($handler) || is_object($handler[0])) {
$handler = [$handler]; $handler = [$handler];
} }
foreach ($handler as $closure) { foreach ($handler as $closure) {
if (in_array($closure, $this->_limits)) { if (in_array($closure, $this->_limits)) {
continue; continue;
} }
$this->_limits[] = $closure; $this->_limits[] = $closure;
} }
} }
/** /**
* @return string * @return string
* 错误信息 * 错误信息
*/ */
public function getError(): string public function getError(): string
{ {
return $this->_error; return $this->_error;
} }
/** /**
* @param Node $node * @param Node $node
* @param string $field * @param string $field
* @return Node * @return Node
*/ */
public function addChild(Node $node, string $field): Node public function addChild(Node $node, string $field): Node
{ {
$field = (string)$field; $field = (string)$field;
/** @var Node $oLod */ /** @var Node $oLod */
$oLod = $this->childes[$field] ?? null; $oLod = $this->childes[$field] ?? null;
if (!empty($oLod)) { if (!empty($oLod)) {
$node = $oLod; $node = $oLod;
} }
$this->childes[$field] = $node; $this->childes[$field] = $node;
return $this->childes[$field]; return $this->childes[$field];
} }
/** /**
* @param string $search * @param string $search
* @return Node|null * @return Node|null
* @throws Exception * @throws Exception
*/ */
public function findNode(string $search): ?Node public function findNode(string $search): ?Node
{ {
if (empty($this->childes)) { if (empty($this->childes)) {
return null; return null;
} }
if (isset($this->childes[$search])) { if (isset($this->childes[$search])) {
return $this->childes[$search]; return $this->childes[$search];
} }
foreach ($this->childes as $key => $val) { foreach ($this->childes as $key => $val) {
if ($search == $key) { if ($search == $key) {
return $this->childes[$key]; return $this->childes[$key];
} }
} }
return null; return null;
} }
/** /**
* @param string $alias * @param string $alias
* @return $this * @return $this
* 别称 * 别称
*/ */
public function alias(string $alias): static public function alias(string $alias): static
{ {
$this->_alias = $alias; $this->_alias = $alias;
return $this; return $this;
} }
/** /**
* @return string * @return string
*/ */
public function getAlias(): string public function getAlias(): string
{ {
return $this->_alias; return $this->_alias;
} }
/** /**
* @param Closure|array $class * @param Closure|array $class
* @return $this * @return $this
*/ */
public function addMiddleware(Closure|array $class): static public function addMiddleware(Closure|array $class): static
{ {
if (empty($class)) return $this; if (empty($class)) return $this;
foreach ($class as $closure) { foreach ($class as $closure) {
if (in_array($closure, $this->middleware)) { if (in_array($closure, $this->middleware)) {
continue; continue;
} }
$this->middleware[] = $closure; $this->middleware[] = $closure;
} }
return $this; return $this;
} }
/** /**
* @return array * @return array
*/ */
public function getMiddleWares(): array public function getMiddleWares(): array
{ {
return $this->middleware; return $this->middleware;
} }
/** /**
* @return mixed * @return mixed
* @throws Exception * @throws Exception
*/ */
public function dispatch(): mixed public function dispatch(): mixed
{ {
Context::setContext('dispatch-param', func_get_args()); Context::setContext('dispatch-param', func_get_args());
if (empty($this->callback)) { if (empty($this->callback)) {
return Json::to(404, $this->errorMsg()); return Json::to(404, $this->errorMsg());
} }
return call_user_func($this->callback, \request()); return call_user_func($this->callback, \request());
} }
/** /**
* @return string * @return string
*/ */
private function errorMsg(): string private function errorMsg(): string
{ {
return $this->_error ?? 'Page not found.'; return $this->_error ?? 'Page not found.';
} }
} }
+2 -1
View File
@@ -12,6 +12,7 @@ use Exception;
use HttpServer\IInterface\Task; use HttpServer\IInterface\Task;
use JetBrains\PhpStorm\Pure; use JetBrains\PhpStorm\Pure;
use ReflectionException; use ReflectionException;
use ReflectionProperty;
use Snowflake\Abstracts\Config; use Snowflake\Abstracts\Config;
use Snowflake\Core\Json; use Snowflake\Core\Json;
use Snowflake\Di\Container; use Snowflake\Di\Container;
@@ -58,7 +59,7 @@ class Snowflake
$attributes = static::getDi()->getClassProperty($class::class); $attributes = static::getDi()->getClassProperty($class::class);
/** /**
* @var string $property * @var string $property
* @var \ReflectionProperty $attribute * @var ReflectionProperty $attribute
*/ */
foreach ($attributes as $property => $attribute) { foreach ($attributes as $property => $attribute) {