changed getDb()

hiqdev · Jan 23, 2017 · 1cd3d7b · 1cd3d7b
1 parent a766a44
commit 1cd3d7b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/AbstractConnection.php b/src/AbstractConnection.php
@@ -11,6 +11,7 @@
 namespace hiqdev\hiart;
 
 use Closure;
+use Yii;
 use hiqdev\hiart\stream\Request;
 use yii\base\Component;
 use yii\base\InvalidParamException;
@@ -36,6 +37,8 @@ abstract class AbstractConnection extends Component
 
     public $activeQueryClass = ActiveQuery::class;
 
+    public static $dbname = 'hiart';
+
     public $name = 'hiart';
 
     public $userAgent = 'HiArt/0.x';
@@ -74,6 +77,11 @@ abstract class AbstractConnection extends Component
      */
     protected $_errorChecker;
 
+    public static function getDb($dbname = null)
+    {
+        return Yii::$app->get($dbname ?: static::$dbname);
+    }
+
     public function setAuth($auth)
     {
         $this->_auth = $auth;

diff --git a/src/ActiveRecord.php b/src/ActiveRecord.php
@@ -10,7 +10,7 @@
 
 namespace hiqdev\hiart;
 
-use Yii;
+use hiqdev\hiart\AbstractConnection;
 use yii\base\InvalidConfigException;
 use yii\base\NotSupportedException;
 use yii\db\ActiveQueryInterface;
@@ -30,7 +30,7 @@ class ActiveRecord extends BaseActiveRecord
      */
     public static function getDb()
     {
-        return Yii::$app->get('hiart');
+        return AbstractConnection::getDb();
     }
 
     /**