优化

Author: onanying

ShardingQuery.php

@@ -178,18 +178,18 @@ public function __construct(array $config)
     public function select()
     {
         $this->stats = $this->stats($this->table);
-        $range       = $this->range($this->stats);
-        $sql         = $this->sql();
-        $data        = [];
+        $this->log(['stats' => $this->stats]);
+        $range = $this->range($this->stats);
+        $this->log(['range' => $range]);
+        $sql  = $this->sql();
+        $data = [];
         foreach ($range as $tableName => $item) {
-            $tmpSql        = str_replace(static::$tableSymbol, $tableName, $sql);
-            $tmpSql        = "{$tmpSql} LIMIT {$item['limit']} OFFSET {$item['offset']}";
-            $result        = call_user_func($this->callback, $tmpSql);
-            $data          = array_merge($data, $result);
-            $this->trace[] = [
-                'sql'      => $tmpSql,
-                'rowCount' => count($result),
-            ];
+            $tmpSql = str_replace(static::$tableSymbol, $tableName, $sql);
+            $tmpSql = "{$tmpSql} LIMIT {$item['limit']} OFFSET {$item['offset']}";
+            $this->log(['sql' => $tmpSql]);
+            $result = call_user_func($this->callback, $tmpSql);
+            $this->log(['resultCount' => count($result)]);
+            $data = array_merge($data, $result);
         }
         return $data;
     }
@@ -208,10 +208,12 @@ protected function stats($table)
             if (!empty($this->where)) {
                 $sql .= " WHERE {$this->where}";
             }
-            $sql               = str_replace(static::$tableSymbol, $tableName, $sql);
-            $result            = call_user_func($this->callback, $sql);
-            $first             = array_pop($result);
-            $count             = array_pop($first);
+            $sql = str_replace(static::$tableSymbol, $tableName, $sql);
+            $this->log(['sql' => $sql]);
+            $result = call_user_func($this->callback, $sql);
+            $first  = array_pop($result);
+            $count  = array_pop($first);
+            $this->log(['result' => $count]);
             $start             = $end;
             $end               += $count;
             $stats[$tableName] = [
@@ -243,31 +245,24 @@ protected function range($stats)
                     'offset' => $start - $item['start'],
                     'limit'  => $end - $start,
                 ];
-                continue;
-            }
-            if ($before) {
+            } elseif ($before) {
                 $tables[$table] = [
                     'offset' => $start - $item['start'],
                     'limit'  => $item['end'] - $start,
                 ];
-                if ($tables[$table]['limit'] == 0) {
-                    unset($tables[$table]);
-                }
-                continue;
-            }
-            if ($after) {
+            } elseif ($after) {
                 $tables[$table] = [
                     'offset' => 0,
                     'limit'  => $end - $item['start'],
                 ];
-                continue;
-            }
-            if ($center) {
+            } elseif ($center) {
                 $tables[$table] = [
                     'offset' => 0,
                     'limit'  => $item['end'] - $item['start'],
                 ];
-                continue;
+            }
+            if (isset($tables[$table]['limit']) && $tables[$table]['limit'] === 0) {
+                unset($tables[$table]);
             }
         }
         return $tables;
@@ -321,6 +316,14 @@ public function count()
         return $number ?: 0;
     }
 
+    /**
+     * 记录日志
+     */
+    protected function log($message)
+    {
+        $this->trace[] = $message;
+    }
+
     /**
      * 获取追踪数据
      * @return array