eee

ActiveQuery.php

@@ -204,7 +204,7 @@ class ActiveQuery extends QueryTrait implements ISqlBuilder
      */
     public function exists(): bool
     {
-        return $this->buildCommand($this->builder->one())->rowCount() > 0;
+        return $this->buildCommand($this->limit(1)->builder->exists())->exists();
     }
 
 

Command.php

@@ -75,6 +75,19 @@ class Command extends Component
         return $this->search('fetch');
     }
 
+    /**
+     * @return bool
+     * @throws Exception
+     */
+    public function exists(): bool
+    {
+        $data = $this->search('fetch');
+        if (!$data) {
+            return false;
+        }
+        return true;
+    }
+
     /**
      * @return mixed
      * @throws
@@ -88,46 +101,11 @@ class Command extends Component
      * @return mixed
      * @throws
      */
-    public function rowCount(): mixed
+    public function rowCount(): int
     {
-        $client = $this->connection->getConnection();
-        try {
-            $client->setAttribute(PDO::MYSQL_ATTR_USE_BUFFERED_QUERY, true);
-            if (($prepare = $client->query($this->sql)) === false) {
-                throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $client->errorInfo()[2]);
-            }
+        $data = $this->search('fetch');
 
-            $prepare->execute($this->params);
-
-            $count = $prepare->rowCount();
-            $prepare->closeCursor();
-
-            $this->connection->println($this->sql, $this->params);
-
-            return $count;
-        } catch (Throwable $throwable) {
-            if ($this->isRefresh($throwable)) return $this->rowCount();
-
-            $errorMsg = $throwable->getMessage() . PHP_EOL . '                 Sql: ' . $this->sql . '.' . json_encode($this->params);
-            $this->getLogger()->failure($errorMsg . PHP_EOL, 'mysql');
-            return 0;
-        } finally {
-            $this->connection->release($client);
-        }
-    }
-
-
-    /**
-     * @return bool
-     * @throws Exception
-     */
-    public function exists(): bool
-    {
-        $total = $this->search('rowCount');
-        if ($total === false) {
-            throw new Exception('Query data is has error.');
-        }
-        return $total > 0;
+        return !$data ? 0 : +current($data);
     }
 
 
@@ -182,7 +160,6 @@ class Command extends Component
     {
         $client = $this->connection->getConnection();
         try {
-            $this->connection->println($this->sql, $this->params);
             if (($prepare = $client->prepare($this->sql)) === false) {
                 throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $prepare->errorInfo()[2]);
             }

Connection.php

@@ -30,6 +30,8 @@ use Kiri\Di\Inject\Container;
 use Swoole\Timer;
 use Database\Base\PDO;
 
+//use PDO;
+
 /**
  * Class Connection
  * @package Database
@@ -122,14 +124,14 @@ class Connection extends Component
      */
     protected function checkClientHealth(Pool $pool): void
     {
-        $pool->flush($this->cds, $this->pool['min'] ?? 1);
-        $length = $pool->size($this->cds);
+        $pool->flush($this->getName(), $this->pool['min'] ?? 1);
+        $length = $pool->size($this->getName());
         for ($i = 0; $i < $length; $i++) {
             try {
                 if (($client = $this->validator($pool)) === false) {
                     break;
                 }
-                $pool->push($this->cds, $client);
+                $pool->push($this->getName(), $client);
             } catch (\Throwable $exception) {
                 if (!str_contains($exception->getMessage(), 'Client timeout.')) {
                     $this->logger->error(throwable($exception), [$this->cds]);
@@ -139,6 +141,15 @@ class Connection extends Component
     }
 
 
+    /**
+     * @return string
+     */
+    private function getName(): string
+    {
+        return 'mysql.' . $this->cds;
+    }
+
+
     /**
      * @param Pool $pool
      * @return PDO|bool
@@ -147,7 +158,7 @@ class Connection extends Component
     protected function validator(Pool $pool): PDO|bool
     {
         /** @var $client PDO */
-        if (($client = $pool->get($this->cds)) === false) {
+        if (($client = $pool->get($this->getName())) === false) {
             return false;
         }
         if ($client->query('select 1') === false) {
@@ -177,7 +188,7 @@ class Connection extends Component
      */
     protected function getNormalClientHealth(): PDO
     {
-        $data = $this->pool()->get($this->cds, $this->waite_time);
+        $data = $this->pool()->get($this->getName(), $this->waite_time);
         if ($data === false) {
             throw new Exception('Client Waite timeout.');
         }
@@ -303,7 +314,7 @@ class Connection extends Component
     public function release(PDO $pdo): void
     {
         if (!$this->inTransaction()) {
-            $this->pool()->push($this->cds, $pdo);
+            $this->pool()->push($this->getName(), $pdo);
         }
     }
 
@@ -315,7 +326,7 @@ class Connection extends Component
      */
     public function clear_connection(): void
     {
-        $this->pool()->flush($this->cds, 0);
+        $this->pool()->flush($this->getName(), 0);
     }
 
 
@@ -327,16 +338,17 @@ class Connection extends Component
         if ($this->timerId > -1) {
             Timer::clear($this->timerId);
         }
-        $this->pool()->close($this->cds);
+        $this->pool()->close($this->getName());
     }
 
 
     /**
      * @return PDO
      */
-    public function newConnect(): PDO
+    public function newConnect(): \PDO
     {
         $pdo = new PDO($this->database, $this->cds, $this->username, $this->password, [
+//        $pdo = new \PDO('mysql:dbname=' . $this->database . ';host=' . $this->cds, $this->username, $this->password, [
             \PDO::ATTR_CASE               => \PDO::CASE_NATURAL,
             \PDO::ATTR_ERRMODE            => \PDO::ERRMODE_EXCEPTION,
             \PDO::ATTR_ORACLE_NULLS       => \PDO::NULL_NATURAL,
@@ -357,8 +369,8 @@ class Connection extends Component
      */
     protected function pool(): Pool
     {
-        if (!$this->connections->hasChannel($this->cds)) {
-            $this->connections->created($this->cds, $this->pool['max'] ?? 1, [$this, 'newConnect']);
+        if (!$this->connections->hasChannel($this->getName())) {
+            $this->connections->created($this->getName(), $this->pool['max'] ?? 1, [$this, 'newConnect']);
         }
         return $this->connections;
     }

Db.php

@@ -115,10 +115,11 @@ class Db extends QueryTrait implements ISqlBuilder
 
     /**
      * @return bool
+     * @throws Exception
      */
     public function exists(): bool
     {
-        return $this->connection->createCommand(SqlBuilder::builder($this)->one())->rowCount() > 0;
+        return $this->connection->createCommand(SqlBuilder::builder($this->limit(1))->exists())->exists();
     }
 
     /**
@@ -151,6 +152,55 @@ class Db extends QueryTrait implements ISqlBuilder
         return $this->connection->createCommand(SqlBuilder::builder($this)->delete())->delete();
     }
 
+
+    /**
+     * @return bool|array|null
+     */
+    public function first(): bool|array|null
+    {
+        return $this->connection->createCommand(SqlBuilder::builder($this)->one())->one();
+    }
+
+
+    /**
+     * @return bool|array
+     */
+    public function get(): bool|array
+    {
+        return $this->connection->createCommand(SqlBuilder::builder($this)->all())->all();
+    }
+
+
+    /**
+     * @param string $sql
+     * @return mixed
+     */
+    public function exec(string $sql): mixed
+    {
+        return $this->connection->createCommand($sql)->exec();
+    }
+
+
+    /**
+     * @param string $sql
+     * @return array|bool|null
+     */
+    public function query(string $sql): array|bool|null
+    {
+        return $this->connection->createCommand($sql)->one();
+    }
+
+
+    /**
+     * @param string $sql
+     * @return array|bool
+     */
+    public function queryAll(string $sql): array|bool
+    {
+        return $this->connection->createCommand($sql)->all();
+    }
+
+
     /**
      * @param string $table
      * @return array|bool|null

SqlBuilder.php

@@ -235,7 +235,17 @@ class SqlBuilder extends Component
      */
     public function count(): string
     {
-        return $this->makeSelect() . $this->make();
+        return $this->makeSelect(['COUNT(*)']) . $this->make();
+    }
+
+
+    /**
+     * @return string
+     * @throws
+     */
+    public function exists(): string
+    {
+        return $this->makeSelect(['0']) . $this->make();
     }