eee

xxx

Signed-off-by: 向林 <as2252258@163.com>

.phpstorm.meta.php

@@ -1,18 +0,0 @@
-<?php
-
-namespace PHPSTORM_META {
-
-    // Reflect
-	use Kiri\Di\Container;
-	use Psr\Container\ContainerInterface;
-
-	override(ContainerInterface::get(0), map('@'));
-	override(Container::make(0), map('@'));
-	override(Container::get(0), map('@'));
-	override(Container::create(0), map('@'));
-//    override(\Hyperf\Utils\Context::get(0), map('@'));
-//    override(\make(0), map('@'));
-    override(\di(0), map('@'));
-    override(\duplicate(0), map('@'));
-
-}

ActiveQuery.php

@@ -22,250 +22,272 @@ class ActiveQuery extends QueryTrait implements ISqlBuilder
 {
 
 
-    public bool     $asArray = FALSE;
-    protected mixed $_mock   = null;
+	public bool     $asArray = FALSE;
+	protected mixed $_mock   = NULL;
 
 
-    /**
-     * @param bool $asArray
-     * @return static
-     */
-    public function asArray(bool $asArray = true): static
-    {
-        $this->asArray = $asArray;
-        return $this;
-    }
+	/**
+	 * @param bool $asArray
+	 *
+	 * @return static
+	 */
+	public function asArray(bool $asArray = TRUE): static
+	{
+		$this->asArray = $asArray;
+		return $this;
+	}
 
-    /**
-     * @param array $methods
-     * @return $this
-     */
-    public function with(array $methods): static
-    {
-        $this->modelClass->setWith($methods);
-        return $this;
-    }
+	/**
+	 * @param array $methods
+	 *
+	 * @return $this
+	 */
+	public function with(array $methods): static
+	{
+		$this->modelClass->setWith($methods);
+		return $this;
+	}
 
 
-    /**
-     * @return ModelInterface|array|null|bool
-     * @throws
-     */
-    public function first(): ModelInterface|null|array|bool
-    {
-        $data = $this->buildCommand($this->builder->one())->one();
-        if (is_array($data)) {
-            return $this->populate($data);
-        }
-        return null;
-    }
+	/**
+	 * @return ModelInterface|array|null|bool
+	 * @throws
+	 */
+	public function first(): ModelInterface|null|array|bool
+	{
+		$data = $this->buildCommand($this->builder->one())->one();
+		if (is_array($data)) {
+			return $this->populate($data);
+		}
+		return NULL;
+	}
 
-    /**
-     * @return bool|Collection
-     */
-    public function get(): bool|Collection
-    {
-        $data = $this->buildCommand($this->builder->all())->all();
-        if (is_array($data)) {
-            return new Collection($this, $this->modelClass, $data);
-        }
-        return false;
-    }
+	/**
+	 * @return bool|Collection
+	 */
+	public function get(): bool|Collection
+	{
+		$data = $this->buildCommand($this->builder->all())->all();
+		if (is_array($data)) {
+			return new Collection($this, $this->modelClass, $data);
+		}
+		return FALSE;
+	}
 
 
-    /**
-     * @param string $column
-     * @param int|float $amount
-     * @return bool
-     */
-    public function increment(string $column, int|float $amount = 1): bool
-    {
-        return (bool)$this->buildCommand($this->builder->mathematics([$column => $amount]))->exec();
-    }
+	/**
+	 * @param string    $column
+	 * @param int|float $amount
+	 *
+	 * @return bool
+	 */
+	public function increment(string $column, int|float $amount = 1): bool
+	{
+		return (bool)$this->buildCommand($this->builder->mathematics([$column => $amount]))->exec();
+	}
 
 
-    /**
-     * @param array $attributes
-     * @return bool
-     */
-    public function increments(array $attributes): bool
-    {
-        return (bool)$this->buildCommand($this->builder->mathematics($attributes))->exec();
-    }
+	/**
+	 * @param array $attributes
+	 *
+	 * @return bool
+	 */
+	public function increments(array $attributes): bool
+	{
+		return (bool)$this->buildCommand($this->builder->mathematics($attributes))->exec();
+	}
 
 
-    /**
-     * @param string $column
-     * @param int|float $amount
-     * @return bool
-     */
-    public function decrement(string $column, int|float $amount = 1): bool
-    {
-        return (bool)$this->buildCommand($this->builder->mathematics([$column => $amount], '-'))->exec();
-    }
+	/**
+	 * @param string    $column
+	 * @param int|float $amount
+	 *
+	 * @return bool
+	 */
+	public function decrement(string $column, int|float $amount = 1): bool
+	{
+		return (bool)$this->buildCommand($this->builder->mathematics([$column => $amount], '-'))->exec();
+	}
 
 
-    /**
-     * @param array $attributes
-     * @return bool
-     */
-    public function decrements(array $attributes): bool
-    {
-        return (bool)$this->buildCommand($this->builder->mathematics($attributes, '-'))->exec();
-    }
+	/**
+	 * @param array $attributes
+	 *
+	 * @return bool
+	 */
+	public function decrements(array $attributes): bool
+	{
+		return (bool)$this->buildCommand($this->builder->mathematics($attributes, '-'))->exec();
+	}
 
 
-    /**
-     * @throws
-     */
-    public function flush(): bool
-    {
-        return (bool)$this->buildCommand($this->builder->truncate())->exec();
-    }
+	/**
+	 * @throws
+	 */
+	public function flush(): bool
+	{
+		return (bool)$this->buildCommand($this->builder->truncate())->exec();
+	}
 
 
-    /**
-     * @param int $size
-     * @param Closure $closure
-     * @return void
-     */
-    public function chunk(int $size, Closure $closure): void
-    {
-        if ($this->offset === -1) $this->offset = 0;
-        $data = $this->offset($this->offset)->limit($size)->get();
-        if (!$data || $data->isEmpty()) {
-            return;
-        }
-        if (Context::inCoroutine()) {
-            Coroutine::create(fn () => $closure($data));
-        } else {
-            call_user_func($closure, $data);
-        }
-        $this->offset += $size;
-        $this->chunk($size, $closure);
-    }
+	/**
+	 * @param int     $size
+	 * @param Closure $closure
+	 *
+	 * @return void
+	 */
+	public function chunk(int $size, Closure $closure): void
+	{
+		if ($this->offset === -1) $this->offset = 0;
+		$data = $this->offset($this->offset)->limit($size)->get();
+		if (!$data || $data->isEmpty()) {
+			return;
+		}
+		if (Context::inCoroutine()) {
+			Coroutine::create(fn() => $closure($data));
+		} else {
+			call_user_func($closure, $data);
+		}
+		$this->offset += $size;
+		$this->chunk($size, $closure);
+	}
 
-    /**
-     * @param string $field
-     * @param string|null $setKey
-     *
-     * @return array|null
-     */
-    public function column(string $field, ?string $setKey = null): ?array
-    {
-        return $this->get()->column($field, $setKey);
-    }
+	/**
+	 * @param string      $field
+	 * @param string|null $setKey
+	 *
+	 * @return array|null
+	 */
+	public function column(string $field, ?string $setKey = NULL): ?array
+	{
+		return $this->get()->column($field, $setKey);
+	}
 
 
-    /**
-     * @param mixed $value
-     * @return $this
-     */
-    public function withMock(mixed $value): static
-    {
-        $this->_mock = $value;
-        return $this;
-    }
+	/**
+	 * @param mixed $value
+	 *
+	 * @return $this
+	 */
+	public function withMock(mixed $value): static
+	{
+		$this->_mock = $value;
+		return $this;
+	}
 
 
-    /**
-     * @return mixed
-     */
-    public function mock(): mixed
-    {
-        return $this->_mock;
-    }
+	/**
+	 * @return mixed
+	 */
+	public function mock(): mixed
+	{
+		return $this->_mock;
+	}
 
 
-    /**
-     * @param array $data
-     * @return ModelInterface|array
-     * @throws
-     */
-    public function populate(array $data): ModelInterface|array
-    {
-        $model = $this->modelClass->populates($data);
+	/**
+	 * @param array $data
+	 *
+	 * @return ModelInterface|array
+	 * @throws
+	 */
+	public function populate(array $data): ModelInterface|array
+	{
+		$model = $this->modelClass->populates($data);
 
-        return $this->asArray ? $model->toArray() : $model;
-    }
+		return $this->asArray ? $model->toArray() : $model;
+	}
 
 
-    /**
-     * @return int
-     * @throws
-     */
-    public function count(): int
-    {
-        return $this->buildCommand($this->builder->count())->rowCount();
-    }
+	/**
+	 * @return string
+	 */
+	public function toSql(): string
+	{
+		return $this->builder->get();
+	}
 
 
-    /**
-     * @param array $data
-     * @return bool
-     * @throws
-     */
-    public function update(array $data): bool
-    {
-        if (count($data) < 1) {
-            return true;
-        }
-        $generate = $this->builder->update($data);
-        if (!is_bool($generate)) {
-            return (bool)$this->buildCommand($generate)->exec();
-        } else {
-            return $generate;
-        }
-    }
-
-    /**
-     * @param array $data
-     * @return bool
-     */
-    public function insert(array $data): bool
-    {
-        $sql = $this->builder->insert($data, isset($data[0]));
-
-        return (bool)$this->buildCommand($sql)->exec();
-    }
-
-    /**
-     * @param string $filed
-     *
-     * @return mixed
-     * @throws
-     */
-    public function value(string $filed): mixed
-    {
-        return $this->first()[$filed] ?? NULL;
-    }
-
-    /**
-     * @return bool
-     * @throws
-     */
-    public function exists(): bool
-    {
-        return $this->buildCommand($this->limit(1)->builder->exists())->exists();
-    }
+	/**
+	 * @return int
+	 * @throws
+	 */
+	public function count(): int
+	{
+		return $this->buildCommand($this->builder->count())->rowCount();
+	}
 
 
-    /**
-     * @param string $sql
-     * @param array $params
-     * @return int|bool
-     */
-    public function execute(string $sql, array $params = []): int|bool
-    {
-        return $this->buildCommand($sql, $params)->exec();
-    }
+	/**
+	 * @param array $data
+	 *
+	 * @return bool
+	 * @throws
+	 */
+	public function update(array $data): bool
+	{
+		if (count($data) < 1) {
+			return TRUE;
+		}
+		$generate = $this->builder->update($data);
+		if (!is_bool($generate)) {
+			return (bool)$this->buildCommand($generate)->exec();
+		} else {
+			return $generate;
+		}
+	}
+
+	/**
+	 * @param array $data
+	 *
+	 * @return bool
+	 */
+	public function insert(array $data): bool
+	{
+		$sql = $this->builder->insert($data, isset($data[0]));
+
+		return (bool)$this->buildCommand($sql)->exec();
+	}
 
 
-    /**
-     * @return bool
-     */
-    public function delete(): bool
-    {
-        return $this->buildCommand($this->builder->delete())->delete();
-    }
+	/**
+	 * @param string $filed
+	 *
+	 * @return mixed
+	 * @throws
+	 */
+	public function value(string $filed): mixed
+	{
+		return $this->first()[$filed] ?? NULL;
+	}
+
+	/**
+	 * @return bool
+	 * @throws
+	 */
+	public function exists(): bool
+	{
+		return $this->buildCommand($this->limit(1)->builder->exists())->exists();
+	}
+
+
+	/**
+	 * @param string $sql
+	 * @param array  $params
+	 *
+	 * @return int|bool
+	 */
+	public function execute(string $sql, array $params = []): int|bool
+	{
+		return $this->buildCommand($sql, $params)->exec();
+	}
+
+
+	/**
+	 * @return bool
+	 */
+	public function delete(): bool
+	{
+		return (bool)$this->buildCommand($this->builder->delete())->exec();
+	}
 }

BackupCommand.php

@@ -31,7 +31,7 @@ class BackupCommand extends Command
     /**
      *
      */
-    protected function configure()
+    protected function configure(): void
     {
         $this->setName('db:backup')
              ->addOption('data', 'd', InputArgument::OPTIONAL)

Base/ActiveQueryInterface.php

@@ -250,7 +250,7 @@ interface ActiveQueryInterface
      *
      * @return QueryTrait
      */
-    public function groupBy(string $name, string $having = NULL): QueryTrait;
+    public function groupBy(string $name, ?string $having = NULL): QueryTrait;
 
     /**
      * @param int $limit
@@ -267,4 +267,4 @@ interface ActiveQueryInterface
     public function offset(int $offset): QueryTrait;
 
 
-}
+}

Base/PDO.php

@@ -5,6 +5,7 @@ namespace Database\Base;
 class PDO extends \PDO
 {
 
+    readonly public string $driver;
 
     /**
      * @param string $database
@@ -12,6 +13,7 @@ class PDO extends \PDO
      * @param string $username
      * @param string $password
      * @param array $options
+     * @param string $driver
      */
     public function __construct(
         readonly public string $database,
@@ -19,10 +21,23 @@ class PDO extends \PDO
         readonly public string $username,
         readonly public string $password,
         readonly public array  $options,
+        string $driver = 'mysql',
     )
     {
-        parent::__construct('mysql:dbname=' . $this->database . ';host=' . $this->host, $this->username, $this->password, $this->options);
+        $this->driver = strtolower($driver);
+        $dsn = $this->buildDsn();
+        parent::__construct($dsn, $this->username, $this->password, $this->options);
     }
 
+    /**
+     * @return string
+     */
+    private function buildDsn(): string
+    {
+        return match ($this->driver) {
+            'pgsql', 'postgresql' => 'pgsql:host=' . $this->host . ';dbname=' . $this->database,
+            default => 'mysql:dbname=' . $this->database . ';host=' . $this->host,
+        };
+    }
 
 }

Collection.php

@@ -41,7 +41,7 @@ class Collection extends AbstractCollection
     public function update(array $attributes): bool
     {
         if ($this->isEmpty()) {
-            return $this->getLogger()->failure('No data by update', 'mysql');
+            return $this->getLogger()->logCategory('No data by update', 'mysql');
         }
         return $this->batch()->update($attributes);
     }
@@ -150,7 +150,7 @@ class Collection extends AbstractCollection
     {
         $model = $this->getModel();
         if ($this->isEmpty()) {
-            return $this->getLogger()->failure('No data by delete', 'mysql');
+            return $this->getLogger()->logCategory('No data by delete', 'mysql');
         }
         if (!$model->hasPrimary()) {
             throw new Exception('Must set primary key. if you want to delete data');

Command.php

@@ -1,230 +1,286 @@
-<?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) {
-            if ($this->isRefresh($throwable)) return $this->search($method);
-
-            $errorMsg = $throwable->getMessage() . PHP_EOL . '                 Sql: ' . $this->sql . '.' . json_encode($this->params);
-
-            return $this->getLogger()->failure($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);
-
-            return $result == 0 ? $prepare->rowCount() : (int)$result;
-        } catch (Throwable $throwable) {
-            if ($this->isRefresh($throwable)) return $this->_prepare();
-
-            $errorMsg = $throwable->getMessage() . PHP_EOL . '                 Sql: ' . $this->sql . '.' . json_encode($this->params);
-
-            return $this->getLogger()->failure($errorMsg . PHP_EOL, 'mysql');
-        } finally {
-            $this->connection->release($client);
-        }
-    }
-
-
-    /**
-     * @param Throwable $throwable
-     * @return bool
-     */
-    protected function isRefresh(Throwable $throwable): bool
-    {
-        if (str_contains($throwable->getMessage(), 'MySQL server has gone away')) {
-            return true;
-        }
-        if (str_contains($throwable->getMessage(), 'Send of 14 bytes failed with errno=32 Broken pipe')) {
-            return true;
-        }
-        if (str_contains($throwable->getMessage(), 'Lost connection to MySQL server during query')) {
-            return true;
-        }
-        return false;
-    }
-
-
-    /**
-     * @return bool
-     * @throws
-     */
-    public function delete(): bool
-    {
-        return (bool)$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');
+			}
+		}
+
+		return $this->getLogger()->logCategory('Max retry exceeded for SQL: ' . $this->sql, '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');
+			}
+		}
+
+		return $this->getLogger()->logCategory('Max retry exceeded for SQL: ' . $this->sql, '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;
+	}
+
+}

Condition/Condition.php

@@ -43,10 +43,9 @@ abstract class Condition extends Component
     public function setValue($params): void
     {
         if (is_array($params)) {
-            $values = [];
-            foreach ($params as $item => $value) {
-                $values[$item] = is_numeric($value) ? $value : '\'' . $value . '\'';
-            }
+            $values = array_map(function ($value) {
+                return is_numeric($value) ? $value : '\'' . $value . '\'';
+            }, $params);
             $this->value = $values;
         } else {
             $this->value = $this->checkIsSqlString($params);

Condition/NotInCondition.php

@@ -17,7 +17,7 @@ class NotInCondition extends Condition
      * @return string|null
      * @throws
      */
-    #[Pure] public function builder(): ?string
+    public function builder(): ?string
     {
         if (!is_array($this->value)) {
             throw new \Exception('Builder data by a empty string. need array');

Connection.php

@@ -14,21 +14,21 @@ namespace Database;
 use Database\Affair\BeginTransaction;
 use Database\Affair\Commit;
 use Database\Affair\Rollback;
+use Database\Base\Driver;
+use Database\Base\PDO;
 use Exception;
-use Kiri;
-use Kiri\Server\Events\OnWorkerExit;
 use Kiri\Abstracts\Component;
 use Kiri\Di\Context;
-use Kiri\Pool\Pool;
-use Kiri\Events\EventProvider;
-use Kiri\Error\StdoutLogger;
-use Psr\Log\LoggerInterface;
-use Kiri\Server\Events\OnWorkerStart;
-use Kiri\Server\Events\OnTaskerStart;
-use Kiri\Server\Events\OnAfterRequest;
 use Kiri\Di\Inject\Container;
+use Kiri\Error\StdoutLogger;
+use Kiri\Events\EventProvider;
+use Kiri\Pool\Pool;
+use Kiri\Server\Events\OnAfterRequest;
+use Kiri\Server\Events\OnTaskerStart;
+use Kiri\Server\Events\OnWorkerExit;
+use Kiri\Server\Events\OnWorkerStart;
+use Psr\Log\LoggerInterface;
 use Swoole\Timer;
-use Database\Base\PDO;
 
 //use PDO;
 
@@ -46,9 +46,10 @@ class Connection extends Component
     public string       $charset     = 'utf-8';
     public string       $tablePrefix = '';
     public string       $database    = '';
+    public string       $driver      = 'mysql';
     public int          $timeout     = 30;
     public int          $waite_time  = 3;
-    public int          $tick_time   = 60;
+    public int          $tick_time   = 60000;
     public int          $idle_count  = 3;
     public int          $idle_time   = 60;
     public array        $pool        = ['max' => 10, 'min' => 1];
@@ -67,6 +68,14 @@ class Connection extends Component
     #[Container(LoggerInterface::class)]
     public StdoutLogger $logger;
 
+
+    /**
+     * @var EventProvider
+     */
+    #[Container(EventProvider::class)]
+    public EventProvider $eventProvider;
+
+
     /**
      * @param Pool $connections
      */
@@ -99,14 +108,13 @@ class Connection extends Component
      */
     public function init(): void
     {
-        $eventProvider = Kiri::getDi()->get(EventProvider::class);
-        $eventProvider->on(BeginTransaction::class, [$this, 'beginTransaction'], 0);
-        $eventProvider->on(Rollback::class, [$this, 'rollback'], 0);
-        $eventProvider->on(Commit::class, [$this, 'commit'], 0);
-        $eventProvider->on(OnAfterRequest::class, [$this, 'clear']);
-        $eventProvider->on(OnWorkerExit::class, [$this, 'disconnect']);
-        $eventProvider->on(OnWorkerStart::class, [$this, 'tick']);
-        $eventProvider->on(OnTaskerStart::class, [$this, 'tick']);
+        $this->eventProvider->on(BeginTransaction::class, [$this, 'beginTransaction'], 0);
+        $this->eventProvider->on(Rollback::class, [$this, 'rollback'], 0);
+        $this->eventProvider->on(Commit::class, [$this, 'commit'], 0);
+        $this->eventProvider->on(OnAfterRequest::class, [$this, 'clear']);
+        $this->eventProvider->on(OnWorkerExit::class, [$this, 'disconnect']);
+        $this->eventProvider->on(OnWorkerStart::class, [$this, 'tick']);
+        $this->eventProvider->on(OnTaskerStart::class, [$this, 'tick']);
     }
 
 
@@ -127,28 +135,28 @@ class Connection extends Component
     protected function checkClientHealth(Pool $pool): void
     {
         $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->getName(), $client);
-            } catch (\Throwable $exception) {
-                if (!str_contains($exception->getMessage(), 'Client timeout.')) {
-                    $this->logger->error(throwable($exception), [$this->cds]);
-                }
-            }
-        }
+//        $length = $pool->size($this->getName());
+//        for ($i = 0; $i < $length; $i++) {
+//            try {
+//                if (($client = $this->validator($pool)) === false) {
+//                    break;
+//                }
+//                $pool->push($this->getName(), $client);
+//            } catch (\Throwable $exception) {
+//                if (!str_contains($exception->getMessage(), 'Client timeout.')) {
+//                    $this->logger->error(throwable($exception), [$this->cds]);
+//                }
+//            }
+//        }
     }
 
 
     /**
      * @return string
      */
-    private function getName(): string
+    public function getName(): string
     {
-        return 'mysql.' . $this->cds;
+        return strtolower($this->driver) . '.' . $this->cds;
     }
 
 
@@ -256,6 +264,7 @@ class Connection extends Component
                 $pdo->rollback();
             }
             $this->release($pdo);
+            Context::remove($this->cds);
         }
     }
 
@@ -275,6 +284,7 @@ class Connection extends Component
                 $pdo->commit();
             }
             $this->release($pdo);
+            Context::remove($this->cds);
         }
     }
 
@@ -294,6 +304,8 @@ class Connection extends Component
             $pdo->rollback();
         }
         $this->release($pdo);
+        Context::remove($this->cds);
+        $this->storey = 0;
     }
 
 
@@ -305,7 +317,7 @@ class Connection extends Component
      */
     public function createCommand(string $sql, array $attributes = []): Command
     {
-        return (new Command(['connection' => $this, 'sql' => $sql]))->bindValues($attributes);
+        return new Command(['connection' => $this, 'sql' => $sql])->bindValues($attributes);
     }
 
 
@@ -349,16 +361,22 @@ class Connection extends Component
      */
     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,
-\PDO::ATTR_STRINGIFY_FETCHES  => false,
-\PDO::ATTR_EMULATE_PREPARES   => true,
-\PDO::ATTR_TIMEOUT            => $this->timeout,
-\PDO::MYSQL_ATTR_INIT_COMMAND => 'SET NAMES ' . $this->charset
-        ]);
+        $driver = strtolower($this->driver);
+        $options = [
+            \PDO::ATTR_CASE              => \PDO::CASE_NATURAL,
+            \PDO::ATTR_ERRMODE           => \PDO::ERRMODE_EXCEPTION,
+            \PDO::ATTR_ORACLE_NULLS      => \PDO::NULL_NATURAL,
+            \PDO::ATTR_STRINGIFY_FETCHES => false,
+            \PDO::ATTR_EMULATE_PREPARES  => true,
+            \PDO::ATTR_TIMEOUT           => $this->timeout,
+        ];
+        
+        // MySQL 特定的选项
+        if ($driver === 'mysql') {
+            $options[\PDO::MYSQL_ATTR_INIT_COMMAND] = 'SET NAMES ' . $this->charset;
+        }
+        
+        $pdo = new PDO($this->database, $this->cds, $this->username, $this->password, $options, $driver);
         foreach ($this->attributes as $key => $attribute) {
             $pdo->setAttribute($key, $attribute);
         }

DatabasesProviders.php

@@ -35,7 +35,7 @@ class DatabasesProviders extends Providers
             return;
         }
         foreach ($databases as $key => $database) {
-            $this->set($key, $this->_settings($database));
+            $this->set($key, $this->_settings($key, $database));
         }
     }
 
@@ -63,23 +63,25 @@ class DatabasesProviders extends Providers
     }
 
 
-    /**
-     * @param array $database
-     * @return array
-     */
-    private function _settings(array $database): array
+	/**
+	 * @param string $key
+	 * @param array  $database
+	 * @return array
+	 */
+    private function _settings(string $key, array $database): array
     {
         $clientPool = $database['pool'] ?? ['min' => 1, 'max' => 5, 'tick' => 60];
         return [
-            'id'          => $database['id'],
+            'id'          => $key,
             'cds'         => $database['cds'],
             'class'       => Connection::class,
             'username'    => $database['username'],
             'password'    => $database['password'],
             'tablePrefix' => $database['tablePrefix'],
             'database'    => $database['database'],
+            'driver'      => $database['driver'] ?? 'mysql',
             'timeout'     => $database['timeout'] ?? 10,
-            'tick_time'   => $database['tick_time'] ?? 60,
+            'tick_time'   => $database['tick_time'] ?? 60000,
             'waite_time'  => $database['waite_time'] ?? 3,
             'pool'        => $clientPool,
             'attributes'  => $database['attributes'] ?? [],

HasCount.php

@@ -4,8 +4,8 @@ declare(strict_types=1);
 namespace Database;
 
 use Database\Traits\HasBase;
-use Exception;
 use Kiri;
+use Kiri\Di\Context;
 
 /**
  * Class HasCount
@@ -20,8 +20,7 @@ class HasCount extends HasBase
      */
     public function get(): array|ModelInterface|null
     {
-        $relation = Kiri::getDi()->get(Relation::class);
-        return $relation->get($this->name);
+        return di(Relation::class)->get($this->name);
     }
 
 }

HasMany.php

@@ -12,6 +12,7 @@ namespace Database;
 use Database\Traits\HasBase;
 use Exception;
 use Kiri;
+use Kiri\Di\Context;
 
 /**
  * Class HasMany
@@ -28,7 +29,6 @@ class HasMany extends HasBase
      */
     public function get(): array|Collection|null
     {
-        $relation = Kiri::getDi()->get(Relation::class);
-        return $relation->get($this->name);
+        return di(Relation::class)->get($this->name);
     }
 }

HasOne.php

@@ -12,6 +12,7 @@ namespace Database;
 use Database\Traits\HasBase;
 use Exception;
 use Kiri;
+use Kiri\Di\Context;
 
 /**
  * Class HasOne
@@ -27,7 +28,6 @@ class HasOne extends HasBase
      */
     public function get(): array|ModelInterface|null
     {
-        $relation = Kiri::getDi()->get(Relation::class);
-        return $relation->first($this->name);
+        return di(Relation::class)->first($this->name);
     }
 }

Model.php

@@ -116,8 +116,9 @@ class Model extends Base\Model
             $connection->commit();
             return $select;
         } catch (\Throwable $throwable) {
-            $connection->rollback();
-            return \Kiri::getLogger()->failure($throwable);
+			$model->getLogger()->json_log($throwable);
+			$connection->rollback();
+            return false;
         }
     }
 
@@ -145,8 +146,9 @@ class Model extends Base\Model
             $connection->commit();
             return $select;
         } catch (\Throwable $throwable) {
-            $connection->rollback();
-            return \Kiri::getLogger()->failure($throwable);
+			$model->getLogger()->json_log($throwable);
+			$connection->rollback();
+			return false;
         }
     }
 
@@ -206,9 +208,10 @@ class Model extends Base\Model
     public static function inserts(array $data): bool
     {
         if (empty($data)) {
-            return trigger_print_error('Insert data empty.', 'mysql');
-        }
-        return static::query()->insert($data);
+            return \Kiri::getLogger()->logCategory('Insert data empty.', 'mysql');
+        } else {
+			return static::query()->insert($data);
+		}
     }
 
     /**
@@ -266,22 +269,23 @@ class Model extends Base\Model
 
 
     /**
-     * @param ModelInterface|string $modelName
+     * @param string $modelName
      * @param string $foreignKey
      * @param string $localKey
      * @return string
      * @throws Exception
      */
-    private function _hasBase(ModelInterface|string $modelName, string $foreignKey, string $localKey): string
+    private function _hasBase(string $modelName, string $foreignKey, string $localKey): string
     {
         if (($value = $this->{$localKey}) === null) {
             throw new Exception("Need join table primary key.");
         }
 
-        $relation = $this->getRelation();
+        $relation = di(Relation::class);
 
-        $primaryKey = $modelName . $foreignKey . $value;
+        $primaryKey = str_replace('\\', '_', $modelName) . '_' . $foreignKey . '_' . $value;
         if (!$relation->hasIdentification($primaryKey)) {
+	        /** @var $modelName ModelInterface */
             $relation->bindIdentification($primaryKey, $modelName::query()->where([$foreignKey => $value]));
         }
         return $primaryKey;
@@ -289,60 +293,61 @@ class Model extends Base\Model
 
 
     /**
-     * @param ModelInterface|string $modelName
+     * @param string $modelName
      * @param string $foreignKey
      * @param string $localKey
      * @return HasOne|ActiveQuery
      * @throws Exception
      */
-    public function hasOne(ModelInterface|string $modelName, string $foreignKey, string $localKey): HasOne|ActiveQuery
+    public function hasOne(string $modelName, string $foreignKey, string $localKey): HasOne|ActiveQuery
     {
         return new HasOne($this->_hasBase($modelName, $foreignKey, $localKey));
     }
 
 
     /**
-     * @param ModelInterface|string $modelName
+     * @param string $modelName
      * @param string $foreignKey
      * @param string $localKey
      * @return ActiveQuery|HasCount
      * @throws Exception
      */
-    public function hasCount(ModelInterface|string $modelName, string $foreignKey, string $localKey): ActiveQuery|HasCount
+    public function hasCount(string $modelName, string $foreignKey, string $localKey): ActiveQuery|HasCount
     {
         return new HasCount($this->_hasBase($modelName, $foreignKey, $localKey));
     }
 
 
     /**
-     * @param ModelInterface|string $modelName
+     * @param string $modelName
      * @param string $foreignKey
      * @param string $localKey
      * @return ActiveQuery|HasMany
      * @throws Exception
      */
-    public function hasMany(ModelInterface|string $modelName, string $foreignKey, string $localKey): ActiveQuery|HasMany
+    public function hasMany(string $modelName, string $foreignKey, string $localKey): ActiveQuery|HasMany
     {
         return new HasMany($this->_hasBase($modelName, $foreignKey, $localKey));
     }
 
     /**
-     * @param ModelInterface|string $modelName
+     * @param string $modelName
      * @param string $foreignKey
      * @param string $localKey
      * @return ActiveQuery|HasMany
      * @throws Exception
      */
-    public function hasIn(ModelInterface|string $modelName, string $foreignKey, string $localKey): ActiveQuery|HasMany
+    public function hasIn(string $modelName, string $foreignKey, string $localKey): ActiveQuery|HasMany
     {
         if (($value = $this->{$localKey}) === null) {
             throw new Exception("Need join table primary key.");
         }
 
-        $relation = $this->getRelation();
+        $relation = di(Relation::class);
 
-        $primaryKey = $modelName . $foreignKey . json_encode($value, JSON_UNESCAPED_UNICODE);
+        $primaryKey = str_replace('\\', '_', $modelName) . '_' . $foreignKey . '_' . implode('_', $value);
         if (!$relation->hasIdentification($primaryKey)) {
+			/** @var $modelName ModelInterface */
             $relation->bindIdentification($primaryKey, $modelName::query()->whereIn($foreignKey, $value));
         }
 

Relation.php

@@ -4,8 +4,10 @@ declare(strict_types=1);
 namespace Database;
 
 
+use Database\Base\ActiveQueryInterface;
 use Kiri\Abstracts\Component;
 use Kiri\Di\Context;
+use Swoole\Coroutine;
 
 /**
  * Class Relation
@@ -34,7 +36,7 @@ class Relation extends Component
      */
     public function hasIdentification(string $identification): bool
     {
-        return isset($this->_query[$identification]) && $this->_query[$identification] instanceof ActiveQuery;
+        return isset($this->_query[$identification]);
     }
 
     /**
@@ -54,12 +56,10 @@ class Relation extends Component
      */
     public function first(string $_identification): mixed
     {
-        if (Context::exists($_identification)) {
-            return Context::get($_identification);
+        if (!Context::exists($_identification)) {
+            Context::set($_identification, $this->_query[$_identification]->first());
         }
-        $activeModel = $this->_query[$_identification]->first();
-        unset($this->_query[$_identification]);
-        return Context::set($_identification, $activeModel);
+        return Context::get($_identification);
     }
 
 
@@ -69,12 +69,10 @@ class Relation extends Component
      */
     public function count(string $_identification): mixed
     {
-        if (Context::exists($_identification)) {
-            return Context::get($_identification);
+        if (!Context::exists($_identification)) {
+            Context::set($_identification, $this->_query[$_identification]->count());
         }
-        $activeModel = $this->_query[$_identification]->count();
-        unset($this->_query[$_identification]);
-        return Context::set($_identification, $activeModel);
+        return Context::get($_identification);
     }
 
 
@@ -86,11 +84,9 @@ class Relation extends Component
     public function get(string $_identification): mixed
     {
         if (Context::exists($_identification)) {
-            return Context::get($_identification);
+            Context::set($_identification, $this->_query[$_identification]->get());
         }
-        $activeModel = $this->_query[$_identification]->get();
-        unset($this->_query[$_identification]);
-        return Context::set($_identification, $activeModel);
+        return Context::get($_identification);
     }
 
 }

SqlBuilder.php

@@ -1,411 +1,471 @@
-<?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->getWhere());
-    }
-
-
-    /**
-     * @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->getParams();
-        $this->query->setParams([]);
-        $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()->failure('None data update.');
-        }
-        return 'UPDATE ' . $this->query->getFrom() . ' 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->getFrom();
-        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->getFrom() . $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 (is_string($value) && $this->isMath($value)) {
-            $keys[] = $key . '=' . $key . ' ' . $value;
-        } else {
-            $this->query->pushParam($value);
-            $keys[] = $key . '= ?';
-        }
-        return $keys;
-    }
-
-
-    /**
-     * @param string $value
-     * @return bool
-     */
-    private function isMath(string $value): bool
-    {
-        return str_starts_with($value, '+ ') || str_starts_with($value, '- ');
-    }
-
-
-    /**
-     * @return string
-     * @throws
-     */
-    public function one(): string
-    {
-        return $this->makeSelect($this->query->getSelect()) . $this->make() . $this->makeLimit($this->query->limit(1));
-    }
-
-
-    /**
-     * @return string
-     * @throws
-     */
-    public function all(): string
-    {
-        return $this->makeSelect($this->query->getSelect()) . $this->make() . $this->makeLimit($this->query);
-    }
-
-
-    /**
-     * @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
-     */
-    public function columns(string $table): string
-    {
-        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->getFrom();
-        if ($this->query->getAlias() != "") {
-            $select .= " AS " . $this->query->getAlias();
-        }
-        if (count($this->query->getJoin()) > 0) {
-            $select .= ' ' . implode(' ', $this->query->getJoin());
-        }
-        return $select;
-    }
-
-
-    /**
-     * @return string
-     */
-    private function makeGroup(): string
-    {
-        if ($this->query->getGroup() != "") {
-            return ' GROUP BY ' . $this->query->getGroup();
-        }
-        return '';
-    }
-
-
-    /**
-     * @return string
-     */
-    private function makeOrder(): string
-    {
-        if (count($this->query->getOrder()) > 0) {
-            return ' ORDER BY ' . implode(',', $this->query->getOrder());
-        }
-        return '';
-    }
-
-
-    /**
-     * @return string
-     */
-    private function makeCondition(): string
-    {
-        $condition = $this->where($this->query->getWhere());
-        if (empty($condition)) {
-            return '';
-        }
-        return ' WHERE ' . $condition;
-    }
-
-
-    private function makeLimit(): string
-    {
-        if ($this->query->getOffset() >= 0 && $this->query->getLimit() >= 1) {
-            return ' LIMIT ' . $this->query->getOffset() . ',' . $this->query->getLimit();
-        }
-        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->getFrom());
-    }
-
-
-    /**
-     * @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;
-    }
-
-
-}
+<?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
+	{
+		$params              = $this->query->params;
+		$this->query->params = [];
+		$array               = $this->makeParams($attributes);
+		foreach ($params as $name => $value) {
+			$this->query->pushParam($value);
+		}
+		return $this->__updateBuilder($array);
+	}
+
+
+	/**
+	 * @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->all();
+	}
+
+
+	/**
+	 * @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
+	{
+		return $this->makeCondition() . ' ' . $this->makeGroup() . ' ' . $this->makeOrder();
+	}
+
+	/**
+	 * @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;
+	}
+
+
+}

Traits/HasBase.php

@@ -12,7 +12,6 @@ namespace Database\Traits;
 use Database\ModelInterface;
 use Database\Collection;
 use Database\Relation;
-use Kiri;
 
 /**
  * Class HasBase
@@ -52,12 +51,16 @@ abstract class HasBase implements \Database\Traits\Relation
      * @param string $name
      * @param array $arguments
      * @return $this|mixed
+     * @throws \Exception
      */
     public function __call(string $name, array $arguments)
     {
         if ($name !== 'get') {
-            $relation = Kiri::getDi()->get(Relation::class);
-            $relation->getQuery($this->name)->$name(...$arguments);
+            $query    = di(Relation::class)->getQuery($this->name);
+            if (is_null($query)) {
+                throw new \Exception('Unknown relation key: ' . $this->name);
+            }
+            $query->$name(...$arguments);
             return $this;
         } else {
             return $this->get();

Traits/When.php

@@ -18,61 +18,80 @@ use JetBrains\PhpStorm\Pure;
 class When
 {
 
-    public ActiveQuery|ISqlBuilder $query;
+	public ActiveQuery|ISqlBuilder $query;
 
 
-    private array $_condition = [];
-
-    private string $else = '';
+	/**
+	 * @var array
+	 */
+	private array $_condition = [] {
+		&get {
+			return $this->_condition;
+		}
+	}
 
 
-    /**
-     * CaseWhen constructor.
-     * @param string $column
-     * @param ActiveQuery|ISqlBuilder $activeQuery
-     */
-    public function __construct(public string $column, public ActiveQuery|ISqlBuilder $activeQuery)
-    {
-        $this->_condition[] = 'CASE ' . $column;
-    }
+	/**
+	 * @var string
+	 */
+	private string $else = '' {
+		get {
+			return $this->else;
+		}
+		set(string $value) {
+			$this->else = $value;
+		}
+	}
 
 
-    /**
-     * @param string|int $condition
-     * @param string $then
-     * @return $this
-     * @throws
-     */
-    public function when(string|int $condition, string $then): static
-    {
-        $this->_condition[] = sprintf('WHEN %s THEN %s', $condition, $then);
-
-        return $this;
-    }
+	/**
+	 * CaseWhen constructor.
+	 * @param string                  $column
+	 * @param ActiveQuery|ISqlBuilder $activeQuery
+	 */
+	public function __construct(public string $column, public ActiveQuery|ISqlBuilder $activeQuery)
+	{
+		$this->_condition[] = 'CASE ' . $column;
+	}
 
 
-    /**
-     * @param string $alias
-     */
-    public function else(string $alias): void
-    {
-        $this->else = $alias;
-    }
+	/**
+	 * @param string|int $condition
+	 * @param string     $then
+	 * @return $this
+	 * @throws
+	 */
+	public function when(string|int $condition, string $then): static
+	{
+		$this->_condition[] = sprintf('WHEN %s THEN %s', $condition, $then);
+
+		return $this;
+	}
 
 
-    /**
-     * @return string
-     */
-    #[Pure] public function end(): string
-    {
-        if (empty($this->_condition)) {
-            return '';
-        }
-        $prefix = implode(' ', $this->_condition);
-        if (!empty($this->else)) {
-            $prefix .= ' ELSE ' . $this->else;
-        }
-        return '(' . $prefix . ' END)';
-    }
+	/**
+	 * @param string $alias
+	 */
+	public function else(string $alias): void
+	{
+		$this->else = $alias;
+	}
+
+
+	/**
+	 * @return string
+	 */
+	#[Pure]
+	public function end(): string
+	{
+		if (empty($this->_condition)) {
+			return '';
+		}
+		$prefix = implode(' ', $this->_condition);
+		if (!empty($this->else)) {
+			$prefix .= ' ELSE ' . $this->else;
+		}
+		return '(' . $prefix . ' END)';
+	}
 
 }

composer.json

@@ -9,10 +9,10 @@
   ],
   "license": "MIT",
   "require": {
-    "php": ">=8.3",
+    "php": ">=8.5",
     "ext-json": "*",
     "ext-pdo": "*",
-    "game-worker/kiri-pool": "~v1.0"
+    "game-worker/kiri-pool": "^v1.0"
   },
   "autoload": {
     "psr-4": {

test.php

@@ -0,0 +1,23 @@
+<?php
+
+
+use Database\Query;
+
+class Users extends \Database\Model
+{
+
+
+	public function hasD()
+	{
+		return $this->hasOne(static::class, 'id', 'id')->with([]);
+	}
+}
+
+Users::query()
+	->select(['*', (new Query(Users::class))
+		->where(['id' => 2])])
+	->from(function (Query $query) {
+		$query->from(Users::class)
+			->where(['id' => 1])
+			->groupBy('name DESC');
+	})->toSql();