改进多数据库连接的切换

top-think · Dec 1, 2017 · dc72288 · dc72288
1 parent 97aa420
commit dc72288
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/db/Query.php b/src/db/Query.php
@@ -232,7 +232,13 @@ public function getTable($name = '')
     public function connect($config = [], $name = false)
     {
         $this->connection = Connection::instance($config, $name);
-        $this->prefix     = $this->connection->getConfig('prefix');
+        $query            = $this->connection->getConfig('query');
+
+        if (__CLASS__ != $query) {
+            return new $query($this->connection);
+        }
+
+        $this->prefix = $this->connection->getConfig('prefix');
 
         return $this;
     }