add 存储过程支持 && 通过class_map使查询后执行事件进行转JSON处理

通过上面的方式删掉了上下文存储statement导致执行查询后无法进行存储过程问题

Author: kvnZero

app/ApiJson/Entity/ConditionEntity.php

@@ -17,6 +17,7 @@ class ConditionEntity
     protected array $group = [];
     protected array $order = [];
     protected array $having = [];
+    protected string $procedure = "";
 
     /**
      * @param array $condition 条件
@@ -155,6 +156,22 @@ public function getOrder(): array
         return $this->order;
     }
 
+    /**
+     * @param string $procedure
+     */
+    public function setProcedure(string $procedure): void
+    {
+        $this->procedure = $procedure;
+    }
+
+    /**
+     * @return string
+     */
+    public function getProcedure(): string
+    {
+        return $this->procedure;
+    }
+
     protected function log(array $condition)
     {
         $this->changeLog[] = [

app/ApiJson/Handle/FunctionProcedureHandle.php

@@ -0,0 +1,23 @@
+<?php
+
+namespace App\ApiJson\Handle;
+
+class FunctionProcedureHandle extends AbstractHandle
+{
+    protected string $keyWord = '@procedure()';
+
+    public function buildModel()
+    {
+        if (!in_array($this->keyWord, array_keys($this->condition->getCondition()))) {
+            return;
+        }
+
+        foreach (array_filter($this->condition->getCondition(), function($key){
+            return $key == $this->keyWord;
+        }, ARRAY_FILTER_USE_KEY) as $key => $value)
+        {
+            $this->condition->setProcedure($value);
+            $this->unsetKey[] = $this->keyWord;
+        }
+    }
+}

app/ApiJson/Method/GetMethod.php

@@ -18,25 +18,32 @@ protected function validateCondition(): bool
 
     protected function process()
     {
-        $handle = new Handle($this->tableEntity->getConditionEntity(), $this->tableEntity);
-        $handle->build();
-
         $queryMany = $this->isQueryMany();
         if (!$queryMany) {
             $this->tableEntity->getConditionEntity()->setLimit(1);
         }
 
+        $handle = new Handle($this->tableEntity->getConditionEntity(), $this->tableEntity);
+        $handle->build();
+
         //该事件鼓励是做语句缓存或者事件触发 不赞成修改语句做法 修改语句应在更上层的QueryHandle事件
         $event = new QueryExecuteBefore($this->query->toSql(), $this->method);
         ApplicationContext::getContainer()->get(EventDispatcherInterface::class)->dispatch($event);
 
         if(is_null($event->result)) {
             $result = $this->query->all();
 
-            $queryEvent = new QueryResult($result);
+            $queryEvent = new QueryResult($result, $this->query->toSql());
             ApplicationContext::getContainer()->get(EventDispatcherInterface::class)->dispatch($queryEvent);
 
             $result = $queryEvent->result;
+
+            if (!empty($this->tableEntity->getConditionEntity()->getProcedure())) {
+                foreach ($result as $i => $item) {
+                    $result[$i]['procedure'] = $this->query->callProcedure($item);
+                }
+            }
+
         } else {
             $result = $event->result;
         }

app/ApiJson/Model/MysqlQuery.php

@@ -87,6 +87,50 @@ public function getBindings(): array
         return $this->db->getBindings();
     }
 
+    public function callProcedure(array $dataItem, bool $query = true): array
+    {
+        if (empty($this->conditionEntity->getProcedure())) {
+            return [];
+        }
+        $count = -1;
+        $list = []; //默认值
+
+        $procedure = $this->conditionEntity->getProcedure();
+        $rule = '/(?<functionName>.+)\((?<args>.+?)\)/';
+        preg_match($rule, $procedure, $match);
+        if (empty($match['functionName'])) {
+            return [];
+        }
+        $args = array_map('trim', explode(',', $match['args']));
+
+        $callArgs = [];
+        foreach ($args as $arg) {
+            if (in_array($arg, array_keys($dataItem))) {
+                $callArgs[] = $dataItem[$arg];
+            } else if ($arg == '@limit') {
+                $callArgs[] = $this->conditionEntity->getLimit();
+            } else if ($arg == '@offset') {
+                $callArgs[] = $this->conditionEntity->getOffset();
+            } else {
+                $callArgs[] = $arg;
+            }
+        }
+
+        $sql = sprintf("CALL %s(%s)", $match['functionName'], join(',', $callArgs));
+
+        if ($query) {
+            $list = Db::select($sql);
+        } else {
+            $count = Db::affectingStatement($sql);
+        }
+
+        return [
+            'count' => $count,
+            'update' => !$query,
+            'list' => $list,
+        ];
+    }
+
     protected function buildQuery(bool $query = true)
     {
         if ($this->build) return;

app/ApiJson/Parse/Handle.php

@@ -20,6 +20,7 @@
 use App\ApiJson\Handle\FunctionLimitHandle;
 use App\ApiJson\Handle\FunctionOffsetHandle;
 use App\ApiJson\Handle\FunctionOrderHandle;
+use App\ApiJson\Handle\FunctionProcedureHandle;
 use App\ApiJson\Handle\WhereBetweenHandle;
 use App\ApiJson\Handle\WhereExistsHandle;
 use App\ApiJson\Handle\WhereHandle;
@@ -54,6 +55,7 @@ class Handle
      * @var AbstractHandle[]
      */
     protected array $methodRules = [
+        FunctionProcedureHandle::class,
         FunctionColumnHandle::class,
         FunctionHavingHandle::class,
         FunctionOffsetHandle::class,

app/ApiJson/Parse/Parse.php

@@ -15,7 +15,8 @@ class Parse
 {
     protected array $tagColumn = [
         'tag' => null,
-        'debug' => false
+        'debug' => false,
+        'other' => []
     ];
 
     protected array $globalKey = [

app/Event/ApiJson/QueryResult.php

@@ -9,7 +9,7 @@
  */
 class QueryResult
 {
-    public function __construct(public array $result)
+    public function __construct(public array $result, public string $sql)
     {
     }
 }

app/Event/StatementComplete.php

@@ -19,6 +19,13 @@ class StatementComplete
      */
     public $connection;
 
+    /**
+     * The query result
+     *
+     * @var array|false
+     */
+    public $result;
+
     /**
      * The PDO statement.
      *
@@ -30,11 +37,13 @@ class StatementComplete
      * Create a new event instance.
      *
      * @param \Hyperf\Database\Connection $connection
+     * @param array|false $result
      * @param \PDOStatement $statement
      */
-    public function __construct($connection, $statement)
+    public function __construct($connection, $result, $statement)
     {
         $this->statement = $statement;
+        $this->result = $result;
         $this->connection = $connection;
     }
 }

app/Listener/DbPreparedListener.php

@@ -3,11 +3,9 @@
 
 namespace App\Listener;
 
-use App\Constants\ConfigCode;
-use App\Event\ApiJson\QueryResult;
+use App\Event\StatementComplete;
 use Hyperf\Event\Annotation\Listener;
 use Hyperf\Event\Contract\ListenerInterface;
-use Hyperf\Utils\Context;
 
 /**
  * @Listener
@@ -17,18 +15,15 @@ class QueryResultTryToJsonListener implements ListenerInterface
     public function listen(): array
     {
         return [
-            QueryResult::class,
+            StatementComplete::class,
         ];
     }
 
     public function process(object $event)
     {
-        if (!$event instanceof QueryResult) return;
-
-        $statement = Context::get(ConfigCode::DB_QUERY_STATEMENT);
-        if (empty($statement)) {
-            return;
-        }
+        if (!$event instanceof StatementComplete) return;
+        if (empty($event->result)) return;
+        $statement = $event->statement;
 
         $columnCount = count(array_keys(current($event->result)));
         $columnMeta = [];

app/Listener/QueryResultTryToJsonListener.php

@@ -286,11 +286,9 @@ public function select(string $query, array $bindings = [], bool $useReadPdo = t
 
             $statement->execute();
 
-            $response = $statement->fetchAll();
+            $result = $statement->fetchAll();
 
-            $this->complete($statement);
-
-            return $response;
+            return $this->complete($result, $statement);
         });
     }
 
@@ -318,9 +316,9 @@ public function cursor(string $query, array $bindings = [], bool $useReadPdo = t
             // Next, we'll execute the query against the database and return the statement
             // so we can return the cursor. The cursor will use a PHP generator to give
             // back one row at a time without using a bunch of memory to render them.
-            $statement->execute();
+            $result = $statement->execute();
 
-            $this->complete($statement);
+            $this->complete($result, $statement);
 
             return $statement;
         });
@@ -990,12 +988,15 @@ protected function prepared(PDOStatement $statement)
         return $statement;
     }
 
-    protected function complete(PDOStatement $statement)
+    protected function complete($result, PDOStatement $statement)
     {
-        $this->event(new StatementComplete(
+        $event = new StatementComplete(
             $this,
+            $result,
             $statement
-        ));
+        );
+        $this->event($event);
+        return $event->result;
     }
 
     /**