Compare commits
63 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 39905b77e3 | |||
| fce82ec56c | |||
| 2970c3bc6b | |||
| a35d0bb921 | |||
| fce2b8bcdb | |||
| ce92f310a5 | |||
| 28fa17ecfb | |||
| 9fc910f10f | |||
| 0a95e722d6 | |||
| 9f2611972f | |||
| b7e59d1c96 | |||
| 9b0ee54608 | |||
| 763a4baa57 | |||
| fe639bbd1f | |||
| e40de4ee4a | |||
| ffc2af69da | |||
| 218a38da48 | |||
| 5acb2f4600 | |||
| d6a07446bb | |||
| 6c936c3896 | |||
| 14448d824a | |||
| 06074c38c1 | |||
| 404b511c5f | |||
| bf4f8538ed | |||
| 8a985a65df | |||
| c7b980d969 | |||
| fea649413c | |||
| 959407a95e | |||
| 0abc7c32f7 | |||
| 9c6340d0b3 | |||
| 0f01416351 | |||
| b1d2c0fb51 | |||
| db18a2d73a | |||
| 3cb3dcf384 | |||
| fd8652ccd1 | |||
| 279151825a | |||
| 3dd2d4abf2 | |||
| 8bf5027582 | |||
| 0b27a7dc7d | |||
| c016853ff8 | |||
| 758c322679 | |||
| cbbc6ea7f1 | |||
| 456c7f29b6 | |||
| 89d5b604d6 | |||
| c943d6258b | |||
| 466cee4ab5 | |||
| ee1a5a993d | |||
| 0dcd548645 | |||
| d796a86ce3 | |||
| ef19079727 | |||
| dcee10b77b | |||
| 2bf66fc17f | |||
| 61f549362d | |||
| 583e2214c3 | |||
| e190c42dd0 | |||
| 190c272ea3 | |||
| 28c64daae7 | |||
| 7c347bc373 | |||
| d5c9b3bc86 | |||
| ef7e85746d | |||
| 89f6949556 | |||
| 555d160b40 | |||
| 45b949747d |
@@ -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('@'));
|
||||
|
||||
}
|
||||
+1
-1
@@ -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();
|
||||
}
|
||||
}
|
||||
|
||||
+1
-1
@@ -31,7 +31,7 @@ class BackupCommand extends Command
|
||||
/**
|
||||
*
|
||||
*/
|
||||
protected function configure()
|
||||
protected function configure(): void
|
||||
{
|
||||
$this->setName('db:backup')
|
||||
->addOption('data', 'd', InputArgument::OPTIONAL)
|
||||
|
||||
@@ -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;
|
||||
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
+849
-826
File diff suppressed because it is too large
Load Diff
+16
-1
@@ -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,
|
||||
};
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
+2
-2
@@ -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');
|
||||
|
||||
+286
-230
@@ -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;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -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);
|
||||
|
||||
@@ -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');
|
||||
|
||||
+48
-36
@@ -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];
|
||||
@@ -134,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;
|
||||
}
|
||||
|
||||
|
||||
@@ -263,6 +264,7 @@ class Connection extends Component
|
||||
$pdo->rollback();
|
||||
}
|
||||
$this->release($pdo);
|
||||
Context::remove($this->cds);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -282,6 +284,7 @@ class Connection extends Component
|
||||
$pdo->commit();
|
||||
}
|
||||
$this->release($pdo);
|
||||
Context::remove($this->cds);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -301,6 +304,8 @@ class Connection extends Component
|
||||
$pdo->rollback();
|
||||
}
|
||||
$this->release($pdo);
|
||||
Context::remove($this->cds);
|
||||
$this->storey = 0;
|
||||
}
|
||||
|
||||
|
||||
@@ -312,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);
|
||||
}
|
||||
|
||||
|
||||
@@ -356,15 +361,22 @@ class Connection extends Component
|
||||
*/
|
||||
public function newConnect(): \PDO
|
||||
{
|
||||
$pdo = new PDO($this->database, $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);
|
||||
}
|
||||
|
||||
+10
-8
@@ -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'] ?? [],
|
||||
|
||||
@@ -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);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
@@ -282,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;
|
||||
@@ -343,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
-402
@@ -1,402 +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 (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
|
||||
{
|
||||
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;
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
@@ -12,8 +12,6 @@ namespace Database\Traits;
|
||||
use Database\ModelInterface;
|
||||
use Database\Collection;
|
||||
use Database\Relation;
|
||||
use Kiri;
|
||||
use Kiri\Di\Context;
|
||||
|
||||
/**
|
||||
* Class HasBase
|
||||
|
||||
+951
-1043
File diff suppressed because it is too large
Load Diff
+65
-46
@@ -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)';
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
+2
-2
@@ -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": {
|
||||
|
||||
Reference in New Issue
Block a user