Compare commits

...

68 Commits

Author SHA1 Message Date
as2252258 b3f50fc6e4 eee 2024-12-23 10:03:35 +08:00
as2252258 7b4604ce98 eee 2024-12-20 18:20:54 +08:00
as2252258 654cc483b3 eee 2024-12-18 11:42:09 +08:00
as2252258 c6b88256e3 eee 2024-12-18 11:31:15 +08:00
as2252258 5b4a6b9caa eee 2024-12-13 15:14:13 +08:00
as2252258 42af2e8bf9 eee 2024-12-13 15:03:09 +08:00
as2252258 db92238e9e eee 2024-12-13 14:45:03 +08:00
as2252258 530104ceb6 eee 2024-12-13 14:44:20 +08:00
as2252258 7ea245fc89 eee 2024-12-13 14:44:00 +08:00
as2252258 0cc1d3d9f7 eee 2024-12-13 14:35:53 +08:00
as2252258 c444de4815 eee 2024-12-13 14:35:04 +08:00
as2252258 263f99c8c3 eee 2024-12-13 14:34:06 +08:00
as2252258 450870ea19 eee 2024-12-13 14:08:04 +08:00
as2252258 b3e3e95081 eee 2024-12-13 14:04:58 +08:00
as2252258 f084c171ff ea 2024-12-12 12:10:19 +08:00
as2252258 7ca732f153 eee 2024-11-18 16:39:59 +08:00
as2252258 f9771fed3f eee 2024-11-18 16:38:09 +08:00
as2252258 8d006568ef eee 2024-11-18 16:35:37 +08:00
as2252258 6304684683 eee 2024-11-06 21:41:43 +08:00
as2252258 528f43c7ac eee 2024-11-06 21:33:36 +08:00
as2252258 622176c3ff eee 2024-11-06 21:28:29 +08:00
as2252258 e343595e7d eee 2024-11-06 21:16:30 +08:00
as2252258 4f31065818 eee 2024-11-06 21:15:11 +08:00
as2252258 4fe1558a7e eee 2024-11-06 21:13:12 +08:00
as2252258 904ba8cc97 eee 2024-11-06 21:11:43 +08:00
as2252258 4e59053d5c eee 2024-11-06 21:06:01 +08:00
as2252258 59c9a2e944 eee 2024-11-06 21:03:11 +08:00
as2252258 f563126cd0 eee 2024-11-06 20:59:30 +08:00
as2252258 41166dd998 eee 2024-11-06 20:53:24 +08:00
as2252258 3d4327a92e eee 2024-11-06 20:52:17 +08:00
as2252258 28beb3678e eee 2024-11-06 20:36:59 +08:00
as2252258 32164f66ab eee 2024-10-24 10:53:00 +08:00
as2252258 3244f8cfac eee 2024-10-23 14:54:04 +08:00
as2252258 8027effa8c eee 2024-10-23 14:51:09 +08:00
as2252258 59375e567e eee 2024-10-23 14:36:55 +08:00
as2252258 eb39acb7e9 eee 2024-10-23 14:36:20 +08:00
as2252258 be3fc004d5 eee 2024-10-23 14:32:09 +08:00
as2252258 bbe0631f5b eee 2024-10-23 14:10:14 +08:00
as2252258 bf5d988ba4 eee 2024-10-11 11:55:32 +08:00
as2252258 4fc9b42540 eee 2024-10-09 16:01:22 +08:00
as2252258 e5453807f2 eee 2024-09-05 15:42:24 +08:00
as2252258 5a6f6da70a eee 2024-09-02 12:12:39 +08:00
as2252258 beb48f41d2 eee 2024-08-27 11:19:22 +08:00
as2252258 8d917b0f92 eee 2024-08-08 11:12:09 +08:00
as2252258 08fbc49091 eee 2024-08-08 10:52:48 +08:00
as2252258 08e49cfa65 eee 2024-07-10 16:18:27 +08:00
as2252258 d58850b158 eee 2024-07-08 12:06:41 +08:00
as2252258 aba45e8cb9 eee 2024-07-08 10:58:52 +08:00
as2252258 ed22201bef eee 2024-07-02 21:38:30 +08:00
as2252258 ccbac52a16 eee 2024-05-01 02:06:14 +08:00
as2252258 9fe0698d1c eee 2024-05-01 02:02:58 +08:00
as2252258 55e1f6235e eee 2024-05-01 01:54:38 +08:00
as2252258 ed0d044223 eee 2024-04-29 21:55:31 +08:00
as2252258 2b85cb4ec3 eee 2024-04-29 21:55:19 +08:00
as2252258 73b9923740 eee 2024-04-29 21:49:08 +08:00
as2252258 b2c8160314 eee 2024-04-26 17:01:25 +08:00
as2252258 199312d326 eee 2024-04-26 15:39:13 +08:00
as2252258 87f568bdfe eee 2024-04-26 15:36:55 +08:00
as2252258 2fd37f90a3 eee 2024-04-26 15:28:40 +08:00
as2252258 c100190155 eee 2024-04-26 15:27:11 +08:00
as2252258 71766ee914 eee 2024-04-26 15:26:10 +08:00
as2252258 3a39eaabf4 eee 2024-04-26 15:25:12 +08:00
as2252258 722b286f91 eee 2024-04-26 15:23:37 +08:00
as2252258 f9ee3aa014 eee 2024-04-26 15:22:16 +08:00
as2252258 4341efcb8c eee 2024-04-26 15:20:22 +08:00
as2252258 8acf74c9ed eee 2024-04-26 15:15:36 +08:00
as2252258 b9750d743b eee 2024-04-26 15:12:31 +08:00
as2252258 ea34371652 eee 2024-04-16 17:24:22 +08:00
17 changed files with 2384 additions and 1892 deletions
+244 -180
View File
@@ -22,208 +22,272 @@ class ActiveQuery extends QueryTrait implements ISqlBuilder
{
public bool $asArray = FALSE;
protected mixed $_mock = null;
public bool $asArray = FALSE;
protected mixed $_mock = NULL;
/**
* @param bool $asArray
* @return static
*/
public function asArray(bool $asArray = true): static
{
$this->asArray = $asArray;
return $this;
}
/**
* @param bool $asArray
*
* @return static
*/
public function asArray(bool $asArray = TRUE): static
{
$this->asArray = $asArray;
return $this;
}
/**
* @param array $methods
* @return $this
*/
public function with(array $methods): static
{
$this->modelClass->setWith($methods);
return $this;
}
/**
* @param array $methods
*
* @return $this
*/
public function with(array $methods): static
{
$this->modelClass->setWith($methods);
return $this;
}
/**
* @return ModelInterface|array|null|bool
* @throws
*/
public function first(): ModelInterface|null|array|bool
{
$data = $this->buildCommand($this->builder->one())->one();
if (is_array($data)) {
return $this->populate($data);
}
return null;
}
/**
* @return ModelInterface|array|null|bool
* @throws
*/
public function first(): ModelInterface|null|array|bool
{
$data = $this->buildCommand($this->builder->one())->one();
if (is_array($data)) {
return $this->populate($data);
}
return NULL;
}
/**
* @return bool|Collection
*/
public function get(): bool|Collection
{
$data = $this->buildCommand($this->builder->all())->all();
if (is_array($data)) {
return new Collection($this, $this->modelClass, $data);
}
return false;
}
/**
* @return bool|Collection
*/
public function get(): bool|Collection
{
$data = $this->buildCommand($this->builder->all())->all();
if (is_array($data)) {
return new Collection($this, $this->modelClass, $data);
}
return FALSE;
}
/**
* @throws
*/
public function flush(): bool
{
return (bool)$this->buildCommand($this->builder->truncate())->exec();
}
/**
* @param string $column
* @param int|float $amount
*
* @return bool
*/
public function increment(string $column, int|float $amount = 1): bool
{
return (bool)$this->buildCommand($this->builder->mathematics([$column => $amount]))->exec();
}
/**
* @param int $size
* @param Closure $closure
* @return void
*/
public function chunk(int $size, Closure $closure): void
{
if ($this->offset === -1) $this->offset = 0;
$data = $this->offset($this->offset)->limit($size)->get();
if (!$data || $data->isEmpty()) {
return;
}
if (Context::inCoroutine()) {
Coroutine::create(fn() => $closure($data));
} else {
call_user_func($closure, $data);
}
$this->offset += $size;
$this->chunk($size, $closure);
}
/**
* @param string $field
* @param string|null $setKey
*
* @return array|null
*/
public function column(string $field, ?string $setKey = null): ?array
{
return $this->get()->column($field, $setKey);
}
/**
* @param array $attributes
*
* @return bool
*/
public function increments(array $attributes): bool
{
return (bool)$this->buildCommand($this->builder->mathematics($attributes))->exec();
}
/**
* @param mixed $value
* @return $this
*/
public function withMock(mixed $value): static
{
$this->_mock = $value;
return $this;
}
/**
* @param string $column
* @param int|float $amount
*
* @return bool
*/
public function decrement(string $column, int|float $amount = 1): bool
{
return (bool)$this->buildCommand($this->builder->mathematics([$column => $amount], '-'))->exec();
}
/**
* @return mixed
*/
public function mock(): mixed
{
return $this->_mock;
}
/**
* @param array $attributes
*
* @return bool
*/
public function decrements(array $attributes): bool
{
return (bool)$this->buildCommand($this->builder->mathematics($attributes, '-'))->exec();
}
/**
* @param array $data
* @return ModelInterface|array
* @throws
*/
public function populate(array $data): ModelInterface|array
{
$model = $this->modelClass->populates($data);
return $this->asArray ? $model->toArray() : $model;
}
/**
* @throws
*/
public function flush(): bool
{
return (bool)$this->buildCommand($this->builder->truncate())->exec();
}
/**
* @return int
* @throws
*/
public function count(): int
{
return $this->buildCommand($this->builder->count())->rowCount();
}
/**
* @param int $size
* @param Closure $closure
*
* @return void
*/
public function chunk(int $size, Closure $closure): void
{
if ($this->offset === -1) $this->offset = 0;
$data = $this->offset($this->offset)->limit($size)->get();
if (!$data || $data->isEmpty()) {
return;
}
if (Context::inCoroutine()) {
Coroutine::create(fn() => $closure($data));
} else {
call_user_func($closure, $data);
}
$this->offset += $size;
$this->chunk($size, $closure);
}
/**
* @param string $field
* @param string|null $setKey
*
* @return array|null
*/
public function column(string $field, ?string $setKey = NULL): ?array
{
return $this->get()->column($field, $setKey);
}
/**
* @param array $data
* @return bool
* @throws
*/
public function update(array $data): bool
{
if (count($data) < 1) {
return true;
}
$generate = $this->builder->update($data);
if (!is_bool($generate)) {
return (bool)$this->buildCommand($generate)->exec();
} else {
return $generate;
}
}
/**
* @param array $data
* @return bool
*/
public function insert(array $data): bool
{
$sql = $this->builder->insert($data, isset($data[0]));
return (bool)$this->buildCommand($sql)->exec();
}
/**
* @param string $filed
*
* @return mixed
* @throws
*/
public function value(string $filed): mixed
{
return $this->first()[$filed] ?? NULL;
}
/**
* @return bool
* @throws
*/
public function exists(): bool
{
return $this->buildCommand($this->builder->one())->rowCount() > 0;
}
/**
* @param mixed $value
*
* @return $this
*/
public function withMock(mixed $value): static
{
$this->_mock = $value;
return $this;
}
/**
* @param string $sql
* @param array $params
* @return int|bool
*/
public function execute(string $sql, array $params = []): int|bool
{
return $this->buildCommand($sql, $params)->exec();
}
/**
* @return mixed
*/
public function mock(): mixed
{
return $this->_mock;
}
/**
* @return bool
*/
public function delete(): bool
{
return $this->buildCommand($this->builder->delete())->delete();
}
/**
* @param array $data
*
* @return ModelInterface|array
* @throws
*/
public function populate(array $data): ModelInterface|array
{
$model = $this->modelClass->populates($data);
return $this->asArray ? $model->toArray() : $model;
}
/**
* @return string
*/
public function toSql(): string
{
return $this->builder->get();
}
/**
* @return int
* @throws
*/
public function count(): int
{
return $this->buildCommand($this->builder->count())->rowCount();
}
/**
* @param array $data
*
* @return bool
* @throws
*/
public function update(array $data): bool
{
if (count($data) < 1) {
return TRUE;
}
$generate = $this->builder->update($data);
if (!is_bool($generate)) {
return (bool)$this->buildCommand($generate)->exec();
} else {
return $generate;
}
}
/**
* @param array $data
*
* @return bool
*/
public function insert(array $data): bool
{
$sql = $this->builder->insert($data, isset($data[0]));
return (bool)$this->buildCommand($sql)->exec();
}
/**
* @param string $filed
*
* @return mixed
* @throws
*/
public function value(string $filed): mixed
{
return $this->first()[$filed] ?? NULL;
}
/**
* @return bool
* @throws
*/
public function exists(): bool
{
return $this->buildCommand($this->limit(1)->builder->exists())->exists();
}
/**
* @param string $sql
* @param array $params
*
* @return int|bool
*/
public function execute(string $sql, array $params = []): int|bool
{
return $this->buildCommand($sql, $params)->exec();
}
/**
* @return bool
*/
public function delete(): bool
{
return $this->buildCommand($this->builder->delete())->delete();
}
}
+3 -2
View File
@@ -221,10 +221,11 @@ interface ActiveQueryInterface
public function whereNotBetween(string $column, int|float $start, int|float $end): QueryTrait;
/**
* @param array $column
* @param array|string $column
* @param mixed|null $value
* @return QueryTrait
*/
public function where(array $column): QueryTrait;
public function where(array|string $column, mixed $value = null): QueryTrait;
/**
+829 -776
View File
File diff suppressed because it is too large Load Diff
+28
View File
@@ -0,0 +1,28 @@
<?php
namespace Database\Base;
class PDO extends \PDO
{
/**
* @param string $database
* @param string $host
* @param string $username
* @param string $password
* @param array $options
*/
public function __construct(
readonly public string $database,
readonly public string $host,
readonly public string $username,
readonly public string $password,
readonly public array $options,
)
{
parent::__construct('mysql:dbname=' . $this->database . ';host=' . $this->host, $this->username, $this->password, $this->options);
}
}
+20 -50
View File
@@ -47,15 +47,6 @@ class Command extends Component
return (bool)$this->_prepare();
}
/**
* @return bool
* @throws
*/
public function save(): bool
{
return (bool)$this->_prepare();
}
/**
* @return bool|array
@@ -75,6 +66,19 @@ class Command extends Component
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
@@ -88,46 +92,11 @@ class Command extends Component
* @return mixed
* @throws
*/
public function rowCount(): mixed
public function rowCount(): int
{
$client = $this->connection->getConnection();
try {
$client->setAttribute(PDO::MYSQL_ATTR_USE_BUFFERED_QUERY, true);
if (($prepare = $client->query($this->sql)) === false) {
throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $client->errorInfo()[2]);
}
$data = $this->search('fetch');
$prepare->execute($this->params);
$count = $prepare->rowCount();
$prepare->closeCursor();
$this->connection->println($this->sql, $this->params);
return $count;
} catch (Throwable $throwable) {
if ($this->isRefresh($throwable)) return $this->rowCount();
$errorMsg = $throwable->getMessage() . PHP_EOL . ' Sql: ' . $this->sql . '.' . json_encode($this->params);
$this->getLogger()->failure($errorMsg . PHP_EOL, 'mysql');
return 0;
} finally {
$this->connection->release($client);
}
}
/**
* @return bool
* @throws Exception
*/
public function exists(): bool
{
$total = $this->search('rowCount');
if ($total === false) {
throw new Exception('Query data is has error.');
}
return $total > 0;
return !$data ? 0 : +current($data);
}
@@ -140,6 +109,7 @@ class Command extends Component
{
$client = $this->connection->getConnection();
try {
$startTime = microtime(true);
if (($prepare = $client->prepare($this->sql)) === false) {
throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $client->errorInfo()[2]);
}
@@ -149,7 +119,7 @@ class Command extends Component
$result = $method == 'rowCount' ? $prepare->rowCount() : $prepare->{$method}(PDO::FETCH_ASSOC);
$prepare->closeCursor();
$this->connection->println($this->sql, $this->params);
$this->connection->println($startTime, microtime(true), $this->sql, $this->params);
return $result;
} catch (Throwable $throwable) {
@@ -182,7 +152,7 @@ class Command extends Component
{
$client = $this->connection->getConnection();
try {
$this->connection->println($this->sql, $this->params);
$startTime = microtime(true);
if (($prepare = $client->prepare($this->sql)) === false) {
throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $prepare->errorInfo()[2]);
}
@@ -193,7 +163,7 @@ class Command extends Component
$result = $client->lastInsertId();
$this->connection->println($this->sql, $this->params);
$this->connection->println($startTime, microtime(true), $this->sql, $this->params);
return $result == 0 ? $prepare->rowCount() : (int)$result;
} catch (Throwable $throwable) {
+71 -50
View File
@@ -21,7 +21,6 @@ use Kiri\Abstracts\Component;
use Kiri\Di\Context;
use Kiri\Pool\Pool;
use Kiri\Events\EventProvider;
use PDO;
use Kiri\Error\StdoutLogger;
use Psr\Log\LoggerInterface;
use Kiri\Server\Events\OnWorkerStart;
@@ -29,6 +28,9 @@ use Kiri\Server\Events\OnTaskerStart;
use Kiri\Server\Events\OnAfterRequest;
use Kiri\Di\Inject\Container;
use Swoole\Timer;
use Database\Base\PDO;
//use PDO;
/**
* Class Connection
@@ -37,25 +39,26 @@ use Swoole\Timer;
class Connection extends Component
{
public string $id = 'db';
public string $cds = '';
public string $password = '';
public string $username = '';
public string $charset = 'utf-8';
public string $tablePrefix = '';
public string $database = '';
public int $timeout = 30;
public int $waite_time = 3;
public int $tick_time = 60;
public int $idle_count = 3;
public int $idle_time = 60;
public array $pool = ['max' => 10, 'min' => 1];
private int $storey = 0;
protected int $timerId = -1;
public bool $enableCache = false;
public string $cacheDriver = 'redis';
public array $attributes = [];
protected \Closure $_println;
public string $id = 'db';
public string $cds = '';
public string $password = '';
public string $username = '';
public string $charset = 'utf-8';
public string $tablePrefix = '';
public string $database = '';
public int $timeout = 30;
public int $waite_time = 3;
public int $tick_time = 60;
public int $idle_count = 3;
public int $idle_time = 60;
public array $pool = ['max' => 10, 'min' => 1];
private int $storey = 0;
protected int $timerId = -1;
public bool $enableCache = false;
public string $cacheDriver = 'redis';
public array $attributes = [];
public array $slave = [];
protected ?\Closure $_println = null;
/**
@@ -64,6 +67,14 @@ class Connection extends Component
#[Container(LoggerInterface::class)]
public StdoutLogger $logger;
/**
* @var EventProvider
*/
#[Container(EventProvider::class)]
public EventProvider $eventProvider;
/**
* @param Pool $connections
*/
@@ -76,14 +87,16 @@ class Connection extends Component
/**
* @param float $startTime
* @param float $endTime
* @param string $sql
* @param array $params
* @return void
*/
public function println(string $sql, array $params = []): void
public function println(float $startTime, float $endTime, string $sql, array $params = []): void
{
if (is_callable($this->_println)) {
call_user_func($this->_println, $sql, $params);
call_user_func($this->_println, $startTime, $endTime, $sql, $params);
}
}
@@ -94,14 +107,13 @@ class Connection extends Component
*/
public function init(): void
{
$eventProvider = Kiri::getDi()->get(EventProvider::class);
$eventProvider->on(BeginTransaction::class, [$this, 'beginTransaction'], 0);
$eventProvider->on(Rollback::class, [$this, 'rollback'], 0);
$eventProvider->on(Commit::class, [$this, 'commit'], 0);
$eventProvider->on(OnAfterRequest::class, [$this, 'clear']);
$eventProvider->on(OnWorkerExit::class, [$this, 'disconnect']);
$eventProvider->on(OnWorkerStart::class, [$this, 'tick']);
$eventProvider->on(OnTaskerStart::class, [$this, 'tick']);
$this->eventProvider->on(BeginTransaction::class, [$this, 'beginTransaction'], 0);
$this->eventProvider->on(Rollback::class, [$this, 'rollback'], 0);
$this->eventProvider->on(Commit::class, [$this, 'commit'], 0);
$this->eventProvider->on(OnAfterRequest::class, [$this, 'clear']);
$this->eventProvider->on(OnWorkerExit::class, [$this, 'disconnect']);
$this->eventProvider->on(OnWorkerStart::class, [$this, 'tick']);
$this->eventProvider->on(OnTaskerStart::class, [$this, 'tick']);
}
@@ -110,7 +122,7 @@ class Connection extends Component
*/
public function tick(): void
{
$this->timerId = Timer::tick($this->tick_time, fn() => $this->checkClientHealth($this->pool()));
$this->timerId = Timer::tick($this->tick_time, fn () => $this->checkClientHealth($this->pool()));
}
@@ -121,14 +133,14 @@ class Connection extends Component
*/
protected function checkClientHealth(Pool $pool): void
{
$pool->flush($this->cds, $this->pool['min'] ?? 1);
$length = $pool->size($this->cds);
$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->cds, $client);
$pool->push($this->getName(), $client);
} catch (\Throwable $exception) {
if (!str_contains($exception->getMessage(), 'Client timeout.')) {
$this->logger->error(throwable($exception), [$this->cds]);
@@ -138,6 +150,15 @@ class Connection extends Component
}
/**
* @return string
*/
private function getName(): string
{
return 'mysql.' . $this->cds;
}
/**
* @param Pool $pool
* @return PDO|bool
@@ -146,7 +167,7 @@ class Connection extends Component
protected function validator(Pool $pool): PDO|bool
{
/** @var $client PDO */
if (($client = $pool->get($this->cds)) === false) {
if (($client = $pool->get($this->getName())) === false) {
return false;
}
if ($client->query('select 1') === false) {
@@ -176,7 +197,7 @@ class Connection extends Component
*/
protected function getNormalClientHealth(): PDO
{
$data = $this->pool()->get($this->cds, $this->waite_time);
$data = $this->pool()->get($this->getName(), $this->waite_time);
if ($data === false) {
throw new Exception('Client Waite timeout.');
}
@@ -302,7 +323,7 @@ class Connection extends Component
public function release(PDO $pdo): void
{
if (!$this->inTransaction()) {
$this->pool()->push($this->cds, $pdo);
$this->pool()->push($this->getName(), $pdo);
}
}
@@ -314,7 +335,7 @@ class Connection extends Component
*/
public function clear_connection(): void
{
$this->pool()->flush($this->cds, 0);
$this->pool()->flush($this->getName(), 0);
}
@@ -326,23 +347,23 @@ class Connection extends Component
if ($this->timerId > -1) {
Timer::clear($this->timerId);
}
$this->pool()->close($this->cds);
$this->pool()->close($this->getName());
}
/**
* @return PDO
*/
public function newConnect(): PDO
public function newConnect(): \PDO
{
$pdo = new PDO('mysql:dbname=' . $this->database . ';host=' . $this->cds, $this->username, $this->password, [
PDO::ATTR_CASE => PDO::CASE_NATURAL,
PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
PDO::ATTR_ORACLE_NULLS => PDO::NULL_NATURAL,
PDO::ATTR_STRINGIFY_FETCHES => false,
PDO::ATTR_EMULATE_PREPARES => true,
PDO::ATTR_TIMEOUT => $this->timeout,
PDO::MYSQL_ATTR_INIT_COMMAND => 'SET NAMES ' . $this->charset
$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
]);
foreach ($this->attributes as $key => $attribute) {
$pdo->setAttribute($key, $attribute);
@@ -356,8 +377,8 @@ class Connection extends Component
*/
protected function pool(): Pool
{
if (!$this->connections->hasChannel($this->cds)) {
$this->connections->created($this->cds, $this->pool['max'] ?? 1, [$this, 'newConnect']);
if (!$this->connections->hasChannel($this->getName())) {
$this->connections->created($this->getName(), $this->pool['max'] ?? 1, [$this, 'newConnect']);
}
return $this->connections;
}
+51 -1
View File
@@ -115,10 +115,11 @@ class Db extends QueryTrait implements ISqlBuilder
/**
* @return bool
* @throws Exception
*/
public function exists(): bool
{
return $this->connection->createCommand(SqlBuilder::builder($this)->one())->rowCount() > 0;
return $this->connection->createCommand(SqlBuilder::builder($this->limit(1))->exists())->exists();
}
/**
@@ -151,6 +152,55 @@ class Db extends QueryTrait implements ISqlBuilder
return $this->connection->createCommand(SqlBuilder::builder($this)->delete())->delete();
}
/**
* @return bool|array|null
*/
public function first(): bool|array|null
{
return $this->connection->createCommand(SqlBuilder::builder($this)->one())->one();
}
/**
* @return bool|array
*/
public function get(): bool|array
{
return $this->connection->createCommand(SqlBuilder::builder($this)->all())->all();
}
/**
* @param string $sql
* @return mixed
*/
public function exec(string $sql): mixed
{
return $this->connection->createCommand($sql)->exec();
}
/**
* @param string $sql
* @return array|bool|null
*/
public function query(string $sql): array|bool|null
{
return $this->connection->createCommand($sql)->one();
}
/**
* @param string $sql
* @return array|bool
*/
public function queryAll(string $sql): array|bool
{
return $this->connection->createCommand($sql)->all();
}
/**
* @param string $table
* @return array|bool|null
+2 -3
View File
@@ -4,8 +4,8 @@ declare(strict_types=1);
namespace Database;
use Database\Traits\HasBase;
use Exception;
use Kiri;
use Kiri\Di\Context;
/**
* Class HasCount
@@ -20,8 +20,7 @@ class HasCount extends HasBase
*/
public function get(): array|ModelInterface|null
{
$relation = Kiri::getDi()->get(Relation::class);
return $relation->get($this->name);
return di(Relation::class)->get($this->name);
}
}
+2 -2
View File
@@ -12,6 +12,7 @@ namespace Database;
use Database\Traits\HasBase;
use Exception;
use Kiri;
use Kiri\Di\Context;
/**
* Class HasMany
@@ -28,7 +29,6 @@ class HasMany extends HasBase
*/
public function get(): array|Collection|null
{
$relation = Kiri::getDi()->get(Relation::class);
return $relation->get($this->name);
return di(Relation::class)->get($this->name);
}
}
+2 -2
View File
@@ -12,6 +12,7 @@ namespace Database;
use Database\Traits\HasBase;
use Exception;
use Kiri;
use Kiri\Di\Context;
/**
* Class HasOne
@@ -27,7 +28,6 @@ class HasOne extends HasBase
*/
public function get(): array|ModelInterface|null
{
$relation = Kiri::getDi()->get(Relation::class);
return $relation->first($this->name);
return di(Relation::class)->first($this->name);
}
}
+14 -14
View File
@@ -266,21 +266,21 @@ class Model extends Base\Model
/**
* @param ModelInterface|string $modelName
* @param string $modelName
* @param string $foreignKey
* @param string $localKey
* @return string
* @throws Exception
*/
private function _hasBase(ModelInterface|string $modelName, string $foreignKey, string $localKey): string
private function _hasBase(string $modelName, string $foreignKey, string $localKey): string
{
if (($value = $this->{$localKey}) === null) {
throw new Exception("Need join table primary key.");
}
$relation = $this->getRelation();
$relation = di(Relation::class);
$primaryKey = $modelName . $foreignKey . $value;
$primaryKey = str_replace('\\', '_', $modelName) . '_' . $foreignKey . '_' . $value;
if (!$relation->hasIdentification($primaryKey)) {
$relation->bindIdentification($primaryKey, $modelName::query()->where([$foreignKey => $value]));
}
@@ -289,59 +289,59 @@ class Model extends Base\Model
/**
* @param ModelInterface|string $modelName
* @param string $modelName
* @param string $foreignKey
* @param string $localKey
* @return HasOne|ActiveQuery
* @throws Exception
*/
public function hasOne(ModelInterface|string $modelName, string $foreignKey, string $localKey): HasOne|ActiveQuery
public function hasOne(string $modelName, string $foreignKey, string $localKey): HasOne|ActiveQuery
{
return new HasOne($this->_hasBase($modelName, $foreignKey, $localKey));
}
/**
* @param ModelInterface|string $modelName
* @param string $modelName
* @param string $foreignKey
* @param string $localKey
* @return ActiveQuery|HasCount
* @throws Exception
*/
public function hasCount(ModelInterface|string $modelName, string $foreignKey, string $localKey): ActiveQuery|HasCount
public function hasCount(string $modelName, string $foreignKey, string $localKey): ActiveQuery|HasCount
{
return new HasCount($this->_hasBase($modelName, $foreignKey, $localKey));
}
/**
* @param ModelInterface|string $modelName
* @param string $modelName
* @param string $foreignKey
* @param string $localKey
* @return ActiveQuery|HasMany
* @throws Exception
*/
public function hasMany(ModelInterface|string $modelName, string $foreignKey, string $localKey): ActiveQuery|HasMany
public function hasMany(string $modelName, string $foreignKey, string $localKey): ActiveQuery|HasMany
{
return new HasMany($this->_hasBase($modelName, $foreignKey, $localKey));
}
/**
* @param ModelInterface|string $modelName
* @param string $modelName
* @param string $foreignKey
* @param string $localKey
* @return ActiveQuery|HasMany
* @throws Exception
*/
public function hasIn(ModelInterface|string $modelName, string $foreignKey, string $localKey): ActiveQuery|HasMany
public function hasIn(string $modelName, string $foreignKey, string $localKey): ActiveQuery|HasMany
{
if (($value = $this->{$localKey}) === null) {
throw new Exception("Need join table primary key.");
}
$relation = $this->getRelation();
$relation = di(Relation::class);
$primaryKey = $modelName . $foreignKey . json_encode($value, JSON_UNESCAPED_UNICODE);
$primaryKey = str_replace('\\', '_', $modelName) . '_' . $foreignKey . '_' . implode('_', $value);
if (!$relation->hasIdentification($primaryKey)) {
$relation->bindIdentification($primaryKey, $modelName::query()->whereIn($foreignKey, $value));
}
+2 -2
View File
@@ -18,10 +18,10 @@ interface ModelInterface
{
/**
* @param array|string|int $param
* @param array|string|int|null $param
* @return ModelInterface|null
*/
public static function findOne(array|string|int $param): ?static;
public static function findOne(array|string|int|null $param): ?static;
/**
+11 -15
View File
@@ -4,8 +4,10 @@ declare(strict_types=1);
namespace Database;
use Database\Base\ActiveQueryInterface;
use Kiri\Abstracts\Component;
use Kiri\Di\Context;
use Swoole\Coroutine;
/**
* Class Relation
@@ -34,7 +36,7 @@ class Relation extends Component
*/
public function hasIdentification(string $identification): bool
{
return isset($this->_query[$identification]) && $this->_query[$identification] instanceof ActiveQuery;
return isset($this->_query[$identification]);
}
/**
@@ -54,12 +56,10 @@ class Relation extends Component
*/
public function first(string $_identification): mixed
{
if (Context::exists($_identification)) {
return Context::get($_identification);
if (!Context::exists($_identification)) {
Context::set($_identification, $this->_query[$_identification]->first());
}
$activeModel = $this->_query[$_identification]->first();
unset($this->_query[$_identification]);
return Context::set($_identification, $activeModel);
return Context::get($_identification);
}
@@ -69,12 +69,10 @@ class Relation extends Component
*/
public function count(string $_identification): mixed
{
if (Context::exists($_identification)) {
return Context::get($_identification);
if (!Context::exists($_identification)) {
Context::set($_identification, $this->_query[$_identification]->count());
}
$activeModel = $this->_query[$_identification]->count();
unset($this->_query[$_identification]);
return Context::set($_identification, $activeModel);
return Context::get($_identification);
}
@@ -86,11 +84,9 @@ class Relation extends Component
public function get(string $_identification): mixed
{
if (Context::exists($_identification)) {
return Context::get($_identification);
Context::set($_identification, $this->_query[$_identification]->get());
}
$activeModel = $this->_query[$_identification]->get();
unset($this->_query[$_identification]);
return Context::set($_identification, $activeModel);
return Context::get($_identification);
}
}
+34 -34
View File
@@ -51,7 +51,7 @@ class SqlBuilder extends Component
*/
public function getCondition(): string
{
return $this->where($this->query->where);
return $this->where($this->query->getWhere());
}
@@ -73,9 +73,9 @@ class SqlBuilder extends Component
*/
public function update(array $attributes): bool|string
{
$conditions = $this->query->params;
$this->query->params = [];
$data = $this->__updateBuilder($this->makeParams($attributes));
$conditions = $this->query->getParams();
$this->query->setParams([]);
$data = $this->__updateBuilder($this->makeParams($attributes));
foreach ($conditions as $condition) {
$this->query->pushParam($condition);
}
@@ -109,7 +109,7 @@ class SqlBuilder extends Component
if (empty($string)) {
return Kiri::getLogger()->failure('None data update.');
}
return 'UPDATE ' . $this->query->from . ' SET ' . implode(',', $string) . $this->make();
return 'UPDATE ' . $this->query->getFrom() . ' SET ' . implode(',', $string) . $this->make();
}
@@ -121,7 +121,7 @@ class SqlBuilder extends Component
*/
public function insert(array $attributes, bool $isBatch = false): string
{
$update = 'INSERT INTO ' . $this->query->from;
$update = 'INSERT INTO ' . $this->query->getFrom();
if ($isBatch === false) {
$attributes = [$attributes];
}
@@ -143,7 +143,7 @@ class SqlBuilder extends Component
*/
public function delete(): string
{
return 'DELETE FROM ' . $this->query->from . $this->make();
return 'DELETE FROM ' . $this->query->getFrom() . $this->make();
}
@@ -189,7 +189,7 @@ class SqlBuilder extends Component
if (is_null($value)) {
return $keys;
}
if (is_string($value) && $this->isMath($value)) {
if (preg_match('/^[+|-]\s\d+$/', $value)) {
$keys[] = $key . '=' . $key . ' ' . $value;
} else {
$this->query->pushParam($value);
@@ -199,23 +199,13 @@ class SqlBuilder extends Component
}
/**
* @param string $value
* @return bool
*/
private function isMath(string $value): bool
{
return str_starts_with($value, '+ ') || str_starts_with($value, '- ');
}
/**
* @return string
* @throws
*/
public function one(): string
{
return $this->makeSelect($this->query->select) . $this->make() . $this->makeLimit($this->query->limit(1));
return $this->makeSelect($this->query->getSelect()) . $this->make() . $this->makeLimit($this->query->limit(1));
}
@@ -225,7 +215,7 @@ class SqlBuilder extends Component
*/
public function all(): string
{
return $this->makeSelect($this->query->select) . $this->make() . $this->makeLimit($this->query);
return $this->makeSelect($this->query->getSelect()) . $this->make() . $this->makeLimit($this->query);
}
@@ -235,7 +225,17 @@ class SqlBuilder extends Component
*/
public function count(): string
{
return $this->makeSelect() . $this->make();
return $this->makeSelect(['COUNT(*)']) . $this->make();
}
/**
* @return string
* @throws
*/
public function exists(): string
{
return $this->makeSelect(['0']) . $this->make();
}
@@ -267,12 +267,12 @@ class SqlBuilder extends Component
*/
private function makeSelect(array $select = ['*']): string
{
$select = "SELECT " . implode(',', $select) . " FROM " . $this->query->from;
if ($this->query->alias != "") {
$select .= " AS " . $this->query->alias;
$select = "SELECT " . implode(',', $select) . " FROM " . $this->query->getFrom();
if ($this->query->getAlias() != "") {
$select .= " AS " . $this->query->getAlias();
}
if (count($this->query->join) > 0) {
$select .= ' ' . implode(' ', $this->query->join);
if (count($this->query->getJoin()) > 0) {
$select .= ' ' . implode(' ', $this->query->getJoin());
}
return $select;
}
@@ -283,8 +283,8 @@ class SqlBuilder extends Component
*/
private function makeGroup(): string
{
if ($this->query->group != "") {
return ' GROUP BY ' . $this->query->group;
if ($this->query->getGroup() != "") {
return ' GROUP BY ' . $this->query->getGroup();
}
return '';
}
@@ -295,8 +295,8 @@ class SqlBuilder extends Component
*/
private function makeOrder(): string
{
if (count($this->query->order) > 0) {
return ' ORDER BY ' . implode(',', $this->query->order);
if (count($this->query->getOrder()) > 0) {
return ' ORDER BY ' . implode(',', $this->query->getOrder());
}
return '';
}
@@ -307,7 +307,7 @@ class SqlBuilder extends Component
*/
private function makeCondition(): string
{
$condition = $this->where($this->query->where);
$condition = $this->where($this->query->getWhere());
if (empty($condition)) {
return '';
}
@@ -317,8 +317,8 @@ class SqlBuilder extends Component
private function makeLimit(): string
{
if ($this->query->offset >= 0 && $this->query->limit >= 1) {
return ' LIMIT ' . $this->query->offset . ',' . $this->query->limit;
if ($this->query->getOffset() >= 0 && $this->query->getLimit() >= 1) {
return ' LIMIT ' . $this->query->getOffset() . ',' . $this->query->getLimit();
}
return '';
}
@@ -344,7 +344,7 @@ class SqlBuilder extends Component
*/
public function truncate(): string
{
return sprintf('TRUNCATE %s', $this->query->from);
return sprintf('TRUNCATE %s', $this->query->getFrom());
}
+7 -2
View File
@@ -13,6 +13,7 @@ use Database\ModelInterface;
use Database\Collection;
use Database\Relation;
use Kiri;
use Kiri\Di\Context;
/**
* Class HasBase
@@ -52,12 +53,16 @@ abstract class HasBase implements \Database\Traits\Relation
* @param string $name
* @param array $arguments
* @return $this|mixed
* @throws \Exception
*/
public function __call(string $name, array $arguments)
{
if ($name !== 'get') {
$relation = Kiri::getDi()->get(Relation::class);
$relation->getQuery($this->name)->$name(...$arguments);
$query = di(Relation::class)->getQuery($this->name);
if (is_null($query)) {
throw new \Exception('Unknown relation key: ' . $this->name);
}
$query->$name(...$arguments);
return $this;
} else {
return $this->get();
+1041 -759
View File
File diff suppressed because it is too large Load Diff
+23
View File
@@ -0,0 +1,23 @@
<?php
use Database\Query;
class Users extends \Database\Model
{
public function hasD()
{
return $this->hasOne(static::class, 'id', 'id')->with([]);
}
}
Users::query()
->select(['*', (new Query(Users::class))
->where(['id' => 2])])
->from(function (Query $query) {
$query->from(Users::class)
->where(['id' => 1])
->groupBy('name DESC');
})->toSql();