zhamao-robot/zhamao-framework
1
0
Fork 0
mirror of https://github.com/zhamao-robot/zhamao-framework.git synced 2026-07-22 08:05:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

add middleware arg trait and annotation trait

Browse Source
This commit is contained in:
crazywhalecc
2023-03-05 22:15:01 +08:00
committed by Jerry
parent b8501e94f0
commit 3e2911b807
7 changed files with 129 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/ZM/Middleware/MiddlewareHandler.php
View File

@@ -5,6 +5,7 @@ declare(strict_types=1);
namespace ZM\Middleware;
use OneBot\Util\Singleton;
use ZM\Annotation\AnnotationBase;
use ZM\Exception\InvalidArgumentException;
class MiddlewareHandler
@@ -71,9 +72,9 @@ class MiddlewareHandler
$this->reg_map[$stack_id][] = [$name, $args];
}
public function getPipeClosure(callable $callback, $stack_id)
public function getPipeClosure(callable $callback, $stack_id, ?AnnotationBase $annotation = null)
{
$pipe_func = function (array $mid_list, ...$args) use ($callback, $stack_id, &$pipe_func) {
$pipe_func = function (array $mid_list, ...$args) use ($callback, $stack_id, $annotation, &$pipe_func) {
$return = true;
try {
while (($item = array_shift($mid_list)) !== null) {
@@ -81,6 +82,9 @@ class MiddlewareHandler
// 如果是 pipeline 形式的中间件，则使用闭包回去
if (class_exists($item[0]) && is_a($item[0], PipelineInterface::class, true)) {
$resolve = resolve($item[0]);
if (method_exists($resolve, 'setAnnotation') && $annotation !== null) {
$resolve->setAnnotation($annotation);
}
if (method_exists($resolve, 'setArgs')) {
$resolve->setArgs($item[1]);
}
@@ -158,6 +162,23 @@ class MiddlewareHandler
return $final_result ?? null;
}
public function processWithAnnotation(AnnotationBase $v, callable $callback, ...$args)
{
$stack_id = $this->getStackId($callback);
unset($this->stack[$stack_id]);
$this->callable_stack[] = $callback;
// 遍历执行before并压栈，并在遇到返回false后停止
try {
$mid_list = ($this->reg_map[$stack_id] ?? []);
$final_result = ($this->getPipeClosure($callback, $stack_id, $v))($mid_list, ...$args);
} finally {
array_pop($this->callable_stack);
}
return $final_result ?? null;
}
/**
* 获取正在运行的回调调用对象，可能是Closure、array、string
*