Compare commits

...

15 Commits

Author SHA1 Message Date
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
9 changed files with 1842 additions and 1909 deletions
+1 -1
View File
@@ -288,6 +288,6 @@ class ActiveQuery extends QueryTrait implements ISqlBuilder
*/ */
public function delete(): bool 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
+288 -256
View File
@@ -1,256 +1,288 @@
<?php <?php
/** /**
* Created by PhpStorm. * Created by PhpStorm.
* User: whwyy * User: whwyy
* Date: 2018/3/30 0030 * Date: 2018/3/30 0030
* Time: 15:23 * Time: 15:23
*/ */
declare(strict_types=1); declare(strict_types=1);
namespace Database; namespace Database;
use Exception; use Exception;
use Kiri\Abstracts\Component; use Kiri\Abstracts\Component;
use Kiri\Di\Container; use Kiri\Di\Container;
use PDO; use PDO;
use Throwable; use Throwable;
/** /**
* Class Command * Class Command
* @package Database * @package Database
*/ */
class Command extends Component class Command extends Component
{ {
public Connection $connection; /** @var int 数据库操作最大重试次数 */
public ?string $sql = ''; private const MAX_RETRIES = 2;
public array $params = [];
public Connection $connection;
/** public ?string $sql = '';
* @param array $params public array $params = [];
* @throws
*/ /**
public function __construct(array $params = []) * @param array $params
{ * @throws
parent::__construct(); */
Container::configure($this, $params); public function __construct(array $params = [])
} {
parent::__construct();
Container::configure($this, $params);
/** }
* @return bool
* @throws
*/ /**
public function incrOrDecr(): bool * @return bool
{ * @throws
return (bool)$this->_prepare(); */
} public function incrOrDecr(): bool
{
return (bool)$this->_prepare();
/** }
* @return bool|array
* @throws
*/ /**
public function all(): bool|array * @return bool|array
{ * @throws
return $this->search('fetchAll'); */
} public function all(): bool|array
{
/** return $this->search('fetchAll');
* @return array|bool|null }
* @throws
*/ /**
public function one(): array|null|bool * @return array|bool|null
{ * @throws
return $this->search('fetch'); */
} public function one(): array|null|bool
{
/** return $this->search('fetch');
* @return bool }
* @throws Exception
*/ /**
public function exists(): bool * @return bool
{ * @throws Exception
$data = $this->search('fetch'); */
if (!$data) { public function exists(): bool
return false; {
} $data = $this->search('fetch');
return true; if (!$data) {
} return false;
}
/** return true;
* @return mixed }
* @throws
*/ /**
public function fetchColumn(): mixed * @return mixed
{ * @throws
return $this->search('fetchColumn'); */
} public function fetchColumn(): mixed
{
/** return $this->search('fetchColumn');
* @return mixed }
* @throws
*/ /**
public function rowCount(): int * @return mixed
{ * @throws
$data = $this->search('fetch'); */
public function rowCount(): int
return !$data ? 0 : +current($data); {
} $data = $this->search('fetch');
return !$data ? 0 : +current($data);
/** }
* @param string $method
* @return mixed
* @throws /**
*/ * 执行查询类 SQL 操作,内置断连重试机制
protected function search(string $method): mixed * 当数据库连接断开时自动重试,断开的连接不再归还池中防止污染连接池
{ * @param string $method PDO 结果获取方法名
$client = $this->connection->getConnection(); * @return mixed
try { * @throws
$startTime = microtime(true); */
if (($prepare = $client->prepare($this->sql)) === false) { protected function search(string $method): mixed
throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $client->errorInfo()[2]); {
} $retryCount = 0;
$prepare->execute($this->params); while ($retryCount <= self::MAX_RETRIES) {
$client = $this->connection->getConnection();
$result = $method == 'rowCount' ? $prepare->rowCount() : $prepare->{$method}(PDO::FETCH_ASSOC); try {
$prepare->closeCursor(); $startTime = microtime(true);
if (($prepare = $client->prepare($this->sql)) === false) {
$this->connection->println($startTime, microtime(true), $this->sql, $this->params); throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $client->errorInfo()[2]);
}
return $result;
} catch (Throwable $throwable) { $prepare->execute($this->params);
$this->getLogger()->json_log($throwable);
$result = $method == 'rowCount' ? $prepare->rowCount() : $prepare->{$method}(PDO::FETCH_ASSOC);
if ($this->isRefresh($throwable)) { $prepare->closeCursor();
return $this->search($method);
} $this->connection->println($startTime, microtime(true), $this->sql, $this->params);
$errorMsg = $throwable->getMessage() . PHP_EOL . ' Sql: ' . $this->sql . '.' . json_encode($this->params); $this->connection->release($client);
return $result;
return $this->getLogger()->logCategory($errorMsg . PHP_EOL, 'mysql'); } catch (Throwable $throwable) {
} finally { $this->getLogger()->json_log($throwable);
$this->connection->release($client);
} if ($this->isRefresh($throwable) && $retryCount < self::MAX_RETRIES) {
} $retryCount++;
$this->connection->connections->abandon($this->connection->getName());
continue;
/** }
* @return bool
* @throws $this->connection->release($client);
*/
public function flush(): bool $errorMsg = $throwable->getMessage() . PHP_EOL . ' Sql: ' . $this->sql . '.' . json_encode($this->params);
{
return (bool)$this->_prepare(); return $this->getLogger()->logCategory($errorMsg . PHP_EOL, 'mysql');
} }
}
/** $errorMsg = 'Max retry exceeded for SQL: ' . $this->sql;
* @return int|bool return $this->getLogger()->logCategory($errorMsg, 'mysql');
* @throws }
*/
private function _prepare(): int|bool
{ /**
$client = $this->connection->getConnection(); * @return bool
try { * @throws
$startTime = microtime(true); */
if (($prepare = $client->prepare($this->sql)) === false) { public function flush(): bool
throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $prepare->errorInfo()[2]); {
} return (bool)$this->_prepare();
if ($prepare->execute($this->params) === false) { }
throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $prepare->errorInfo()[2]);
}
$prepare->closeCursor(); /**
* 执行写入类 SQL 操作,内置断连重试机制
$result = $client->lastInsertId(); * 当数据库连接断开时自动重试,断开的连接不再归还池中防止污染连接池
* @return int|bool
$this->connection->println($startTime, microtime(true), $this->sql, $this->params); * @throws
*/
return $result == 0 ? $prepare->rowCount() : (int)$result; private function _prepare(): int|bool
} catch (Throwable $throwable) { {
$this->getLogger()->json_log($throwable); $retryCount = 0;
if ($this->isRefresh($throwable)) { while ($retryCount <= self::MAX_RETRIES) {
return $this->_prepare(); $client = $this->connection->getConnection();
} try {
$startTime = microtime(true);
$errorMsg = $throwable->getMessage() . PHP_EOL . ' Sql: ' . $this->sql . '.' . json_encode($this->params, JSON_UNESCAPED_UNICODE); if (($prepare = $client->prepare($this->sql)) === false) {
throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $prepare->errorInfo()[2]);
return $this->getLogger()->logCategory($errorMsg . PHP_EOL, 'mysql'); }
} finally { if ($prepare->execute($this->params) === false) {
$this->connection->release($client); throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $prepare->errorInfo()[2]);
} }
}
$prepare->closeCursor();
/** $result = $client->lastInsertId();
* @param Throwable $throwable
* @return bool $this->connection->println($startTime, microtime(true), $this->sql, $this->params);
*/ if (str_starts_with($this->sql, 'DELETE')) {
protected function isRefresh(Throwable $throwable): bool $this->connection->release($client);
{ return $prepare->rowCount();
$message = $throwable->getMessage(); }
// MySQL 错误处理 $this->connection->release($client);
if (str_contains($message, 'MySQL server has gone away')) { return $result == 0 ? $prepare->rowCount() : (int)$result;
return true; } catch (Throwable $throwable) {
} $this->getLogger()->json_log($throwable);
if (str_contains($message, 'Send of 14 bytes failed with errno=32 Broken pipe')) {
return true; if ($this->isRefresh($throwable) && $retryCount < self::MAX_RETRIES) {
} $retryCount++;
if (str_contains($message, 'Lost connection to MySQL server during query')) { $this->connection->connections->abandon($this->connection->getName());
return true; continue;
} }
// PostgreSQL 错误处理 $this->connection->release($client);
if (str_contains($message, 'server closed the connection unexpectedly')) {
return true; $errorMsg = $throwable->getMessage() . PHP_EOL . ' Sql: ' . $this->sql . '.' . json_encode($this->params, JSON_UNESCAPED_UNICODE);
}
if (str_contains($message, 'Connection refused')) { return $this->getLogger()->logCategory($errorMsg . PHP_EOL, 'mysql');
return true; }
} }
if (str_contains($message, 'Broken pipe')) {
return true; $errorMsg = 'Max retry exceeded for SQL: ' . $this->sql;
} return $this->getLogger()->logCategory($errorMsg, 'mysql');
if (str_contains($message, 'connection to server was lost')) { }
return true;
}
/**
return false; * @param Throwable $throwable
} * @return bool
*/
protected function isRefresh(Throwable $throwable): bool
/** {
* @return bool $message = $throwable->getMessage();
* @throws
*/ // MySQL 错误处理
public function delete(): bool if (str_contains($message, 'MySQL server has gone away')) {
{ return true;
return (bool)$this->_prepare(); }
} 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 int|bool return true;
*/ }
public function exec(): int|bool
{ // PostgreSQL 错误处理
return $this->_prepare(); if (str_contains($message, 'server closed the connection unexpectedly')) {
} return true;
}
/** if (str_contains($message, 'Connection refused')) {
* @param array $data return true;
* @return $this }
*/ if (str_contains($message, 'Broken pipe')) {
public function bindValues(array $data = []): static return true;
{ }
if (count($data) > 0) { if (str_contains($message, 'connection to server was lost')) {
$this->params = array_merge($this->params, $data); return true;
} }
return $this;
} 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;
}
}
+59 -55
View File
@@ -248,61 +248,65 @@ class Connection extends Component
} }
/** /**
* @throws * @throws
* 事务回滚 * 事务回滚
*/ */
public function rollback(): void public function rollback(): void
{ {
$this->storey--; $this->storey--;
if ($this->storey == 0) { if ($this->storey == 0) {
if (!Context::exists($this->cds)) { if (!Context::exists($this->cds)) {
return; return;
} }
$pdo = $this->getTransactionClient(); $pdo = $this->getTransactionClient();
if ($pdo->inTransaction()) { if ($pdo->inTransaction()) {
$pdo->rollback(); $pdo->rollback();
} }
$this->release($pdo); $this->release($pdo);
} Context::remove($this->cds);
} }
}
/**
* @throws /**
* 事务提交 * @throws
*/ * 事务提交
public function commit(): void */
{ public function commit(): void
$this->storey--; {
if ($this->storey == 0) { $this->storey--;
if (!Context::exists($this->cds)) { if ($this->storey == 0) {
return; if (!Context::exists($this->cds)) {
} return;
$pdo = $this->getTransactionClient(); }
if ($pdo->inTransaction()) { $pdo = $this->getTransactionClient();
$pdo->commit(); if ($pdo->inTransaction()) {
} $pdo->commit();
$this->release($pdo); }
} $this->release($pdo);
} Context::remove($this->cds);
}
}
/**
* @return void
* @throws /**
*/ * @return void
public function clear(): void * @throws
{ */
/** @var PDO $pdo */ public function clear(): void
$pdo = Context::get($this->cds); {
if ($pdo === null) { /** @var PDO $pdo */
return; $pdo = Context::get($this->cds);
} if ($pdo === null) {
if ($this->inTransaction()) { return;
$pdo->rollback(); }
} if ($this->inTransaction()) {
$this->release($pdo); $pdo->rollback();
} }
$this->release($pdo);
Context::remove($this->cds);
$this->storey = 0;
}
/** /**
+2
View File
@@ -285,6 +285,7 @@ class Model extends Base\Model
$primaryKey = str_replace('\\', '_', $modelName) . '_' . $foreignKey . '_' . $value; $primaryKey = str_replace('\\', '_', $modelName) . '_' . $foreignKey . '_' . $value;
if (!$relation->hasIdentification($primaryKey)) { if (!$relation->hasIdentification($primaryKey)) {
/** @var $modelName ModelInterface */
$relation->bindIdentification($primaryKey, $modelName::query()->where([$foreignKey => $value])); $relation->bindIdentification($primaryKey, $modelName::query()->where([$foreignKey => $value]));
} }
return $primaryKey; return $primaryKey;
@@ -346,6 +347,7 @@ class Model extends Base\Model
$primaryKey = str_replace('\\', '_', $modelName) . '_' . $foreignKey . '_' . implode('_', $value); $primaryKey = str_replace('\\', '_', $modelName) . '_' . $foreignKey . '_' . implode('_', $value);
if (!$relation->hasIdentification($primaryKey)) { if (!$relation->hasIdentification($primaryKey)) {
/** @var $modelName ModelInterface */
$relation->bindIdentification($primaryKey, $modelName::query()->whereIn($foreignKey, $value)); $relation->bindIdentification($primaryKey, $modelName::query()->whereIn($foreignKey, $value));
} }
+470 -462
View File
@@ -1,462 +1,470 @@
<?php <?php
/** @noinspection ALL */ /** @noinspection ALL */
declare(strict_types=1); declare(strict_types=1);
namespace Database; namespace Database;
use JetBrains\PhpStorm\Pure; use JetBrains\PhpStorm\Pure;
use Kiri; use Kiri;
use Kiri\Abstracts\Component; use Kiri\Abstracts\Component;
use Database\Traits\QueryTrait;
/**
* Class SqlBuilder /**
* @package Database * Class SqlBuilder
*/ * @package Database
class SqlBuilder extends Component */
{ class SqlBuilder extends Component
{
/**
* @var ActiveQuery|Query|ISqlBuilder|null /**
*/ * @var ActiveQuery|Query|ISqlBuilder|null
public ActiveQuery|Query|ISqlBuilder|null $query; */
public ActiveQuery|Query|ISqlBuilder|null $query;
/**
* @param ActiveQuery|Query|ISqlBuilder|null $config /**
*/ * @param ActiveQuery|Query|ISqlBuilder|null $config
public function __construct(ActiveQuery|Query|null|ISqlBuilder $config) */
{ public function __construct(ActiveQuery|Query|null|ISqlBuilder $config)
parent::__construct(); {
parent::__construct();
$this->query = $config;
} $this->query = $config;
}
/**
* @param ISqlBuilder|null $query /**
* @return $this * @param ISqlBuilder|null $query
* @throws * @return $this
*/ * @throws
public static function builder(ISqlBuilder|null $query): static */
{ public static function builder(ISqlBuilder|null $query): static
return new static($query); {
} return new static($query);
}
/**
* @return string /**
* @throws * @return string
*/ * @throws
public function getCondition(): string */
{ public function getCondition(): string
return $this->where($this->query->getWhere()); {
} return $this->where($this->query->where);
}
/**
* @param array $compiler /**
* @return string * @param array $compiler
* @throws * @return string
*/ * @throws
public function hashCompiler(array $compiler): string */
{ public function hashCompiler(array $compiler): string
return $this->where($compiler); {
} return $this->where($compiler);
}
/**
* @param array $attributes /**
* @return bool|array * @param array $attributes
* @throws * @return bool|array
*/ * @throws
public function update(array $attributes): bool|string */
{ public function update(array $attributes): bool|string
$conditions = $this->query->getParams(); {
$this->query->setParams([]); $conditions = $this->query;
$data = $this->__updateBuilder($this->makeParams($attributes)); $this->query = [];
foreach ($conditions as $condition) { $data = $this->__updateBuilder($this->makeParams($attributes));
$this->query->pushParam($condition); foreach ($conditions as $condition) {
} $this->query->pushParam($condition);
return $data; }
} return $data;
}
/**
* @param array $attributes /**
* @param string $opera * @param array $attributes
* @return bool|array * @param string $opera
* @throws * @return bool|array
*/ * @throws
public function mathematics(array $attributes, string $opera = '+'): bool|string */
{ public function mathematics(array $attributes, string $opera = '+'): bool|string
$string = []; {
foreach ($attributes as $attribute => $value) { $string = [];
$string[] = $attribute . '=' . $attribute . $opera . $value; foreach ($attributes as $attribute => $value) {
} $string[] = $attribute . '=' . $attribute . $opera . $value;
return $this->__updateBuilder($string); }
} return $this->__updateBuilder($string);
}
/**
* @param array $string /**
* @return string|bool * @param array $string
* @throws * @return string|bool
*/ * @throws
private function __updateBuilder(array $string): string|bool */
{ private function __updateBuilder(array $string): string|bool
if (empty($string)) { {
return Kiri::getLogger()->logCategory('None data update.'); if (empty($string)) {
} return Kiri::getLogger()->logCategory('None data update.');
return 'UPDATE ' . $this->query->getFrom() . ' SET ' . implode(',', $string) . $this->make(); }
} return 'UPDATE ' . $this->getFromAlias() . ' SET ' . implode(',', $string) . $this->make();
}
/**
* @param array $attributes /**
* @param false $isBatch * @param array $attributes
* @return array * @param false $isBatch
* @throws * @return array
*/ * @throws
public function insert(array $attributes, bool $isBatch = false): string */
{ public function insert(array $attributes, bool $isBatch = false): string
$update = 'INSERT INTO ' . $this->query->getFrom(); {
if ($isBatch === false) { $update = 'INSERT INTO ' . $this->getFromAlias();
$attributes = [$attributes]; if ($isBatch === false) {
} $attributes = [$attributes];
$update .= '(' . implode(',', $this->getFields($attributes)) . ') VALUES '; }
$update .= '(' . implode(',', $this->getFields($attributes)) . ') VALUES ';
$keys = [];
foreach ($attributes as $attribute) { $keys = [];
$_keys = $this->makeParams($attribute, true); foreach ($attributes as $attribute) {
$_keys = $this->makeParams($attribute, true);
$keys[] = implode(',', $_keys);
} $keys[] = implode(',', $_keys);
return $update . '(' . implode('),(', $keys) . ')'; }
} return $update . '(' . implode('),(', $keys) . ')';
}
/**
* @return string /**
* @throws * @return string
*/ * @throws
public function delete(): string */
{ public function delete(): string
return 'DELETE FROM ' . $this->query->getFrom() . $this->make(); {
} return 'DELETE FROM ' . $this->getFromAlias() . $this->make();
}
/**
* @param $attributes /**
* @return array * @param $attributes
*/ * @return array
#[Pure] private function getFields(array $attributes): array */
{ #[Pure] private function getFields(array $attributes): array
return array_keys(current($attributes)); {
} return array_keys(current($attributes));
}
/**
* @param array $attributes /**
* @param bool $isInsert * @param array $attributes
* @return array[] * @param bool $isInsert
* a=:b, * @return array[]
*/ * a=:b,
private function makeParams(array $attributes, bool $isInsert = false): array */
{ private function makeParams(array $attributes, bool $isInsert = false): array
$keys = []; {
foreach ($attributes as $key => $value) { $keys = [];
if ($isInsert === true) { foreach ($attributes as $key => $value) {
$keys[] = '?'; if ($isInsert === true) {
$this->query->pushParam($value); $keys[] = '?';
} else { $this->query->pushParam($value);
$keys = $this->resolveParams($key, $value, $keys); } else {
} $keys = $this->resolveParams($key, $value, $keys);
} }
return $keys; }
} return $keys;
}
/**
* @param string $key /**
* @param mixed $value * @param string $key
* @param array $keys * @param mixed $value
* @return array * @param array $keys
*/ * @return array
private function resolveParams(string $key, mixed $value, array $keys): array */
{ private function resolveParams(string $key, mixed $value, array $keys): array
if (is_null($value)) { {
return $keys; if (is_null($value)) {
} return $keys;
if (preg_match('/^[+|-]\s\d+$/', (string)$value)) { }
$keys[] = $key . '=' . $key . ' ' . $value; if (preg_match('/^[+|-]\s\d+$/', (string)$value)) {
} else { $keys[] = $key . '=' . $key . ' ' . $value;
$this->query->pushParam($value); } else {
$keys[] = $key . '= ?'; $this->query->pushParam($value);
} $keys[] = $key . '= ?';
return $keys; }
} return $keys;
}
/**
* @return string /**
* @throws * @return string
*/ * @throws
public function one(): string */
{ public function one(): string
$this->query->offset(0)->limit(1); {
return $this->makeSelect($this->query->getSelect()) . $this->make() . $this->makeLimit(); $this->query->offset(0)->limit(1);
} return $this->makeSelect($this->query) . $this->make() . $this->makeLimit();
}
/**
* @return string /**
* @throws * @return string
*/ * @throws
public function all(): string */
{ public function all(): string
return $this->makeSelect($this->query->getSelect()) . $this->make() . $this->makeLimit(); {
} return $this->makeSelect($this->query) . $this->make() . $this->makeLimit();
}
/**
* @return string /**
* @throws * @return string
*/ * @throws
public function count(): string */
{ public function count(): string
return $this->makeSelect(['COUNT(*)']) . $this->make(); {
} return $this->makeSelect(['COUNT(*)']) . $this->make();
}
/**
* @return string /**
* @throws * @return string
*/ * @throws
public function exists(): string */
{ public function exists(): string
return $this->makeSelect(['0']) . $this->make(); {
} return $this->makeSelect(['0']) . $this->make();
}
/**
* @param string $table /**
* @return string * @param string $table
* @throws * @return string
*/ * @throws
public function columns(string $table): string */
{ public function columns(string $table): string
$driver = $this->getDriver(); {
if (in_array($driver, ['pgsql', 'postgresql'])) { $driver = $this->getDriver();
// PostgreSQL 使用 information_schema if (in_array($driver, ['pgsql', 'postgresql'])) {
$tableName = trim($table, '"`'); $tableName = trim($table, '"`');
return "SELECT return "SELECT
column_name as Field, column_name as Field,
data_type as Type, data_type as Type,
is_nullable as Null, is_nullable as Null,
column_default as Default, column_default as Default,
'' as Extra, '' as Extra,
'' as Key '' as Key
FROM information_schema.columns FROM information_schema.columns
WHERE table_name = '$tableName' WHERE table_name = '$tableName'
ORDER BY ordinal_position"; ORDER BY ordinal_position";
} }
// MySQL 使用 SHOW FULL FIELDS return 'SHOW FULL FIELDS FROM ' . $table;
return 'SHOW FULL FIELDS FROM ' . $table; }
}
/**
/** * @return string
* @return string * @throws
* @throws */
*/ private function make(): string
private function make(): string {
{ $select = $this->makeCondition();
$select = $this->makeCondition(); $select .= $this->makeGroup();
$select .= $this->makeGroup(); $select .= $this->makeOrder();
$select .= $this->makeOrder(); return $select;
return $select; }
}
/**
/** * @param mixed $select 列名数组或 QueryTrait 对象
* @param array $select * @return string
* @return string */
*/ private function makeSelect(mixed $select = ['*']): string
private function makeSelect(array $select = ['*']): string {
{ if (!is_array($select)) {
$select = "SELECT " . implode(',', $select) . " FROM " . $this->query->getFrom(); $select = ['*'];
if ($this->query->getAlias() != "") { }
$select .= " AS " . $this->query->getAlias(); $sql = "SELECT " . implode(',', $select) . " FROM " . $this->getFromAlias();
} if ($this->query->join !== []) {
if (count($this->query->getJoin()) > 0) { $sql .= ' ' . implode(' ', $this->query->join);
$select .= ' ' . implode(' ', $this->query->getJoin()); }
} return $sql;
return $select; }
}
/**
/** * @return string
* @return string */
*/ private function makeGroup(): string
private function makeGroup(): string {
{ if ($this->query->group !== '' && $this->query->group !== '0') {
if ($this->query->getGroup() != "") { return ' GROUP BY ' . $this->query->group;
return ' GROUP BY ' . $this->query->getGroup(); }
} return '';
return ''; }
}
/**
/** * @return string
* @return string */
*/ private function makeOrder(): string
private function makeOrder(): string {
{ if (count($this->query->order) > 0) {
if (count($this->query->getOrder()) > 0) { return ' ORDER BY ' . implode(',', $this->query->order);
return ' ORDER BY ' . implode(',', $this->query->getOrder()); }
} return '';
return ''; }
}
/**
/** * @return string
* @return string */
*/ private function makeCondition(): string
private function makeCondition(): string {
{ $condition = $this->where($this->query->where);
$condition = $this->where($this->query->getWhere()); if (empty($condition)) {
if (empty($condition)) { return '';
return ''; }
} return ' WHERE ' . $condition;
return ' WHERE ' . $condition; }
}
/**
/** * @return string
* @return string * @throws
* @throws */
*/ private function makeLimit(): string
private function makeLimit(): string {
{ if ($this->query->offset >= 0 && $this->query->limit >= 1) {
if ($this->query->getOffset() >= 0 && $this->query->getLimit() >= 1) { $driver = $this->getDriver();
$driver = $this->getDriver(); if (in_array($driver, ['pgsql', 'postgresql'])) {
if (in_array($driver, ['pgsql', 'postgresql'])) { return ' LIMIT ' . $this->query->limit . ' OFFSET ' . $this->query->offset;
// PostgreSQL 使用 LIMIT ... OFFSET ... 语法 }
return ' LIMIT ' . $this->query->getLimit() . ' OFFSET ' . $this->query->getOffset(); return ' LIMIT ' . $this->query->offset . ',' . $this->query->limit;
} }
// MySQL 使用 LIMIT offset,limit 语法 return '';
return ' LIMIT ' . $this->query->getOffset() . ',' . $this->query->getLimit(); }
}
return '';
} /**
* @param false $isCount
* @return string
/** * @throws
* @param false $isCount */
* @return string public function get(bool $isCount = false): string
* @throws {
*/ if ($isCount === false) {
public function get(bool $isCount = false): string return $this->all();
{ }
if ($isCount === false) { return $this->count();
return $this->all(); }
}
return $this->count();
} /**
* @return string
* @throws
/** */
* @return string public function truncate(): string
* @throws {
*/ return sprintf('TRUNCATE %s', $this->getFromAlias());
public function truncate(): string }
{
return sprintf('TRUNCATE %s', $this->query->getFrom());
} /**
* @param array $where
* @return string
/** */
* @param array $where private function where(array $where): string
* @return string {
*/ if (count($where) < 1) {
private function where(array $where): string return '';
{ }
if (count($where) < 1) { $_tmp = [];
return ''; foreach ($where as $key => $value) {
} $_tmp[] = $this->resolveCondition($key, $value);
$_tmp = []; }
foreach ($where as $key => $value) { return implode(' AND ', $_tmp);
$_tmp[] = $this->resolveCondition($key, $value); }
}
return implode(' AND ', $_tmp);
} /**
* @param string $field
* @param mixed $condition
/** * @return string
* @param string $field */
* @param mixed $condition private function resolveCondition(string|int $field, mixed $condition): string
* @return string {
*/ if (is_string($field)) {
private function resolveCondition(string|int $field, mixed $condition): string $this->query->pushParam($condition);
{ return $field . ' = ?';
if (is_string($field)) { } else if (is_string($condition)) {
$this->query->pushParam($condition); return $condition;
return $field . ' = ?'; } else {
} else if (is_string($condition)) { return implode(' AND ', $this->_hashMap($condition));
return $condition; }
} else { }
return implode(' AND ', $this->_hashMap($condition));
}
} /**
* @param $condition
* @return array
/** */
* @param $condition private function _hashMap(array $condition): array
* @return array {
*/ $_array = [];
private function _hashMap(array $condition): array foreach ($condition as $key => $value) {
{ $this->query->pushParam($value);
$_array = []; $_array[] = $key . '= ?';
foreach ($condition as $key => $value) { }
$this->query->pushParam($value); return $_array;
$_array[] = $key . '= ?'; }
}
return $_array; /**
} * 获取数据库驱动类型
* @return string
/** * @throws
* 获取数据库驱动类型 */
* @return string private function getDriver(): string
* @throws {
*/ try {
private function getDriver(): string if ($this->query instanceof ActiveQuery && $this->query->modelClass !== null) {
{ if ($this->query->modelClass instanceof Model) {
try { $connection = $this->query->modelClass->getConnection();
// 尝试从 query 对象获取 connection if ($connection instanceof Connection) {
if ($this->query instanceof ActiveQuery && $this->query->modelClass !== null) { return strtolower($connection->driver ?? 'mysql');
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');
// 如果是 Db 查询,尝试获取默认 connection }
if (method_exists($this->query, 'getConnection')) { }
$connection = $this->query->getConnection(); } catch (\Throwable $e) {
if ($connection instanceof Connection) { $this->getLogger()->json_log($e);
return strtolower($connection->driver ?? 'mysql'); }
} return 'mysql';
} }
} catch (\Throwable $e) {
// 忽略错误,返回默认值
$this->getLogger()->json_log($e); /**
} * 获取 FROM 子句(表名 + 别名)
// 默认返回 mysql * QueryTrait 对象有 $from 和 $alias 属性
return 'mysql'; * @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;
}
}
+113 -253
View File
@@ -27,26 +27,93 @@ use Kiri\Abstracts\Component;
*/ */
abstract class QueryTrait extends Component implements ActiveQueryInterface, ISqlBuilder abstract class QueryTrait extends Component implements ActiveQueryInterface, ISqlBuilder
{ {
protected array $where = []; public array $where = [] {
protected array $select = ['t1.*']; &get {
protected array $join = []; return $this->where;
protected array $order = []; }
protected int $offset = -1; }
protected int $limit = -1; public array $select = ['t1.*'] {
protected string $group = ''; &get {
protected string $from = ''; return $this->select;
protected string $alias = ''; }
protected array $filter = []; }
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 bool $lock = FALSE;
protected SqlBuilder $builder; protected SqlBuilder $builder;
protected array $params = []; public array $params = [] {
protected array $_alias = []; &get {
return $this->params;
}
}
protected array $_alias = [] {
&get {
return $this->_alias;
}
}
/** /**
* @var ModelInterface|string|null * @var ModelInterface|string|null
*/ */
protected ModelInterface|string|null $modelClass; protected ModelInterface|string|null $modelClass {
get {
return $this->modelClass;
}
}
/** /**
@@ -63,29 +130,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 Closure|Query $callback
* @param array $attributes * @param array $attributes
* *
* @return $this * @return $this
*/ */
@@ -93,7 +140,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
{ {
if ($callback instanceof Query) { if ($callback instanceof Query) {
$this->where[] = 'NOT EXISTS(' . $callback->build() . ')'; $this->where[] = 'NOT EXISTS(' . $callback->build() . ')';
$this->mergeParams($callback->getParams()); $this->mergeParams($callback->params);
} else { } else {
$this->where[] = 'NOT EXISTS(' . $this->makeClosureFunction($callback) . ')'; $this->where[] = 'NOT EXISTS(' . $this->makeClosureFunction($callback) . ')';
} }
@@ -106,7 +153,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param Closure|Query $callback * @param Closure|Query $callback
* @param array $attributes * @param array $attributes
* *
* @return $this * @return $this
*/ */
@@ -114,7 +161,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
{ {
if ($callback instanceof Query) { if ($callback instanceof Query) {
$this->where[] = 'EXISTS(' . $callback->build() . ')'; $this->where[] = 'EXISTS(' . $callback->build() . ')';
$this->mergeParams($callback->getParams()); $this->mergeParams($callback->params);
} else { } else {
$this->where[] = 'EXISTS(' . $this->makeClosureFunction($callback) . ')'; $this->where[] = 'EXISTS(' . $this->makeClosureFunction($callback) . ')';
} }
@@ -125,194 +172,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 * @param array $params
* @return void * @return void
@@ -326,7 +185,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param string $column * @param string $column
* @param callable $callable * @param callable $callable
* *
* @return $this * @return $this
@@ -465,7 +324,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
{ {
if ($tableName instanceof Query) { if ($tableName instanceof Query) {
$this->from = '(' . $tableName->build() . ')'; $this->from = '(' . $tableName->build() . ')';
$this->mergeParams($tableName->getParams()); $this->mergeParams($tableName->params);
} else if ($tableName instanceof Closure) { } else if ($tableName instanceof Closure) {
$this->from = '(' . $this->makeClosureFunction($tableName) . ')'; $this->from = '(' . $this->makeClosureFunction($tableName) . ')';
} else if (class_exists($tableName)) { } else if (class_exists($tableName)) {
@@ -480,8 +339,8 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param string $tableName * @param string $tableName
* @param string $alias * @param string $alias
* @param array $on * @param array $on
* @param array $param * @param array $param
* *
* @return $this * @return $this
* $query->join([$tableName, ['userId'=>'uuvOd']], $param) * $query->join([$tableName, ['userId'=>'uuvOd']], $param)
@@ -563,8 +422,8 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param string $tableName * @param string $tableName
* @param string $alias * @param string $alias
* @param array $onCondition * @param array $onCondition
* @param array $param * @param array $param
* *
* @return $this * @return $this
* @throws Exception * @throws Exception
@@ -585,8 +444,8 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param string $tableName * @param string $tableName
* @param string $alias * @param string $alias
* @param array $onCondition * @param array $onCondition
* @param array $param * @param array $param
* *
* @return $this * @return $this
* @throws Exception * @throws Exception
@@ -607,8 +466,8 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param string $tableName * @param string $tableName
* @param string $alias * @param string $alias
* @param array $onCondition * @param array $onCondition
* @param array $param * @param array $param
* *
* @return $this * @return $this
* @throws Exception * @throws Exception
@@ -651,8 +510,8 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param string $lngField * @param string $lngField
* @param string $latField * @param string $latField
* @param int|float $lng1 * @param int|float $lng1
* @param int|float $lat1 * @param int|float $lat1
* *
@@ -668,7 +527,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param string|array $column * @param string|array $column
* @param string $sort * @param string $sort
* *
* @return $this * @return $this
* *
@@ -736,8 +595,8 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param array|Closure|string $conditionArray * @param array|Closure|string $conditionArray
* @param string $opera * @param string $opera
* @param null $value * @param null $value
* *
* @return QueryTrait * @return QueryTrait
* @throws * @throws
@@ -816,7 +675,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param string $columns * @param string $columns
* @param array|Closure|Query $value * @param array|Closure|Query $value
* *
* @return $this * @return $this
@@ -825,12 +684,13 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
public function whereIn(string $columns, array|Closure|Query $value): static public function whereIn(string $columns, array|Closure|Query $value): static
{ {
if (is_array($value)) { 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) . ')'; $this->where[] = $columns . ' IN (' . implode(',', $value) . ')';
} else if ($value instanceof Query) { } else if ($value instanceof Query) {
$this->where[] = $columns . ' IN (' . $value->build() . ')'; $this->where[] = $columns . ' IN (' . $value->build() . ')';
$this->mergeParams($value->getParams()); $this->mergeParams($value->params);
} else { } else {
$this->where[] = $columns . ' IN (' . $this->makeClosureFunction($value) . ')'; $this->where[] = $columns . ' IN (' . $this->makeClosureFunction($value) . ')';
} }
@@ -844,13 +704,13 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
*/ */
public function queryInstance(): Query public function queryInstance(): Query
{ {
return new Query(); return new Query;
} }
/** /**
* @param string $columns * @param string $columns
* @param array $value * @param array $value
* *
* @return $this * @return $this
*/ */
@@ -865,7 +725,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param string $column * @param string $column
* @param int|float $start * @param int|float $start
* @param int|float $end * @param int|float $end
* *
@@ -886,7 +746,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param string $column * @param string $column
* @param int|float $start * @param int|float $start
* @param int|float $end * @param int|float $end
* *
@@ -920,7 +780,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param array|string $column * @param array|string $column
* @param mixed $value * @param mixed $value
* *
* @return $this * @return $this
*/ */
@@ -940,7 +800,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param string $column * @param string $column
* @param string $opera * @param string $opera
* @param mixed $value * @param mixed $value
* *
* @return $this * @return $this
*/ */
@@ -978,13 +838,13 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
} else { } else {
$generate->addArray($closure); $generate->addArray($closure);
} }
$this->mergeParams($generate->getParams()); $this->mergeParams($generate->params);
return $generate->build(); return $generate->build();
} }
/** /**
* @param string $name * @param string $name
* @param string|null $having * @param string|null $having
* *
* @return $this * @return $this
@@ -1060,7 +920,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param string $column * @param string $column
* @param mixed $value * @param mixed $value
* @param string $opera * @param string $opera
* *
* @return string * @return string
@@ -1084,7 +944,7 @@ abstract class QueryTrait extends Component implements ActiveQueryInterface, ISq
/** /**
* @param string $querySql * @param string $querySql
* @param array $params * @param array $params
* *
* @return Command * @return Command
*/ */
+65 -46
View File
@@ -18,61 +18,80 @@ use JetBrains\PhpStorm\Pure;
class When class When
{ {
public ActiveQuery|ISqlBuilder $query; public ActiveQuery|ISqlBuilder $query;
private array $_condition = []; /**
* @var array
private string $else = ''; */
private array $_condition = [] {
&get {
return $this->_condition;
}
}
/** /**
* CaseWhen constructor. * @var string
* @param string $column */
* @param ActiveQuery|ISqlBuilder $activeQuery private string $else = '' {
*/ get {
public function __construct(public string $column, public ActiveQuery|ISqlBuilder $activeQuery) return $this->else;
{ }
$this->_condition[] = 'CASE ' . $column; set(string $value) {
} $this->else = $value;
}
}
/** /**
* @param string|int $condition * CaseWhen constructor.
* @param string $then * @param string $column
* @return $this * @param ActiveQuery|ISqlBuilder $activeQuery
* @throws */
*/ public function __construct(public string $column, public ActiveQuery|ISqlBuilder $activeQuery)
public function when(string|int $condition, string $then): static {
{ $this->_condition[] = 'CASE ' . $column;
$this->_condition[] = sprintf('WHEN %s THEN %s', $condition, $then); }
return $this;
}
/** /**
* @param string $alias * @param string|int $condition
*/ * @param string $then
public function else(string $alias): void * @return $this
{ * @throws
$this->else = $alias; */
} public function when(string|int $condition, string $then): static
{
$this->_condition[] = sprintf('WHEN %s THEN %s', $condition, $then);
return $this;
}
/** /**
* @return string * @param string $alias
*/ */
#[Pure] public function end(): string public function else(string $alias): void
{ {
if (empty($this->_condition)) { $this->else = $alias;
return ''; }
}
$prefix = implode(' ', $this->_condition);
if (!empty($this->else)) { /**
$prefix .= ' ELSE ' . $this->else; * @return string
} */
return '(' . $prefix . ' END)'; #[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", "license": "MIT",
"require": { "require": {
"php": ">=8.4", "php": ">=8.5",
"ext-json": "*", "ext-json": "*",
"ext-pdo": "*", "ext-pdo": "*",
"game-worker/kiri-pool": "^v1.0" "game-worker/kiri-pool": "^v1.0"