Compare commits

..

36 Commits

Author SHA1 Message Date
as2252258 39905b77e3 eee 2026-07-03 14:35:45 +08:00
as2252258 fce82ec56c eee 2026-06-30 23:27:42 +08:00
as2252258 2970c3bc6b eee 2026-06-30 23:16:18 +08:00
as2252258 a35d0bb921 eee 2026-06-30 23:15:59 +08:00
as2252258 fce2b8bcdb eee 2026-06-30 23:15:06 +08:00
as2252258 ce92f310a5 eee 2026-06-30 23:14:22 +08:00
as2252258 28fa17ecfb eee 2026-06-30 23:12:17 +08:00
as2252258 9fc910f10f eee 2026-06-30 23:11:55 +08:00
as2252258 0a95e722d6 eee 2026-06-30 23:09:52 +08:00
as2252258 9f2611972f eee 2026-06-30 23:08:13 +08:00
as2252258 b7e59d1c96 eee 2026-06-30 23:06:55 +08:00
as2252258 9b0ee54608 eee 2026-06-30 23:03:51 +08:00
as2252258 763a4baa57 eee 2026-06-30 23:01:40 +08:00
as2252258 fe639bbd1f eee 2026-06-30 23:01:11 +08:00
as2252258 e40de4ee4a eee 2026-06-30 23:00:25 +08:00
as2252258 ffc2af69da eee 2026-06-30 22:54:04 +08:00
as2252258 218a38da48 eee 2026-06-30 22:46:19 +08:00
as2252258 5acb2f4600 eee 2026-06-30 22:43:20 +08:00
as2252258 d6a07446bb eee 2026-06-30 22:35:02 +08:00
as2252258 6c936c3896 eee 2026-06-30 22:31:47 +08:00
as2252258 14448d824a eee 2026-06-30 22:07:04 +08:00
as2252258 06074c38c1 eee 2026-06-24 22:00:50 +08:00
as2252258 404b511c5f eee 2026-06-24 21:53:42 +08:00
as2252258 bf4f8538ed eee 2026-06-24 20:11:10 +08:00
as2252258 8a985a65df eee 2026-06-12 23:57:19 +08:00
as2252258 c7b980d969 eee 2026-04-23 00:23:32 +08:00
as2252258 fea649413c eee 2026-04-23 00:20:00 +08:00
as2252258 959407a95e eee 2026-04-23 00:19:10 +08:00
as2252258 0abc7c32f7 eee 2026-04-21 14:59:51 +08:00
as2252258 9c6340d0b3 eee 2026-01-11 19:59:20 +08:00
as2252258 0f01416351 eee 2026-01-11 19:57:24 +08:00
as2252258 b1d2c0fb51 eee 2026-01-11 19:56:26 +08:00
as2252258 db18a2d73a eee 2026-01-11 19:55:00 +08:00
as2252258 3cb3dcf384 eee 2026-01-11 19:53:35 +08:00
as2252258 fd8652ccd1 eee 2026-01-11 19:32:32 +08:00
as2252258 279151825a eee 2026-01-11 19:29:35 +08:00
10 changed files with 1807 additions and 1862 deletions
+1 -1
View File
@@ -288,6 +288,6 @@ class ActiveQuery extends QueryTrait implements ISqlBuilder
*/
public function delete(): bool
{
return $this->buildCommand($this->builder->delete())->delete();
return (bool)$this->buildCommand($this->builder->delete())->exec();
}
}
+843 -835
View File
File diff suppressed because it is too large Load Diff
+286 -256
View File
@@ -1,256 +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) {
$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);
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 (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;
}
}
+5 -1
View File
@@ -154,7 +154,7 @@ class Connection extends Component
/**
* @return string
*/
private function getName(): string
public function getName(): string
{
return strtolower($this->driver) . '.' . $this->cds;
}
@@ -264,6 +264,7 @@ class Connection extends Component
$pdo->rollback();
}
$this->release($pdo);
Context::remove($this->cds);
}
}
@@ -283,6 +284,7 @@ class Connection extends Component
$pdo->commit();
}
$this->release($pdo);
Context::remove($this->cds);
}
}
@@ -302,6 +304,8 @@ class Connection extends Component
$pdo->rollback();
}
$this->release($pdo);
Context::remove($this->cds);
$this->storey = 0;
}
+8 -7
View File
@@ -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,15 +63,16 @@ 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'],
+2
View File
@@ -285,6 +285,7 @@ class Model extends Base\Model
$primaryKey = str_replace('\\', '_', $modelName) . '_' . $foreignKey . '_' . $value;
if (!$relation->hasIdentification($primaryKey)) {
/** @var $modelName ModelInterface */
$relation->bindIdentification($primaryKey, $modelName::query()->where([$foreignKey => $value]));
}
return $primaryKey;
@@ -346,6 +347,7 @@ class Model extends Base\Model
$primaryKey = str_replace('\\', '_', $modelName) . '_' . $foreignKey . '_' . implode('_', $value);
if (!$relation->hasIdentification($primaryKey)) {
/** @var $modelName ModelInterface */
$relation->bindIdentification($primaryKey, $modelName::query()->whereIn($foreignKey, $value));
}
+471 -462
View File
@@ -1,462 +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()->logCategory('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 (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->getSelect()) . $this->make() . $this->makeLimit();
}
/**
* @return string
* @throws
*/
public function all(): string
{
return $this->makeSelect($this->query->getSelect()) . $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->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;
}
/**
* @return string
* @throws
*/
private function makeLimit(): string
{
if ($this->query->getOffset() >= 0 && $this->query->getLimit() >= 1) {
$driver = $this->getDriver();
if (in_array($driver, ['pgsql', 'postgresql'])) {
// PostgreSQL 使用 LIMIT ... OFFSET ... 语法
return ' LIMIT ' . $this->query->getLimit() . ' OFFSET ' . $this->query->getOffset();
}
// MySQL 使用 LIMIT offset,limit 语法
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;
}
/**
* 获取数据库驱动类型
* @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
{
$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;
}
}
+125 -253
View File
@@ -27,26 +27,105 @@ use Kiri\Abstracts\Component;
*/
abstract class QueryTrait extends Component implements ActiveQueryInterface, ISqlBuilder
{
protected array $where = [];
protected array $select = ['t1.*'];
protected array $join = [];
protected array $order = [];
protected int $offset = -1;
protected int $limit = -1;
protected string $group = '';
protected string $from = '';
protected string $alias = '';
protected array $filter = [];
public array $where = [] {
&get {
return $this->where;
}
}
public array $select = ['t1.*'] {
&get {
return $this->select;
}
}
public array $join = [] {
&get {
return $this->join;
}
}
public array $order = [] {
&get {
return $this->order;
}
}
public int $offset = -1 {
get {
return $this->offset;
}
set {
$this->offset = $value;
}
}
public int $limit = -1 {
get {
return $this->limit;
}
set {
$this->limit = $value;
}
}
public string $group = '' {
get {
return $this->group;
}
set {
$this->group = $value;
}
}
public string $from = '' {
get {
return $this->from;
}
set {
$this->from = $value;
}
}
public string $alias = '' {
get {
return $this->alias;
}
set {
$this->alias = $value;
}
}
protected array $filter = [] {
&get {
return $this->filter;
}
}
protected bool $lock = FALSE;
protected SqlBuilder $builder;
protected array $params = [];
protected array $_alias = [];
public array $params = [] {
&get {
return $this->params;
}
}
protected array $_alias = [] {
&get {
return $this->_alias;
}
}
/**
* @var ModelInterface|string|null
*/
protected ModelInterface|string|null $modelClass;
protected ModelInterface|string|null $modelClass {
get {
return $this->modelClass;
}
}
public function __get(string $name)
{
if (property_exists($this, $name)) {
return $this->$name;
}
if (method_exists($this, $name)) {
return $this->$name();
}
return parent::__get($name); // TODO: Change the autogenerated stub
}
/**
@@ -63,29 +142,9 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
}
/**
* @return mixed
*/
public function getModelClass(): mixed
{
return $this->modelClass;
}
/**
* @param array $where
*
* @return void
*/
public function setWhere(array $where): void
{
$this->where = $where;
}
/**
* @param Closure|Query $callback
* @param array $attributes
* @param array $attributes
*
* @return $this
*/
@@ -93,7 +152,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
{
if ($callback instanceof Query) {
$this->where[] = 'NOT EXISTS(' . $callback->build() . ')';
$this->mergeParams($callback->getParams());
$this->mergeParams($callback->params);
} else {
$this->where[] = 'NOT EXISTS(' . $this->makeClosureFunction($callback) . ')';
}
@@ -106,7 +165,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param Closure|Query $callback
* @param array $attributes
* @param array $attributes
*
* @return $this
*/
@@ -114,7 +173,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
{
if ($callback instanceof Query) {
$this->where[] = 'EXISTS(' . $callback->build() . ')';
$this->mergeParams($callback->getParams());
$this->mergeParams($callback->params);
} else {
$this->where[] = 'EXISTS(' . $this->makeClosureFunction($callback) . ')';
}
@@ -125,194 +184,6 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
}
/**
* @param array $select
*
* @return void
*/
public function setSelect(array $select): void
{
$this->select = $select;
}
/**
* @param array $join
*
* @return void
*/
public function setJoin(array $join): void
{
$this->join = $join;
}
/**
* @param array $order
*
* @return void
*/
public function setOrder(array $order): void
{
$this->order = $order;
}
/**
* @param int $offset
*
* @return void
*/
public function setOffset(int $offset): void
{
$this->offset = $offset;
}
/**
* @param int $limit
*
* @return void
*/
public function setLimit(int $limit): void
{
$this->limit = $limit;
}
/**
* @param string $group
*
* @return void
*/
public function setGroup(string $group): void
{
$this->group = $group;
}
/**
* @param string $from
*
* @return void
*/
public function setFrom(string $from): void
{
$this->from = $from;
}
/**
* @param string $alias
*
* @return void
*/
public function setAlias(string $alias): void
{
$this->alias = $alias;
}
/**
* @param array $params
*
* @return void
*/
public function setParams(array $params): void
{
$this->params = $params;
}
/**
* @return array
*/
public function getWhere(): array
{
return $this->where;
}
/**
* @return array|string[]
*/
public function getSelect(): array
{
return $this->select;
}
/**
* @return array
*/
public function getJoin(): array
{
return $this->join;
}
/**
* @return array
*/
public function getOrder(): array
{
return $this->order;
}
/**
* @return int
*/
public function getOffset(): int
{
return $this->offset;
}
/**
* @return int
*/
public function getLimit(): int
{
return $this->limit;
}
/**
* @return string
*/
public function getGroup(): string
{
return $this->group;
}
/**
* @return string
*/
public function getFrom(): string
{
return $this->from;
}
/**
* @return string
*/
public function getAlias(): string
{
return $this->alias;
}
/**
* @return array
*/
public function getParams(): array
{
return $this->params;
}
/**
* @param array $params
* @return void
@@ -326,7 +197,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param string $column
* @param string $column
* @param callable $callable
*
* @return $this
@@ -465,7 +336,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
{
if ($tableName instanceof Query) {
$this->from = '(' . $tableName->build() . ')';
$this->mergeParams($tableName->getParams());
$this->mergeParams($tableName->params);
} else if ($tableName instanceof Closure) {
$this->from = '(' . $this->makeClosureFunction($tableName) . ')';
} else if (class_exists($tableName)) {
@@ -480,8 +351,8 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param string $tableName
* @param string $alias
* @param array $on
* @param array $param
* @param array $on
* @param array $param
*
* @return $this
* $query->join([$tableName, ['userId'=>'uuvOd']], $param)
@@ -563,8 +434,8 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param string $tableName
* @param string $alias
* @param array $onCondition
* @param array $param
* @param array $onCondition
* @param array $param
*
* @return $this
* @throws Exception
@@ -585,8 +456,8 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param string $tableName
* @param string $alias
* @param array $onCondition
* @param array $param
* @param array $onCondition
* @param array $param
*
* @return $this
* @throws Exception
@@ -607,8 +478,8 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param string $tableName
* @param string $alias
* @param array $onCondition
* @param array $param
* @param array $onCondition
* @param array $param
*
* @return $this
* @throws Exception
@@ -651,8 +522,8 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param string $lngField
* @param string $latField
* @param string $lngField
* @param string $latField
* @param int|float $lng1
* @param int|float $lat1
*
@@ -668,7 +539,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param string|array $column
* @param string $sort
* @param string $sort
*
* @return $this
*
@@ -736,8 +607,8 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param array|Closure|string $conditionArray
* @param string $opera
* @param null $value
* @param string $opera
* @param null $value
*
* @return QueryTrait
* @throws
@@ -816,7 +687,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param string $columns
* @param string $columns
* @param array|Closure|Query $value
*
* @return $this
@@ -825,12 +696,13 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
public function whereIn(string $columns, array|Closure|Query $value): static
{
if (is_array($value)) {
if (count($value) < 1) throw new Exception('Value must be an not empty array');
if (count($value) < 1)
throw new Exception('Value must be an not empty array');
$this->where[] = $columns . ' IN (' . implode(',', $value) . ')';
} else if ($value instanceof Query) {
$this->where[] = $columns . ' IN (' . $value->build() . ')';
$this->mergeParams($value->getParams());
$this->mergeParams($value->params);
} else {
$this->where[] = $columns . ' IN (' . $this->makeClosureFunction($value) . ')';
}
@@ -844,13 +716,13 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
*/
public function queryInstance(): Query
{
return new Query();
return new Query;
}
/**
* @param string $columns
* @param array $value
* @param array $value
*
* @return $this
*/
@@ -865,7 +737,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param string $column
* @param string $column
* @param int|float $start
* @param int|float $end
*
@@ -886,7 +758,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param string $column
* @param string $column
* @param int|float $start
* @param int|float $end
*
@@ -920,7 +792,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param array|string $column
* @param mixed $value
* @param mixed $value
*
* @return $this
*/
@@ -940,7 +812,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param string $column
* @param string $opera
* @param mixed $value
* @param mixed $value
*
* @return $this
*/
@@ -978,13 +850,13 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
} else {
$generate->addArray($closure);
}
$this->mergeParams($generate->getParams());
$this->mergeParams($generate->params);
return $generate->build();
}
/**
* @param string $name
* @param string $name
* @param string|null $having
*
* @return $this
@@ -1060,7 +932,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param string $column
* @param mixed $value
* @param mixed $value
* @param string $opera
*
* @return string
@@ -1084,7 +956,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/**
* @param string $querySql
* @param array $params
* @param array $params
*
* @return Command
*/
+65 -46
View File
@@ -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)';
}
}
+1 -1
View File
@@ -9,7 +9,7 @@
],
"license": "MIT",
"require": {
"php": ">=8.4",
"php": ">=8.5",
"ext-json": "*",
"ext-pdo": "*",
"game-worker/kiri-pool": "^v1.0"