eee

Command.php

@@ -1,260 +1,288 @@
-<?php
-/**
- * Created by PhpStorm.
- * User: whwyy
- * Date: 2018/3/30 0030
- * Time: 15:23
- */
-declare(strict_types=1);
-
-namespace Database;
-
-
-use Exception;
-use Kiri\Abstracts\Component;
-use Kiri\Di\Container;
-use PDO;
-use Throwable;
-
-/**
- * Class Command
- * @package Database
- */
-class Command extends Component
-{
-	
-	public Connection $connection;
-	public ?string    $sql    = '';
-	public array      $params = [];
-	
-	/**
-	 * @param array $params
-	 * @throws
-	 */
-	public function __construct(array $params = [])
-	{
-		parent::__construct();
-		Container::configure($this, $params);
-	}
-	
-	
-	/**
-	 * @return bool
-	 * @throws
-	 */
-	public function incrOrDecr(): bool
-	{
-		return (bool)$this->_prepare();
-	}
-	
-	
-	/**
-	 * @return bool|array
-	 * @throws
-	 */
-	public function all(): bool|array
-	{
-		return $this->search('fetchAll');
-	}
-	
-	/**
-	 * @return array|bool|null
-	 * @throws
-	 */
-	public function one(): array|null|bool
-	{
-		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
-	 */
-	public function fetchColumn(): mixed
-	{
-		return $this->search('fetchColumn');
-	}
-	
-	/**
-	 * @return mixed
-	 * @throws
-	 */
-	public function rowCount(): int
-	{
-		$data = $this->search('fetch');
-		
-		return !$data ? 0 : +current($data);
-	}
-	
-	
-	/**
-	 * @param string $method
-	 * @return mixed
-	 * @throws
-	 */
-	protected function search(string $method): mixed
-	{
-		$client = $this->connection->getConnection();
-		try {
-			$startTime = microtime(true);
-			if (($prepare = $client->prepare($this->sql)) === false) {
-				throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $client->errorInfo()[2]);
-			}
-			
-			$prepare->execute($this->params);
-			
-			$result = $method == 'rowCount' ? $prepare->rowCount() : $prepare->{$method}(PDO::FETCH_ASSOC);
-			$prepare->closeCursor();
-			
-			$this->connection->println($startTime, microtime(true), $this->sql, $this->params);
-			
-			return $result;
-		} catch (Throwable $throwable) {
-			$this->getLogger()->json_log($throwable);
-			
-			if ($this->isRefresh($throwable)) {
-				return $this->search($method);
-			}
-			
-			$errorMsg = $throwable->getMessage() . PHP_EOL . '                 Sql: ' . $this->sql . '.' . json_encode($this->params);
-			
-			return $this->getLogger()->logCategory($errorMsg . PHP_EOL, 'mysql');
-		} finally {
-			$this->connection->release($client);
-		}
-	}
-	
-	
-	/**
-	 * @return bool
-	 * @throws
-	 */
-	public function flush(): bool
-	{
-		return (bool)$this->_prepare();
-	}
-	
-	
-	/**
-	 * @return int|bool
-	 * @throws
-	 */
-	private function _prepare(): int|bool
-	{
-		$client = $this->connection->getConnection();
-		try {
-			$startTime = microtime(true);
-			if (($prepare = $client->prepare($this->sql)) === false) {
-				throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $prepare->errorInfo()[2]);
-			}
-			if ($prepare->execute($this->params) === false) {
-				throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $prepare->errorInfo()[2]);
-			}
-			
-			$prepare->closeCursor();
-			
-			$result = $client->lastInsertId();
-			
-			$this->connection->println($startTime, microtime(true), $this->sql, $this->params);
-			if (str_starts_with($this->sql, 'DELETE')) {
-				return $prepare->rowCount();
-			}
-			
-			return $result == 0 ? $prepare->rowCount() : (int)$result;
-		} catch (Throwable $throwable) {
-			$this->getLogger()->json_log($throwable);
-			
-			if ($this->isRefresh($throwable)) {
-				return $this->_prepare();
-			}
-			
-			$errorMsg = $throwable->getMessage() . PHP_EOL . '                 Sql: ' . $this->sql . '.' . json_encode($this->params, JSON_UNESCAPED_UNICODE);
-			
-			return $this->getLogger()->logCategory($errorMsg . PHP_EOL, 'mysql');
-		} finally {
-			$this->connection->release($client);
-		}
-	}
-	
-	
-	/**
-	 * @param Throwable $throwable
-	 * @return bool
-	 */
-	protected function isRefresh(Throwable $throwable): bool
-	{
-		$message = $throwable->getMessage();
-		
-		// MySQL 错误处理
-		if (str_contains($message, 'MySQL server has gone away')) {
-			return true;
-		}
-		if (str_contains($message, 'Send of 14 bytes failed with errno=32 Broken pipe')) {
-			return true;
-		}
-		if (str_contains($message, 'Lost connection to MySQL server during query')) {
-			return true;
-		}
-		
-		// PostgreSQL 错误处理
-		if (str_contains($message, 'server closed the connection unexpectedly')) {
-			return true;
-		}
-		if (str_contains($message, 'Connection refused')) {
-			return true;
-		}
-		if (str_contains($message, 'Broken pipe')) {
-			return true;
-		}
-		if (str_contains($message, 'connection to server was lost')) {
-			return true;
-		}
-		
-		return false;
-	}
-	
-	
-	/**
-	 * @return bool
-	 * @throws
-	 */
-	public function delete(): bool
-	{
-		return $this->_prepare();
-	}
-	
-	
-	/**
-	 * @return int|bool
-	 */
-	public function exec(): int|bool
-	{
-		return $this->_prepare();
-	}
-	
-	/**
-	 * @param array $data
-	 * @return $this
-	 */
-	public function bindValues(array $data = []): static
-	{
-		if (count($data) > 0) {
-			$this->params = array_merge($this->params, $data);
-		}
-		return $this;
-	}
-	
-}
+<?php
+/**
+ * Created by PhpStorm.
+ * User: whwyy
+ * Date: 2018/3/30 0030
+ * Time: 15:23
+ */
+declare(strict_types=1);
+
+namespace Database;
+
+
+use Exception;
+use Kiri\Abstracts\Component;
+use Kiri\Di\Container;
+use PDO;
+use Throwable;
+
+/**
+ * Class Command
+ * @package Database
+ */
+class Command extends Component
+{
+
+	/** @var int 数据库操作最大重试次数 */
+	private const MAX_RETRIES = 2;
+
+	public Connection $connection;
+	public ?string    $sql    = '';
+	public array      $params = [];
+	
+	/**
+	 * @param array $params
+	 * @throws
+	 */
+	public function __construct(array $params = [])
+	{
+		parent::__construct();
+		Container::configure($this, $params);
+	}
+	
+	
+	/**
+	 * @return bool
+	 * @throws
+	 */
+	public function incrOrDecr(): bool
+	{
+		return (bool)$this->_prepare();
+	}
+	
+	
+	/**
+	 * @return bool|array
+	 * @throws
+	 */
+	public function all(): bool|array
+	{
+		return $this->search('fetchAll');
+	}
+	
+	/**
+	 * @return array|bool|null
+	 * @throws
+	 */
+	public function one(): array|null|bool
+	{
+		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
+	 */
+	public function fetchColumn(): mixed
+	{
+		return $this->search('fetchColumn');
+	}
+	
+	/**
+	 * @return mixed
+	 * @throws
+	 */
+	public function rowCount(): int
+	{
+		$data = $this->search('fetch');
+		
+		return !$data ? 0 : +current($data);
+	}
+	
+	
+	/**
+	 * 执行查询类 SQL 操作，内置断连重试机制
+	 * 当数据库连接断开时自动重试，断开的连接不再归还池中防止污染连接池
+	 * @param string $method PDO 结果获取方法名
+	 * @return mixed
+	 * @throws
+	 */
+	protected function search(string $method): mixed
+	{
+		$retryCount = 0;
+
+		while ($retryCount <= self::MAX_RETRIES) {
+			$client = $this->connection->getConnection();
+			try {
+				$startTime = microtime(true);
+				if (($prepare = $client->prepare($this->sql)) === false) {
+					throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $client->errorInfo()[2]);
+				}
+				
+				$prepare->execute($this->params);
+				
+				$result = $method == 'rowCount' ? $prepare->rowCount() : $prepare->{$method}(PDO::FETCH_ASSOC);
+				$prepare->closeCursor();
+				
+				$this->connection->println($startTime, microtime(true), $this->sql, $this->params);
+				
+				$this->connection->release($client);
+				return $result;
+			} catch (Throwable $throwable) {
+				$this->getLogger()->json_log($throwable);
+				
+				if ($this->isRefresh($throwable) && $retryCount < self::MAX_RETRIES) {
+					$retryCount++;
+					$this->connection->connections->abandon($this->connection->getName());
+					continue;
+				}
+				
+				$this->connection->release($client);
+				
+				$errorMsg = $throwable->getMessage() . PHP_EOL . '                 Sql: ' . $this->sql . '.' . json_encode($this->params);
+				
+				return $this->getLogger()->logCategory($errorMsg . PHP_EOL, 'mysql');
+			}
+		}
+
+		$errorMsg = 'Max retry exceeded for SQL: ' . $this->sql;
+		return $this->getLogger()->logCategory($errorMsg, 'mysql');
+	}
+	
+	
+	/**
+	 * @return bool
+	 * @throws
+	 */
+	public function flush(): bool
+	{
+		return (bool)$this->_prepare();
+	}
+	
+	
+	/**
+	 * 执行写入类 SQL 操作，内置断连重试机制
+	 * 当数据库连接断开时自动重试，断开的连接不再归还池中防止污染连接池
+	 * @return int|bool
+	 * @throws
+	 */
+	private function _prepare(): int|bool
+	{
+		$retryCount = 0;
+
+		while ($retryCount <= self::MAX_RETRIES) {
+			$client = $this->connection->getConnection();
+			try {
+				$startTime = microtime(true);
+				if (($prepare = $client->prepare($this->sql)) === false) {
+					throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $prepare->errorInfo()[2]);
+				}
+				if ($prepare->execute($this->params) === false) {
+					throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $prepare->errorInfo()[2]);
+				}
+				
+				$prepare->closeCursor();
+				
+				$result = $client->lastInsertId();
+				
+				$this->connection->println($startTime, microtime(true), $this->sql, $this->params);
+				if (str_starts_with($this->sql, 'DELETE')) {
+					$this->connection->release($client);
+					return $prepare->rowCount();
+				}
+				
+				$this->connection->release($client);
+				return $result == 0 ? $prepare->rowCount() : (int)$result;
+			} catch (Throwable $throwable) {
+				$this->getLogger()->json_log($throwable);
+				
+				if ($this->isRefresh($throwable) && $retryCount < self::MAX_RETRIES) {
+					$retryCount++;
+					$this->connection->connections->abandon($this->connection->getName());
+					continue;
+				}
+				
+				$this->connection->release($client);
+				
+				$errorMsg = $throwable->getMessage() . PHP_EOL . '                 Sql: ' . $this->sql . '.' . json_encode($this->params, JSON_UNESCAPED_UNICODE);
+				
+				return $this->getLogger()->logCategory($errorMsg . PHP_EOL, 'mysql');
+			}
+		}
+
+		$errorMsg = 'Max retry exceeded for SQL: ' . $this->sql;
+		return $this->getLogger()->logCategory($errorMsg, 'mysql');
+	}
+	
+	
+	/**
+	 * @param Throwable $throwable
+	 * @return bool
+	 */
+	protected function isRefresh(Throwable $throwable): bool
+	{
+		$message = $throwable->getMessage();
+		
+		// MySQL 错误处理
+		if (str_contains($message, 'MySQL server has gone away')) {
+			return true;
+		}
+		if (str_contains($message, 'Send of 14 bytes failed with errno=32 Broken pipe')) {
+			return true;
+		}
+		if (str_contains($message, 'Lost connection to MySQL server during query')) {
+			return true;
+		}
+		
+		// PostgreSQL 错误处理
+		if (str_contains($message, 'server closed the connection unexpectedly')) {
+			return true;
+		}
+		if (str_contains($message, 'Connection refused')) {
+			return true;
+		}
+		if (str_contains($message, 'Broken pipe')) {
+			return true;
+		}
+		if (str_contains($message, 'connection to server was lost')) {
+			return true;
+		}
+		
+		return false;
+	}
+	
+	
+	/**
+	 * @return bool
+	 * @throws
+	 */
+	public function delete(): bool
+	{
+		return $this->_prepare();
+	}
+	
+	
+	/**
+	 * @return int|bool
+	 */
+	public function exec(): int|bool
+	{
+		return $this->_prepare();
+	}
+	
+	/**
+	 * @param array $data
+	 * @return $this
+	 */
+	public function bindValues(array $data = []): static
+	{
+		if (count($data) > 0) {
+			$this->params = array_merge($this->params, $data);
+		}
+		return $this;
+	}
+	
+}

Connection.php

@@ -248,61 +248,65 @@ class Connection extends Component
     }
 
 
-    /**
-     * @throws
-     * 事务回滚
-     */
-    public function rollback(): void
-    {
-        $this->storey--;
-        if ($this->storey == 0) {
-            if (!Context::exists($this->cds)) {
-                return;
-            }
-            $pdo = $this->getTransactionClient();
-            if ($pdo->inTransaction()) {
-                $pdo->rollback();
-            }
-            $this->release($pdo);
-        }
-    }
-
-    /**
-     * @throws
-     * 事务提交
-     */
-    public function commit(): void
-    {
-        $this->storey--;
-        if ($this->storey == 0) {
-            if (!Context::exists($this->cds)) {
-                return;
-            }
-            $pdo = $this->getTransactionClient();
-            if ($pdo->inTransaction()) {
-                $pdo->commit();
-            }
-            $this->release($pdo);
-        }
-    }
-
-
-    /**
-     * @return void
-     * @throws
-     */
-    public function clear(): void
-    {
-        /** @var PDO $pdo */
-        $pdo = Context::get($this->cds);
-        if ($pdo === null) {
-            return;
-        }
-        if ($this->inTransaction()) {
-            $pdo->rollback();
-        }
-        $this->release($pdo);
-    }
+    /**
+     * @throws
+     * 事务回滚
+     */
+    public function rollback(): void
+    {
+        $this->storey--;
+        if ($this->storey == 0) {
+            if (!Context::exists($this->cds)) {
+                return;
+            }
+            $pdo = $this->getTransactionClient();
+            if ($pdo->inTransaction()) {
+                $pdo->rollback();
+            }
+            $this->release($pdo);
+            Context::remove($this->cds);
+        }
+    }
+
+    /**
+     * @throws
+     * 事务提交
+     */
+    public function commit(): void
+    {
+        $this->storey--;
+        if ($this->storey == 0) {
+            if (!Context::exists($this->cds)) {
+                return;
+            }
+            $pdo = $this->getTransactionClient();
+            if ($pdo->inTransaction()) {
+                $pdo->commit();
+            }
+            $this->release($pdo);
+            Context::remove($this->cds);
+        }
+    }
+
+
+    /**
+     * @return void
+     * @throws
+     */
+    public function clear(): void
+    {
+        /** @var PDO $pdo */
+        $pdo = Context::get($this->cds);
+        if ($pdo === null) {
+            return;
+        }
+        if ($this->inTransaction()) {
+            $pdo->rollback();
+        }
+        $this->release($pdo);
+        Context::remove($this->cds);
+        $this->storey = 0;
+    }
 
 
     /**

SqlBuilder.php

@@ -1,462 +1,470 @@
-<?php
-/** @noinspection ALL */
-
-declare(strict_types=1);
-
-namespace Database;
-
-
-use JetBrains\PhpStorm\Pure;
-use Kiri;
-use Kiri\Abstracts\Component;
-
-
-/**
- * Class SqlBuilder
- * @package Database
- */
-class SqlBuilder extends Component
-{
-
-    /**
-     * @var ActiveQuery|Query|ISqlBuilder|null
-     */
-    public ActiveQuery|Query|ISqlBuilder|null $query;
-
-
-    /**
-     * @param ActiveQuery|Query|ISqlBuilder|null $config
-     */
-    public function __construct(ActiveQuery|Query|null|ISqlBuilder $config)
-    {
-        parent::__construct();
-
-        $this->query = $config;
-    }
-
-
-    /**
-     * @param ISqlBuilder|null $query
-     * @return $this
-     * @throws
-     */
-    public static function builder(ISqlBuilder|null $query): static
-    {
-        return new static($query);
-    }
-
-
-    /**
-     * @return string
-     * @throws
-     */
-    public function getCondition(): string
-    {
-        return $this->where($this->query);
-    }
-
-
-    /**
-     * @param array $compiler
-     * @return string
-     * @throws
-     */
-    public function hashCompiler(array $compiler): string
-    {
-        return $this->where($compiler);
-    }
-
-
-    /**
-     * @param array $attributes
-     * @return bool|array
-     * @throws
-     */
-    public function update(array $attributes): bool|string
-    {
-        $conditions  = $this->query;
-        $this->query = [];
-        $data        = $this->__updateBuilder($this->makeParams($attributes));
-        foreach ($conditions as $condition) {
-            $this->query->pushParam($condition);
-        }
-        return $data;
-    }
-
-
-    /**
-     * @param array $attributes
-     * @param string $opera
-     * @return bool|array
-     * @throws
-     */
-    public function mathematics(array $attributes, string $opera = '+'): bool|string
-    {
-        $string = [];
-        foreach ($attributes as $attribute => $value) {
-            $string[] = $attribute . '=' . $attribute . $opera . $value;
-        }
-        return $this->__updateBuilder($string);
-    }
-
-
-    /**
-     * @param array $string
-     * @return string|bool
-     * @throws
-     */
-    private function __updateBuilder(array $string): string|bool
-    {
-        if (empty($string)) {
-            return Kiri::getLogger()->logCategory('None data update.');
-        }
-        return 'UPDATE ' . $this->query . ' SET ' . implode(',', $string) . $this->make();
-    }
-
-
-    /**
-     * @param array $attributes
-     * @param false $isBatch
-     * @return array
-     * @throws
-     */
-    public function insert(array $attributes, bool $isBatch = false): string
-    {
-        $update = 'INSERT INTO ' . $this->query;
-        if ($isBatch === false) {
-            $attributes = [$attributes];
-        }
-        $update .= '(' . implode(',', $this->getFields($attributes)) . ') VALUES ';
-
-        $keys = [];
-        foreach ($attributes as $attribute) {
-            $_keys = $this->makeParams($attribute, true);
-
-            $keys[] = implode(',', $_keys);
-        }
-        return $update . '(' . implode('),(', $keys) . ')';
-    }
-
-
-    /**
-     * @return string
-     * @throws
-     */
-    public function delete(): string
-    {
-        return 'DELETE FROM ' . $this->query . $this->make();
-    }
-
-
-    /**
-     * @param $attributes
-     * @return array
-     */
-    #[Pure] private function getFields(array $attributes): array
-    {
-        return array_keys(current($attributes));
-    }
-
-
-    /**
-     * @param array $attributes
-     * @param bool $isInsert
-     * @return array[]
-     * a=:b,
-     */
-    private function makeParams(array $attributes, bool $isInsert = false): array
-    {
-        $keys = [];
-        foreach ($attributes as $key => $value) {
-            if ($isInsert === true) {
-                $keys[] = '?';
-                $this->query->pushParam($value);
-            } else {
-                $keys = $this->resolveParams($key, $value, $keys);
-            }
-        }
-        return $keys;
-    }
-
-
-    /**
-     * @param string $key
-     * @param mixed $value
-     * @param array $keys
-     * @return array
-     */
-    private function resolveParams(string $key, mixed $value, array $keys): array
-    {
-        if (is_null($value)) {
-            return $keys;
-        }
-        if (preg_match('/^[+|-]\s\d+$/', (string)$value)) {
-            $keys[] = $key . '=' . $key . ' ' . $value;
-        } else {
-            $this->query->pushParam($value);
-            $keys[] = $key . '= ?';
-        }
-        return $keys;
-    }
-
-
-    /**
-     * @return string
-     * @throws
-     */
-    public function one(): string
-    {
-        $this->query->offset(0)->limit(1);
-        return $this->makeSelect($this->query) . $this->make() . $this->makeLimit();
-    }
-
-
-    /**
-     * @return string
-     * @throws
-     */
-    public function all(): string
-    {
-        return $this->makeSelect($this->query) . $this->make() . $this->makeLimit();
-    }
-
-
-    /**
-     * @return string
-     * @throws
-     */
-    public function count(): string
-    {
-        return $this->makeSelect(['COUNT(*)']) . $this->make();
-    }
-
-
-    /**
-     * @return string
-     * @throws
-     */
-    public function exists(): string
-    {
-        return $this->makeSelect(['0']) . $this->make();
-    }
-
-
-    /**
-     * @param string $table
-     * @return string
-     * @throws
-     */
-    public function columns(string $table): string
-    {
-        $driver = $this->getDriver();
-        if (in_array($driver, ['pgsql', 'postgresql'])) {
-            // PostgreSQL 使用 information_schema
-            $tableName = trim($table, '"`');
-            return "SELECT 
-                column_name as Field,
-                data_type as Type,
-                is_nullable as Null,
-                column_default as Default,
-                '' as Extra,
-                '' as Key
-            FROM information_schema.columns 
-            WHERE table_name = '$tableName' 
-            ORDER BY ordinal_position";
-        }
-        // MySQL 使用 SHOW FULL FIELDS
-        return 'SHOW FULL FIELDS FROM ' . $table;
-    }
-
-
-    /**
-     * @return string
-     * @throws
-     */
-    private function make(): string
-    {
-        $select = $this->makeCondition();
-        $select .= $this->makeGroup();
-        $select .= $this->makeOrder();
-        return $select;
-    }
-
-    /**
-     * @param array $select
-     * @return string
-     */
-    private function makeSelect(array $select = ['*']): string
-    {
-        $select = "SELECT " . implode(',', $select) . " FROM " . $this->query;
-        if ($this->query != "") {
-            $select .= " AS " . $this->query;
-        }
-        if (count($this->query) > 0) {
-            $select .= ' ' . implode(' ', $this->query);
-        }
-        return $select;
-    }
-
-
-    /**
-     * @return string
-     */
-    private function makeGroup(): string
-    {
-        if ($this->query != "") {
-            return ' GROUP BY ' . $this->query;
-        }
-        return '';
-    }
-
-
-    /**
-     * @return string
-     */
-    private function makeOrder(): string
-    {
-        if (count($this->query) > 0) {
-            return ' ORDER BY ' . implode(',', $this->query);
-        }
-        return '';
-    }
-
-
-    /**
-     * @return string
-     */
-    private function makeCondition(): string
-    {
-        $condition = $this->where($this->query);
-        if (empty($condition)) {
-            return '';
-        }
-        return ' WHERE ' . $condition;
-    }
-
-
-    /**
-     * @return string
-     * @throws
-     */
-    private function makeLimit(): string
-    {
-        if ($this->query >= 0 && $this->query >= 1) {
-            $driver = $this->getDriver();
-            if (in_array($driver, ['pgsql', 'postgresql'])) {
-                // PostgreSQL 使用 LIMIT ... OFFSET ... 语法
-                return ' LIMIT ' . $this->query . ' OFFSET ' . $this->query;
-            }
-            // MySQL 使用 LIMIT offset,limit 语法
-            return ' LIMIT ' . $this->query . ',' . $this->query;
-        }
-        return '';
-    }
-
-
-    /**
-     * @param false $isCount
-     * @return string
-     * @throws
-     */
-    public function get(bool $isCount = false): string
-    {
-        if ($isCount === false) {
-            return $this->all();
-        }
-        return $this->count();
-    }
-
-
-    /**
-     * @return string
-     * @throws
-     */
-    public function truncate(): string
-    {
-        return sprintf('TRUNCATE %s', $this->query);
-    }
-
-
-    /**
-     * @param array $where
-     * @return string
-     */
-    private function where(array $where): string
-    {
-        if (count($where) < 1) {
-            return '';
-        }
-        $_tmp = [];
-        foreach ($where as $key => $value) {
-            $_tmp[] = $this->resolveCondition($key, $value);
-        }
-        return implode(' AND ', $_tmp);
-    }
-
-
-    /**
-     * @param string $field
-     * @param mixed $condition
-     * @return string
-     */
-    private function resolveCondition(string|int $field, mixed $condition): string
-    {
-        if (is_string($field)) {
-            $this->query->pushParam($condition);
-            return $field . ' = ?';
-        } else if (is_string($condition)) {
-            return $condition;
-        } else {
-            return implode(' AND ', $this->_hashMap($condition));
-        }
-    }
-
-
-    /**
-     * @param $condition
-     * @return array
-     */
-    private function _hashMap(array $condition): array
-    {
-        $_array = [];
-        foreach ($condition as $key => $value) {
-            $this->query->pushParam($value);
-            $_array[] = $key . '= ?';
-        }
-        return $_array;
-    }
-
-    /**
-     * 获取数据库驱动类型
-     * @return string
-     * @throws
-     */
-    private function getDriver(): string
-    {
-        try {
-            // 尝试从 query 对象获取 connection
-            if ($this->query instanceof ActiveQuery && $this->query->modelClass !== null) {
-                if ($this->query->modelClass instanceof Model) {
-                    $connection = $this->query->modelClass->getConnection();
-                    if ($connection instanceof Connection) {
-                        return strtolower($connection->driver ?? 'mysql');
-                    }
-                }
-            }
-            // 如果是 Db 查询，尝试获取默认 connection
-            if (method_exists($this->query, 'getConnection')) {
-                $connection = $this->query->getConnection();
-                if ($connection instanceof Connection) {
-                    return strtolower($connection->driver ?? 'mysql');
-                }
-            }
-        } catch (\Throwable $e) {
-            // 忽略错误，返回默认值
-			$this->getLogger()->json_log($e);
-		}
-        // 默认返回 mysql
-        return 'mysql';
-    }
-
-
-}
+<?php
+/** @noinspection ALL */
+
+declare(strict_types=1);
+
+namespace Database;
+
+
+use JetBrains\PhpStorm\Pure;
+use Kiri;
+use Kiri\Abstracts\Component;
+use Database\Traits\QueryTrait;
+
+
+/**
+ * Class SqlBuilder
+ * @package Database
+ */
+class SqlBuilder extends Component
+{
+
+    /**
+     * @var ActiveQuery|Query|ISqlBuilder|null
+     */
+    public ActiveQuery|Query|ISqlBuilder|null $query;
+
+
+    /**
+     * @param ActiveQuery|Query|ISqlBuilder|null $config
+     */
+    public function __construct(ActiveQuery|Query|null|ISqlBuilder $config)
+    {
+        parent::__construct();
+
+        $this->query = $config;
+    }
+
+
+    /**
+     * @param ISqlBuilder|null $query
+     * @return $this
+     * @throws
+     */
+    public static function builder(ISqlBuilder|null $query): static
+    {
+        return new static($query);
+    }
+
+
+    /**
+     * @return string
+     * @throws
+     */
+    public function getCondition(): string
+    {
+        return $this->where($this->query->where);
+    }
+
+
+    /**
+     * @param array $compiler
+     * @return string
+     * @throws
+     */
+    public function hashCompiler(array $compiler): string
+    {
+        return $this->where($compiler);
+    }
+
+
+    /**
+     * @param array $attributes
+     * @return bool|array
+     * @throws
+     */
+    public function update(array $attributes): bool|string
+    {
+        $conditions  = $this->query;
+        $this->query = [];
+        $data        = $this->__updateBuilder($this->makeParams($attributes));
+        foreach ($conditions as $condition) {
+            $this->query->pushParam($condition);
+        }
+        return $data;
+    }
+
+
+    /**
+     * @param array $attributes
+     * @param string $opera
+     * @return bool|array
+     * @throws
+     */
+    public function mathematics(array $attributes, string $opera = '+'): bool|string
+    {
+        $string = [];
+        foreach ($attributes as $attribute => $value) {
+            $string[] = $attribute . '=' . $attribute . $opera . $value;
+        }
+        return $this->__updateBuilder($string);
+    }
+
+
+    /**
+     * @param array $string
+     * @return string|bool
+     * @throws
+     */
+    private function __updateBuilder(array $string): string|bool
+    {
+        if (empty($string)) {
+            return Kiri::getLogger()->logCategory('None data update.');
+        }
+        return 'UPDATE ' . $this->getFromAlias() . ' SET ' . implode(',', $string) . $this->make();
+    }
+
+
+    /**
+     * @param array $attributes
+     * @param false $isBatch
+     * @return array
+     * @throws
+     */
+    public function insert(array $attributes, bool $isBatch = false): string
+    {
+        $update = 'INSERT INTO ' . $this->getFromAlias();
+        if ($isBatch === false) {
+            $attributes = [$attributes];
+        }
+        $update .= '(' . implode(',', $this->getFields($attributes)) . ') VALUES ';
+
+        $keys = [];
+        foreach ($attributes as $attribute) {
+            $_keys = $this->makeParams($attribute, true);
+
+            $keys[] = implode(',', $_keys);
+        }
+        return $update . '(' . implode('),(', $keys) . ')';
+    }
+
+
+    /**
+     * @return string
+     * @throws
+     */
+    public function delete(): string
+    {
+        return 'DELETE FROM ' . $this->getFromAlias() . $this->make();
+    }
+
+
+    /**
+     * @param $attributes
+     * @return array
+     */
+    #[Pure] private function getFields(array $attributes): array
+    {
+        return array_keys(current($attributes));
+    }
+
+
+    /**
+     * @param array $attributes
+     * @param bool $isInsert
+     * @return array[]
+     * a=:b,
+     */
+    private function makeParams(array $attributes, bool $isInsert = false): array
+    {
+        $keys = [];
+        foreach ($attributes as $key => $value) {
+            if ($isInsert === true) {
+                $keys[] = '?';
+                $this->query->pushParam($value);
+            } else {
+                $keys = $this->resolveParams($key, $value, $keys);
+            }
+        }
+        return $keys;
+    }
+
+
+    /**
+     * @param string $key
+     * @param mixed $value
+     * @param array $keys
+     * @return array
+     */
+    private function resolveParams(string $key, mixed $value, array $keys): array
+    {
+        if (is_null($value)) {
+            return $keys;
+        }
+        if (preg_match('/^[+|-]\s\d+$/', (string)$value)) {
+            $keys[] = $key . '=' . $key . ' ' . $value;
+        } else {
+            $this->query->pushParam($value);
+            $keys[] = $key . '= ?';
+        }
+        return $keys;
+    }
+
+
+    /**
+     * @return string
+     * @throws
+     */
+    public function one(): string
+    {
+        $this->query->offset(0)->limit(1);
+        return $this->makeSelect($this->query) . $this->make() . $this->makeLimit();
+    }
+
+
+    /**
+     * @return string
+     * @throws
+     */
+    public function all(): string
+    {
+        return $this->makeSelect($this->query) . $this->make() . $this->makeLimit();
+    }
+
+
+    /**
+     * @return string
+     * @throws
+     */
+    public function count(): string
+    {
+        return $this->makeSelect(['COUNT(*)']) . $this->make();
+    }
+
+
+    /**
+     * @return string
+     * @throws
+     */
+    public function exists(): string
+    {
+        return $this->makeSelect(['0']) . $this->make();
+    }
+
+
+    /**
+     * @param string $table
+     * @return string
+     * @throws
+     */
+    public function columns(string $table): string
+    {
+        $driver = $this->getDriver();
+        if (in_array($driver, ['pgsql', 'postgresql'])) {
+            $tableName = trim($table, '"`');
+            return "SELECT 
+                column_name as Field,
+                data_type as Type,
+                is_nullable as Null,
+                column_default as Default,
+                '' as Extra,
+                '' as Key
+            FROM information_schema.columns 
+            WHERE table_name = '$tableName' 
+            ORDER BY ordinal_position";
+        }
+        return 'SHOW FULL FIELDS FROM ' . $table;
+    }
+
+
+    /**
+     * @return string
+     * @throws
+     */
+    private function make(): string
+    {
+        $select = $this->makeCondition();
+        $select .= $this->makeGroup();
+        $select .= $this->makeOrder();
+        return $select;
+    }
+
+    /**
+     * @param mixed $select 列名数组或 QueryTrait 对象
+     * @return string
+     */
+    private function makeSelect(mixed $select = ['*']): string
+    {
+        if (!is_array($select)) {
+            $select = ['*'];
+        }
+        $sql = "SELECT " . implode(',', $select) . " FROM " . $this->getFromAlias();
+        if ($this->query->join !== []) {
+            $sql .= ' ' . implode(' ', $this->query->join);
+        }
+        return $sql;
+    }
+
+
+    /**
+     * @return string
+     */
+    private function makeGroup(): string
+    {
+        if ($this->query->group !== '' && $this->query->group !== '0') {
+            return ' GROUP BY ' . $this->query->group;
+        }
+        return '';
+    }
+
+
+    /**
+     * @return string
+     */
+    private function makeOrder(): string
+    {
+        if (count($this->query->order) > 0) {
+            return ' ORDER BY ' . implode(',', $this->query->order);
+        }
+        return '';
+    }
+
+
+    /**
+     * @return string
+     */
+    private function makeCondition(): string
+    {
+        $condition = $this->where($this->query->where);
+        if (empty($condition)) {
+            return '';
+        }
+        return ' WHERE ' . $condition;
+    }
+
+
+    /**
+     * @return string
+     * @throws
+     */
+    private function makeLimit(): string
+    {
+        if ($this->query->offset >= 0 && $this->query->limit >= 1) {
+            $driver = $this->getDriver();
+            if (in_array($driver, ['pgsql', 'postgresql'])) {
+                return ' LIMIT ' . $this->query->limit . ' OFFSET ' . $this->query->offset;
+            }
+            return ' LIMIT ' . $this->query->offset . ',' . $this->query->limit;
+        }
+        return '';
+    }
+
+
+    /**
+     * @param false $isCount
+     * @return string
+     * @throws
+     */
+    public function get(bool $isCount = false): string
+    {
+        if ($isCount === false) {
+            return $this->all();
+        }
+        return $this->count();
+    }
+
+
+    /**
+     * @return string
+     * @throws
+     */
+    public function truncate(): string
+    {
+        return sprintf('TRUNCATE %s', $this->getFromAlias());
+    }
+
+
+    /**
+     * @param array $where
+     * @return string
+     */
+    private function where(array $where): string
+    {
+        if (count($where) < 1) {
+            return '';
+        }
+        $_tmp = [];
+        foreach ($where as $key => $value) {
+            $_tmp[] = $this->resolveCondition($key, $value);
+        }
+        return implode(' AND ', $_tmp);
+    }
+
+
+    /**
+     * @param string $field
+     * @param mixed $condition
+     * @return string
+     */
+    private function resolveCondition(string|int $field, mixed $condition): string
+    {
+        if (is_string($field)) {
+            $this->query->pushParam($condition);
+            return $field . ' = ?';
+        } else if (is_string($condition)) {
+            return $condition;
+        } else {
+            return implode(' AND ', $this->_hashMap($condition));
+        }
+    }
+
+
+    /**
+     * @param $condition
+     * @return array
+     */
+    private function _hashMap(array $condition): array
+    {
+        $_array = [];
+        foreach ($condition as $key => $value) {
+            $this->query->pushParam($value);
+            $_array[] = $key . '= ?';
+        }
+        return $_array;
+    }
+
+    /**
+     * 获取数据库驱动类型
+     * @return string
+     * @throws
+     */
+    private function getDriver(): string
+    {
+        try {
+            if ($this->query instanceof ActiveQuery && $this->query->modelClass !== null) {
+                if ($this->query->modelClass instanceof Model) {
+                    $connection = $this->query->modelClass->getConnection();
+                    if ($connection instanceof Connection) {
+                        return strtolower($connection->driver ?? 'mysql');
+                    }
+                }
+            }
+            if (method_exists($this->query, 'getConnection')) {
+                $connection = $this->query->getConnection();
+                if ($connection instanceof Connection) {
+                    return strtolower($connection->driver ?? 'mysql');
+                }
+            }
+        } catch (\Throwable $e) {
+			$this->getLogger()->json_log($e);
+		}
+        return 'mysql';
+    }
+
+
+	/**
+	 * 获取 FROM 子句（表名 + 别名）
+	 * QueryTrait 对象有 $from 和 $alias 属性
+	 * @return string
+	 */
+	private function getFromAlias(): string
+	{
+		$from = $this->query->from;
+		if ($this->query->alias !== '' && $this->query->alias !== '0') {
+			$from .= ' AS ' . $this->query->alias;
+		}
+		return $from;
+	}
+
+
+}