新增支持查询器别名，减少查询器连贯操作重复构建步骤

imiphp · Oct 25, 2019 · a8b6e5e · a8b6e5e
1 parent 6e84883
commit a8b6e5e
Show file tree

Hide file tree

Showing 9 changed files with 236 additions and 34 deletions.
diff --git a/src/Db/Query/Builder/InsertBuilder.php b/src/Db/Query/Builder/InsertBuilder.php
@@ -38,7 +38,7 @@ public function build(...$args)
                 else
                 {
                     $fields[] = $this->parseKeyword($k);
-                    $valueParam = $this->query->getAutoParamName();
+                    $valueParam = ':' . $k;
                     $valueParams[] = $valueParam;
                     $this->params[$valueParam] = $v;
                 }

diff --git a/src/Db/Query/Builder/ReplaceBuilder.php b/src/Db/Query/Builder/ReplaceBuilder.php
@@ -42,7 +42,7 @@ public function build(...$args)
                 }
                 else
                 {
-                    $valueParam = $this->query->getAutoParamName();
+                    $valueParam = ':' . $k;
                     $this->params[$valueParam] = $v;
                     $setStrs[] = $this->parseKeyword($k) . ' = ' . $valueParam;
                 }

diff --git a/src/Db/Query/Builder/UpdateBuilder.php b/src/Db/Query/Builder/UpdateBuilder.php
@@ -34,7 +34,7 @@ public function build(...$args)
             }
             else
             {
-                $valueParam = $this->query->getAutoParamName();
+                $valueParam = ':' . $k;
                 $valueParams[] = $valueParam;
                 $this->params[$valueParam] = $v;
                 $setStrs[] = $this->parseKeyword($k) . ' = ' . $valueParam;

diff --git a/src/Db/Query/Interfaces/IQuery.php b/src/Db/Query/Interfaces/IQuery.php
@@ -594,5 +594,19 @@ public function setFieldInc($fieldName, float $incValue = 1);
      */
     public function setFieldDec($fieldName, float $decValue = 1);
 
+    /**
+     * 获取自动起名的参数名称
+     * @return string
+     */
     public function getAutoParamName();
+
+    /**
+     * 查询器别名
+     *
+     * @param string $name
+     * @param callable $callable
+     * @return static
+     */
+    public function alias($name, $callable);
+
 }
diff --git a/src/Db/Query/Query.php b/src/Db/Query/Query.php
@@ -91,6 +91,27 @@ class Query implements IQuery
      */
     protected $dbParamInc = 0;
 
+    /**
+     * 查询器别名集合
+     *
+     * @var \Imi\Db\Query\Interfaces\IQuery[]
+     */
+    protected static $aliasMap = [];
+
+    /**
+     * 当前别名
+     *
+     * @var string
+     */
+    protected $alias;
+
+    /**
+     * 别名 Sql 集合
+     *
+     * @var string[]
+     */
+    protected static $aliasSqls = [];
+
     public function __construct(IDb $db = null, $modelClass = null, $poolName = null, $queryType = null)
     {
         $this->db = $db;
@@ -111,6 +132,13 @@ public function __init()
         }
     }
 
+    public function __clone()
+    {
+        $this->isInitDb = false;
+        $this->db = null;
+        $this->option = clone $this->option;
+    }
+
     /**
      * 获取所有操作的记录
      * @return QueryOption
@@ -856,8 +884,15 @@ public function getBinds()
      */
     public function select(): IResult
     {
-        $builder = new SelectBuilder($this);
-        $sql = $builder->build();
+        if($this->alias && isset(static::$aliasSqls[$this->alias]))
+        {
+            $sql = static::$aliasSqls[$this->alias];
+        }
+        else
+        {
+            $builder = new SelectBuilder($this);
+            static::$aliasSqls[$this->alias] = $sql = $builder->build();
+        }
         if(!$this->isInitQueryType && !$this->isInTransaction())
         {
             $this->queryType = QueryType::READ;
@@ -899,8 +934,29 @@ public function paginate($page, $count, $options = []): IPaginateResult
      */
     public function insert($data = null): IResult
     {
-        $builder = new InsertBuilder($this);
-        $sql = $builder->build($data);
+        if($this->alias && isset(static::$aliasSqls[$this->alias]))
+        {
+            $sql = static::$aliasSqls[$this->alias];
+            $bindValues = [];
+            $numberKey = isset($data[0]);
+            foreach($data as $k => $v)
+            {
+                if($numberKey)
+                {
+                    $bindValues[':' . ($k + 1)] = $v;
+                }
+                else
+                {
+                    $bindValues[':' . $k] = $v;
+                }
+            }
+            $this->bindValues($bindValues);
+        }
+        else
+        {
+            $builder = new InsertBuilder($this);
+            static::$aliasSqls[$this->alias] = $sql = $builder->build($data);
+        }
         return $this->execute($sql);
     }
 
@@ -926,8 +982,21 @@ public function batchInsert($data = null): IResult
      */
     public function update($data = null): IResult
     {
-        $builder = new UpdateBuilder($this);
-        $sql = $builder->build($data);
+        if($this->alias && isset(static::$aliasSqls[$this->alias]))
+        {
+            $sql = static::$aliasSqls[$this->alias];
+            $bindValues = [];
+            foreach($data as $k => $v)
+            {
+                $bindValues[':' . $k] = $v;
+            }
+            $this->bindValues($bindValues);
+        }
+        else
+        {
+            $builder = new UpdateBuilder($this);
+            static::$aliasSqls[$this->alias] = $sql = $builder->build($data);
+        }
         return $this->execute($sql);
     }
 
@@ -939,8 +1008,21 @@ public function update($data = null): IResult
      */
     public function replace($data = null): IResult
     {
-        $builder = new ReplaceBuilder($this);
-        $sql = $builder->build($data);
+        if($this->alias && isset(static::$aliasSqls[$this->alias]))
+        {
+            $sql = static::$aliasSqls[$this->alias];
+            $bindValues = [];
+            foreach($data as $k => $v)
+            {
+                $bindValues[':' . $k] = $v;
+            }
+            $this->bindValues($bindValues);
+        }
+        else
+        {
+            $builder = new ReplaceBuilder($this);
+            static::$aliasSqls[$this->alias] = $sql = $builder->build($data);
+        }
         return $this->execute($sql);
     }
 
@@ -950,10 +1032,16 @@ public function replace($data = null): IResult
      */
     public function delete(): IResult
     {
-        $builder = new DeleteBuilder($this);
-        $sql = $builder->build();
+        if($this->alias && isset(static::$aliasSqls[$this->alias]))
+        {
+            $sql = static::$aliasSqls[$this->alias];
+        }
+        else
+        {
+            $builder = new DeleteBuilder($this);
+            static::$aliasSqls[$this->alias] = $sql = $builder->build();
+        }
         $result = $this->execute($sql);
-        $this->__init();
         return $result;
     }
 
@@ -1158,4 +1246,23 @@ private function isInTransaction()
             return false;
         }
     }
+
+    /**
+     * 查询器别名
+     *
+     * @param string $name
+     * @param callable $callable
+     * @return static
+     */
+    public function alias($name, $callable)
+    {
+        if(!isset(static::$aliasMap[$name]))
+        {
+            $callable($this);
+            $this->alias = $name;
+            static::$aliasMap[$name] = $this;
+        }
+        return clone static::$aliasMap[$name];
+    }
+
 }
diff --git a/src/Db/Query/QueryOption.php b/src/Db/Query/QueryOption.php
@@ -74,4 +74,10 @@ public function __construct()
     {
         $this->table = new Table();
     }
+
+    public function __clone()
+    {
+        $this->table = clone $this->table;
+    }
+
 }