新增支持配置数据库名

Yurunsoft · Dec 21, 2021 · 2ec5d3e · 2ec5d3e
1 parent cd7d5d8
commit 2ec5d3e
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -33,6 +33,7 @@
     // 'port'            => 6041,
     // 'user'            => 'root',
     // 'password'        => 'taosdata',
+    // 'db'              => 'database'
     // 'ssl'             => false,
     // 'timestampFormat' => \Yurun\TDEngine\Constants\TimeStampFormat::LOCAL_STRING,
     // 'keepAlive'       => true,
@@ -52,6 +53,7 @@ $client = new \Yurun\TDEngine\Client(new \Yurun\TDEngine\ClientConfig([
     // 'port'            => 6041,
     // 'user'            => 'root',
     // 'password'        => 'taosdata',
+    // 'db'              => 'database'
     // 'ssl'             => false,
     // 'timestampFormat' => \Yurun\TDEngine\Constants\TimeStampFormat::LOCAL_STRING,
     // 'keepAlive'       => true,

diff --git a/src/Client.php b/src/Client.php
@@ -91,6 +91,11 @@ public function sql(string $sql): SqlResult
             default:
                 throw new InvalidArgumentException(sprintf('Invalid timestampFormat %s', $config->getTimestampFormat()));
         }
+        $db = $config->getDb();
+        if ('' !== $db)
+        {
+            $path .= '/' . $db;
+        }
         $result = $this->request($path, $sql);
 
         return new SqlResult($result);

diff --git a/src/ClientConfig.php b/src/ClientConfig.php
@@ -33,6 +33,11 @@ class ClientConfig
      */
     protected $password = 'taosdata';
 
+    /**
+     * @var string
+     */
+    protected $db = '';
+
     /**
      * @var bool
      */
@@ -119,6 +124,18 @@ public function setPassword(string $password): self
         return $this;
     }
 
+    public function getDb(): string
+    {
+        return $this->db;
+    }
+
+    public function setDb(string $db): self
+    {
+        $this->db = $db;
+
+        return $this;
+    }
+
     public function getSsl(): bool
     {
         return $this->ssl;