This commit is contained in:
as2252258@163.com
2021-02-23 00:33:45 +08:00
parent e9f1312b56
commit b84ca315b6
+247 -244
View File
@@ -21,290 +21,293 @@ use Snowflake\Snowflake;
class Annotation extends Component class Annotation extends Component
{ {
private array $_annotations = []; private array $_annotations = [];
private array $_classes = []; private array $_classes = [];
private array $_targets = []; private array $_targets = [];
private array $_methods = []; private array $_methods = [];
/** /**
* @param array $handler * @param array $handler
* @param $name * @param $name
*/ */
public function addMethodAttribute(array $handler, $name) public function addMethodAttribute(array $handler, $name)
{ {
$this->_methods[get_class($handler[0])][$name] = $handler; $this->_methods[get_class($handler[0])][$name] = $handler;
} }
/** /**
* @param string $className * @param string $className
* @return array 根据类名获取注解 * @return array 根据类名获取注解
* 根据类名获取注解 * 根据类名获取注解
*/ */
public function getMethods(string $className): array public function getMethods(string $className): array
{ {
return $this->_methods[$className] ?? []; return $this->_methods[$className] ?? [];
} }
/** /**
* @param string $path * @param string $path
* @param string $namespace * @param string $namespace
* @param string $alias * @param string $alias
* @return $this * @return $this
* @throws ReflectionException|NotFindPropertyException * @throws ReflectionException|NotFindPropertyException
*/ */
public function readControllers(string $path, string $namespace, string $alias = 'root'): static public function readControllers(string $path, string $namespace, string $alias = 'root'): static
{ {
return $this->scanDir(glob($path . '*'), $namespace, $alias); return $this->scanDir(glob($path . '*'), $namespace, $alias);
} }
/** /**
* @param string $alias * @param string $alias
* @return array * @return array
*/ */
public function getAlias(string $alias = 'root'): array public function getAlias(string $alias = 'root'): array
{ {
if (!isset($this->_annotations[$alias])) { if (!isset($this->_annotations[$alias])) {
return []; return [];
} }
return $this->_annotations[$alias]; return $this->_annotations[$alias];
} }
/** /**
* @param array $paths * @param array $paths
* @param string $namespace * @param string $namespace
* @param string $alias * @param string $alias
* @return $this * @return $this
* @throws ReflectionException|NotFindPropertyException * @throws ReflectionException|NotFindPropertyException
* @throws NotFindClassException * @throws NotFindClassException
*/ */
private function scanDir(array $paths, string $namespace, string $alias): static private function scanDir(array $paths, string $namespace, string $alias): static
{ {
if (!isset($this->_annotations[$alias])) { if (!isset($this->_annotations[$alias])) {
$this->_annotations[$alias] = []; $this->_annotations[$alias] = [];
} }
foreach ($paths as $path) { foreach ($paths as $path) {
$explode = explode('/', $path); $explode = explode('/', $path);
$explode_pop = array_pop($explode); $explode_pop = array_pop($explode);
if (is_file($path)) { if (is_file($path)) {
if (!str_contains($path, '.php')) { if (!str_contains($path, '.php')) {
continue; continue;
} }
$explode_pop = str_replace('.php', '', $explode_pop); $explode_pop = str_replace('.php', '', $explode_pop);
$this->getReflect($namespace . '\\' . $explode_pop, $alias); $this->getReflect($namespace . '\\' . $explode_pop, $alias);
} else { } else {
$this->scanDir(glob($path . '/*'), $namespace . '\\' . $explode_pop, $alias); $this->scanDir(glob($path . '/*'), $namespace . '\\' . $explode_pop, $alias);
} }
} }
return $this; return $this;
} }
/** /**
* @param string $class * @param string $class
* @param string $alias * @param string $alias
* @return array * @return array
* @throws Exception * @throws Exception
*/ */
private function getReflect(string $class, string $alias): array private function getReflect(string $class, string $alias): array
{ {
try { try {
$reflect = $this->reflectClass($class); $reflect = $this->reflectClass($class);
if (empty($reflect)) { if (empty($reflect)) {
return []; return [];
} }
if (!$reflect->isInstantiable()) {
return [];
}
$construct = $reflect->getConstructor(); $construct = $reflect->getConstructor();
if (!empty($construct) && count($params = $construct->getParameters()) > 0) { if (!empty($construct) && count($params = $construct->getParameters()) > 0) {
if (!$params[0]->isOptional()) { if (!$params[0]->isOptional()) {
return []; return [];
} }
} }
$object = $reflect->newInstance(); $object = $reflect->newInstance();
$this->resolveMethod($reflect, $class, $alias, $object); $this->resolveMethod($reflect, $class, $alias, $object);
return $this->targets($reflect); return $this->targets($reflect);
} catch (\Throwable $throwable) { } catch (\Throwable $throwable) {
$this->addError($throwable); $this->addError($throwable);
return []; return [];
} }
} }
/** /**
* @param ReflectionClass $reflect * @param ReflectionClass $reflect
* @param $class * @param $class
* @param $alias * @param $alias
* @param $object * @param $object
* @throws NotFindPropertyException * @throws NotFindPropertyException
*/ */
private function resolveMethod(ReflectionClass $reflect, $class, $alias, $object) private function resolveMethod(ReflectionClass $reflect, $class, $alias, $object)
{ {
foreach ($reflect->getMethods(ReflectionMethod::IS_PUBLIC) as $method) { foreach ($reflect->getMethods(ReflectionMethod::IS_PUBLIC) as $method) {
if ($method->class != $class) { if ($method->class != $class) {
continue; continue;
} }
$tmp = $this->resolveAnnotations($method, $alias, $object); $tmp = $this->resolveAnnotations($method, $alias, $object);
if (empty($tmp)) { if (empty($tmp)) {
continue; continue;
} }
$this->_classes[$reflect->getName()][$method->getName()] = $tmp; $this->_classes[$reflect->getName()][$method->getName()] = $tmp;
} }
$this->resolveProperty($reflect, $object); $this->resolveProperty($reflect, $object);
} }
/** /**
* @param ReflectionClass $reflectionClass * @param ReflectionClass $reflectionClass
* @param $object * @param $object
* @throws NotFindPropertyException * @throws NotFindPropertyException
*/ */
private function resolveProperty(ReflectionClass $reflectionClass, $object) private function resolveProperty(ReflectionClass $reflectionClass, $object)
{ {
$property = $reflectionClass->getProperties(); $property = $reflectionClass->getProperties();
foreach ($property as $value) { foreach ($property as $value) {
if ($value->isStatic()) continue; if ($value->isStatic()) continue;
$attributes = $value->getAttributes(); $attributes = $value->getAttributes();
if (count($attributes) < 1) { if (count($attributes) < 1) {
continue; continue;
} }
foreach ($attributes as $attribute) { foreach ($attributes as $attribute) {
/** @var IAnnotation $annotation */ /** @var IAnnotation $annotation */
$annotation = $this->instance($attribute); $annotation = $this->instance($attribute);
if (empty($annotation)) { if (empty($annotation)) {
continue; continue;
} }
$annotation = $annotation->execute([$object, $value->getName()]); $annotation = $annotation->execute([$object, $value->getName()]);
if ($value->isPublic()) { if ($value->isPublic()) {
$object->{$value->getName()} = $annotation; $object->{$value->getName()} = $annotation;
} else { } else {
$name = 'set' . ucfirst($value->getName()); $name = 'set' . ucfirst($value->getName());
if (!method_exists($object, $name)) { if (!method_exists($object, $name)) {
throw new NotFindPropertyException('set property need method ' . $name); throw new NotFindPropertyException('set property need method ' . $name);
} }
$object->$name($annotation); $object->$name($annotation);
} }
} }
} }
} }
/** /**
* @param string $class * @param string $class
* @return ReflectionClass|null * @return ReflectionClass|null
* @throws ReflectionException * @throws ReflectionException
* @throws NotFindClassException * @throws NotFindClassException
*/ */
private function reflectClass(string $class): ?ReflectionClass private function reflectClass(string $class): ?ReflectionClass
{ {
return Snowflake::getDi()->getReflect($class); return Snowflake::getDi()->getReflect($class);
} }
/** /**
* @param ReflectionMethod $method * @param ReflectionMethod $method
* @param $alias * @param $alias
* @param $object * @param $object
* @return array * @return array
*/ */
private function resolveAnnotations(ReflectionMethod $method, $alias, $object): array private function resolveAnnotations(ReflectionMethod $method, $alias, $object): array
{ {
$attributes = $method->getAttributes(); $attributes = $method->getAttributes();
if (count($attributes) < 1) { if (count($attributes) < 1) {
return []; return [];
} }
$names = []; $names = [];
foreach ($attributes as $attribute) { foreach ($attributes as $attribute) {
/** @var IAnnotation $class */ /** @var IAnnotation $class */
$class = $this->instance($attribute); $class = $this->instance($attribute);
if ($class === null) { if ($class === null) {
continue; continue;
} }
$names[$attribute->getName()] = $class->execute([$object, $method->getName()]); $names[$attribute->getName()] = $class->execute([$object, $method->getName()]);
} }
$tmp['handler'] = [$object, $method->getName()]; $tmp['handler'] = [$object, $method->getName()];
$tmp['attributes'] = $names; $tmp['attributes'] = $names;
$this->_annotations[$alias][] = $tmp; $this->_annotations[$alias][] = $tmp;
return $tmp; return $tmp;
} }
/** /**
* @param $className * @param $className
* @param string $method * @param string $method
* @return array * @return array
*/ */
public function getByClass($className, $method = ''): array public function getByClass($className, $method = ''): array
{ {
if (!isset($this->_classes[$className])) { if (!isset($this->_classes[$className])) {
return []; return [];
} }
if (empty($method)) { if (empty($method)) {
return $this->_classes[$className]; return $this->_classes[$className];
} }
foreach ($this->_classes[$className] as $_method => $class) { foreach ($this->_classes[$className] as $_method => $class) {
if ($method == $_method) { if ($method == $_method) {
return [$class]; return [$class];
} }
} }
return []; return [];
} }
/** /**
* @param ReflectionClass $reflect * @param ReflectionClass $reflect
* @return array * @return array
*/ */
private function targets(ReflectionClass $reflect): array private function targets(ReflectionClass $reflect): array
{ {
$name = $reflect->getName(); $name = $reflect->getName();
if (!isset($this->_classes[$name])) { if (!isset($this->_classes[$name])) {
$this->_classes[$name] = []; $this->_classes[$name] = [];
} }
$attributes = $reflect->getAttributes(); $attributes = $reflect->getAttributes();
if (count($attributes) > 0) { if (count($attributes) > 0) {
if (!isset($this->_targets[$name])) { if (!isset($this->_targets[$name])) {
$this->_targets[$name] = []; $this->_targets[$name] = [];
} }
foreach ($attributes as $attribute) { foreach ($attributes as $attribute) {
$class = $this->instance($attribute); $class = $this->instance($attribute);
if ($class === null) { if ($class === null) {
continue; continue;
} }
$this->_targets[$name][] = $class; $this->_targets[$name][] = $class;
} }
} }
return []; return [];
} }
/** /**
* @param ReflectionAttribute $attribute * @param ReflectionAttribute $attribute
* @return array|object|null * @return array|object|null
*/ */
private function instance(ReflectionAttribute $attribute): array|object|null private function instance(ReflectionAttribute $attribute): array|object|null
{ {
if (!class_exists($attribute->getName())) { if (!class_exists($attribute->getName())) {
return null; return null;
} }
return $attribute->newInstance(); return $attribute->newInstance();
} }
} }