增加 join, trace 功能

Author: onanying

README.md

@@ -1,6 +1,6 @@
 ## ShardingQuery
 
-根据时间分表的数据查询类
+分表数据查询类
 
 ## 使用方法
 
@@ -14,10 +14,16 @@ $shardingQuery = new ShardingQuery([
         'order_201805',
         'order_201804',
     ],
-    'field'    => '*',
-    'where'    => 'member_id = 10001',
-    'offset'   => 32,
-    'limit'    => 3,
+    'field'    => '{table}.*, u.name',
+    'leftJoin' => [
+        'user AS u ON u.member_id = {table}.member_id',
+    ],
+    'where'    => '{table}.member_id = 10001 AND status = 1',
+    'order'    => '{table}.add_time DESC',
+    'offset'   => 0,
+    'limit'    => 10,
 ]);
 $res           = $shardingQuery->select();
+$count         = $shardingQuery->count();
+$trace         = $shardingQuery->trace();
 ```

ShardingQuery.php

@@ -1,8 +1,7 @@
 <?php
 
 /**
- * 根据时间同构分表的查询类
- * @package app\common\library
+ * 分表数据查询类
  * @author LIUJIAN <coder.keda@gmail.com>
  */
 class ShardingQuery
@@ -28,6 +27,30 @@ class ShardingQuery
      */
     public $field;
 
+    /**
+     * join
+     * @var array
+     */
+    public $innerJoin;
+
+    /**
+     * join
+     * @var array
+     */
+    public $leftJoin;
+
+    /**
+     * join
+     * @var array
+     */
+    public $rightJoin;
+
+    /**
+     * join
+     * @var array
+     */
+    public $fullJoin;
+
     /**
      * 条件
      * @var string
@@ -58,6 +81,12 @@ class ShardingQuery
      */
     protected $stats;
 
+    /**
+     * 追踪
+     * @var array
+     */
+    protected $trace;
+
     /**
      * ShardingQuery constructor.
      * @param array $config
@@ -84,6 +113,30 @@ public function __construct(array $config)
                     }
                     $this->$key = $value;
                     break;
+                case 'innerJoin':
+                    if (!is_array($value)) {
+                        throw new \RuntimeException("'innerJoin' is not a array type.");
+                    }
+                    $this->$key = $value;
+                    break;
+                case 'leftJoin':
+                    if (!is_array($value)) {
+                        throw new \RuntimeException("'leftJoin' is not a array type.");
+                    }
+                    $this->$key = $value;
+                    break;
+                case 'rightJoin':
+                    if (!is_array($value)) {
+                        throw new \RuntimeException("'rightJoin' is not a array type.");
+                    }
+                    $this->$key = $value;
+                    break;
+                case 'fullJoin':
+                    if (!is_array($value)) {
+                        throw new \RuntimeException("'fullJoin' is not a array type.");
+                    }
+                    $this->$key = $value;
+                    break;
                 case 'where':
                     if (!is_string($value)) {
                         throw new \RuntimeException("'where' is not a string type.");
@@ -120,18 +173,17 @@ public function select()
     {
         $this->stats = $this->stats($this->table);
         $range       = $this->range($this->stats);
+        $sql         = $this->sql();
         $data        = [];
         foreach ($range as $tableName => $item) {
-            $sql = "SELECT {$this->field} FROM `{$tableName}`";
-            if (!empty($this->where)) {
-                $sql .= " WHERE {$this->where}";
-            }
-            if (!empty($this->order)) {
-                $sql .= " ORDER BY {$this->order}";
-            }
-            $sql    = "{$sql} LIMIT {$item['limit']} OFFSET {$item['offset']}";
-            $result = call_user_func($this->callback, $sql);
-            $data   = array_merge($data, $result);
+            $sql           = str_replace('{table}', $tableName, $sql);
+            $sql           = "{$sql} LIMIT {$item['limit']} OFFSET {$item['offset']}";
+            $result        = call_user_func($this->callback, $sql);
+            $data          = array_merge($data, $result);
+            $this->trace[] = [
+                'sql'      => $sql,
+                'rowCount' => count($result),
+            ];
         }
         return $data;
     }
@@ -214,6 +266,42 @@ protected function range($stats)
         return $tables;
     }
 
+    /**
+     * 生成sql
+     * @return string
+     */
+    protected function sql()
+    {
+        $sql = "SELECT {$this->field} FROM `{table}`";
+        if (!empty($this->innerJoin)) {
+            foreach ($this->innerJoin as $item) {
+                $sql .= " INNER JOIN {$item}";
+            }
+        }
+        if (!empty($this->leftJoin)) {
+            foreach ($this->leftJoin as $item) {
+                $sql .= " LEFT JOIN {$item}";
+            }
+        }
+        if (!empty($this->rightJoin)) {
+            foreach ($this->rightJoin as $item) {
+                $sql .= " RIGHT JOIN {$item}";
+            }
+        }
+        if (!empty($this->fullJoin)) {
+            foreach ($this->fullJoin as $item) {
+                $sql .= " FULL JOIN {$item}";
+            }
+        }
+        if (!empty($this->where)) {
+            $sql .= " WHERE {$this->where}";
+        }
+        if (!empty($this->order)) {
+            $sql .= " ORDER BY {$this->order}";
+        }
+        return $sql;
+    }
+
     /**
      * 获取数据总数
      * @return int
@@ -226,4 +314,13 @@ public function count()
         return $number ?: 0;
     }
 
+    /**
+     * 获取追踪数据
+     * @return array
+     */
+    public function trace()
+    {
+        return $this->trace;
+    }
+
 }