modify
This commit is contained in:
+17
-2
@@ -4,18 +4,25 @@
|
||||
namespace Annotation;
|
||||
|
||||
|
||||
use Snowflake\Aop;
|
||||
use Snowflake\IAspect;
|
||||
use Snowflake\Snowflake;
|
||||
|
||||
defined('ASPECT_ERROR') or define('ASPECT_ERROR', 'Aspect annotation must implement ');
|
||||
|
||||
/**
|
||||
* Class Aspect
|
||||
* @package Annotation
|
||||
*/
|
||||
#[\Attribute(\Attribute::TARGET_METHOD | \Attribute::TARGET_FUNCTION)] class Aspect extends Attribute
|
||||
#[\Attribute(\Attribute::TARGET_METHOD)] class Aspect extends Attribute
|
||||
{
|
||||
|
||||
|
||||
/**
|
||||
* Aspect constructor.
|
||||
* @param string $aspect
|
||||
*/
|
||||
public function __construct()
|
||||
public function __construct(public string $aspect)
|
||||
{
|
||||
}
|
||||
|
||||
@@ -26,6 +33,14 @@ namespace Annotation;
|
||||
*/
|
||||
public function execute(array $handler): mixed
|
||||
{
|
||||
if (!in_array(IAspect::class, class_implements($this->aspect))) {
|
||||
throw new \Exception(ASPECT_ERROR . IAspect::class);
|
||||
}
|
||||
/** @var Aop $aop */
|
||||
$aop = Snowflake::app()->get('aop');
|
||||
|
||||
$aop->aop_add($handler, $this->aspect);
|
||||
|
||||
return parent::execute($handler); // TODO: Change the autogenerated stub
|
||||
}
|
||||
|
||||
|
||||
@@ -0,0 +1,65 @@
|
||||
<?php
|
||||
|
||||
|
||||
namespace Snowflake;
|
||||
|
||||
|
||||
use Snowflake\Abstracts\Component;
|
||||
|
||||
defined('ASPECT_ERROR') or define('ASPECT_ERROR', 'Aspect annotation must implement ');
|
||||
|
||||
|
||||
/**
|
||||
* Class Aop
|
||||
* @package Snowflake
|
||||
*/
|
||||
class Aop extends Component
|
||||
{
|
||||
|
||||
|
||||
private array $_aop = [];
|
||||
|
||||
|
||||
/**
|
||||
* @param $className
|
||||
* @param null $method
|
||||
*/
|
||||
public function aop_add(array $handler, string $aspect)
|
||||
{
|
||||
[$class, $method] = $handler;
|
||||
if (!isset($this->_aop[$aspect])) {
|
||||
$this->_aop[$aspect] = [];
|
||||
}
|
||||
|
||||
$this->_aop[get_class($class) . '::' . $method][] = $aspect;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @return mixed|void
|
||||
* @throws \ReflectionException
|
||||
*/
|
||||
final public function dispatch()
|
||||
{
|
||||
$get_args = func_get_args();
|
||||
[$class, $method] = array_shift($get_args);
|
||||
|
||||
$aopName = get_class($class) . '::' . $method;
|
||||
if (!isset($this->_aop[$aopName])) {
|
||||
return call_user_func($get_args, ...$get_args);
|
||||
}
|
||||
|
||||
$reflect = new \ReflectionClass($this->_aop[$aopName]);
|
||||
if (!$reflect->hasMethod('invoke')) {
|
||||
throw new \Exception(ASPECT_ERROR . IAspect::class);
|
||||
}
|
||||
$method = $reflect->getMethod('invoke');
|
||||
|
||||
$data = $method->invokeArgs($reflect->newInstance([$class, $method]), $get_args);
|
||||
if ($method->getReturnType() !== null) {
|
||||
return $data;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
@@ -0,0 +1,55 @@
|
||||
<?php
|
||||
|
||||
|
||||
namespace Snowflake\Error;
|
||||
|
||||
|
||||
use Snowflake\IAspect;
|
||||
|
||||
|
||||
/**
|
||||
* Class LoggerAspect
|
||||
* @package Snowflake\Error
|
||||
*/
|
||||
class LoggerAspect implements IAspect
|
||||
{
|
||||
|
||||
private string $className = '';
|
||||
private string $methodName = '';
|
||||
|
||||
|
||||
/**
|
||||
* LoggerAspect constructor.
|
||||
* @param array $handler
|
||||
*/
|
||||
public function __construct(public array $handler, $needReturn)
|
||||
{
|
||||
$this->className = get_class($this->handler[0]);
|
||||
$this->methodName = $this->handler[1];
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @return mixed|void
|
||||
*/
|
||||
public function invoke()
|
||||
{
|
||||
$startTime = microtime(true);
|
||||
|
||||
$data = call_user_func($this->handler, func_get_args());
|
||||
|
||||
$this->print_runtime($startTime);
|
||||
|
||||
return $data;
|
||||
}
|
||||
|
||||
|
||||
private function print_runtime($startTime)
|
||||
{
|
||||
$runTime = round(microtime(true) - $startTime, 6);
|
||||
echo sprintf('run %s::%s use time %6f', $this->className, $this->methodName, $runTime);
|
||||
echo PHP_EOL;
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
@@ -0,0 +1,23 @@
|
||||
<?php
|
||||
|
||||
|
||||
namespace Snowflake;
|
||||
|
||||
|
||||
interface IAspect
|
||||
{
|
||||
|
||||
|
||||
/**
|
||||
* IAspect constructor.
|
||||
* @param array $handler
|
||||
*/
|
||||
public function __construct(array $handler, bool $needRetruen);
|
||||
|
||||
|
||||
/**
|
||||
* @return mixed|void
|
||||
*/
|
||||
public function invoke();
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user