Compare commits

...

28 Commits

Author SHA1 Message Date
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
12 changed files with 142 additions and 69 deletions
+43 -1
View File
@@ -73,6 +73,48 @@ class ActiveQuery extends QueryTrait implements ISqlBuilder
} }
/**
* @param string $column
* @param int|float $amount
* @return bool
*/
public function increment(string $column, int|float $amount = 1): bool
{
return (bool)$this->buildCommand($this->builder->mathematics([$column => $amount]))->exec();
}
/**
* @param array $attributes
* @return bool
*/
public function increments(array $attributes): bool
{
return (bool)$this->buildCommand($this->builder->mathematics($attributes))->exec();
}
/**
* @param string $column
* @param int|float $amount
* @return bool
*/
public function decrement(string $column, int|float $amount = 1): bool
{
return (bool)$this->buildCommand($this->builder->mathematics([$column => $amount], '-'))->exec();
}
/**
* @param array $attributes
* @return bool
*/
public function decrements(array $attributes): bool
{
return (bool)$this->buildCommand($this->builder->mathematics($attributes, '-'))->exec();
}
/** /**
* @throws * @throws
*/ */
@@ -95,7 +137,7 @@ class ActiveQuery extends QueryTrait implements ISqlBuilder
return; return;
} }
if (Context::inCoroutine()) { if (Context::inCoroutine()) {
Coroutine::create(fn() => $closure($data)); Coroutine::create(fn () => $closure($data));
} else { } else {
call_user_func($closure, $data); call_user_func($closure, $data);
} }
+22 -14
View File
@@ -1,4 +1,5 @@
<?php /** @noinspection ALL */ <?php
/** @noinspection ALL */
/** /**
* Created by PhpStorm. * Created by PhpStorm.
* User: whwyy * User: whwyy
@@ -22,6 +23,7 @@ use Database\ModelInterface;
use Database\Relation; use Database\Relation;
use Database\SqlBuilder; use Database\SqlBuilder;
use Exception; use Exception;
use Kiri\Di\Context;
use Kiri; use Kiri;
use Kiri\Abstracts\Component; use Kiri\Abstracts\Component;
use ReturnTypeWillChange; use ReturnTypeWillChange;
@@ -222,8 +224,8 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, \
*/ */
public function getPrimaryValue(): ?int public function getPrimaryValue(): ?int
{ {
if ($this->hasPrimary()) { if ($this->hasPrimary() && isset($this->_oldAttributes[$this->getPrimary()])) {
return (int)$this->_oldAttributes[$this->getPrimary()] ?? null; return (int)$this->_oldAttributes[$this->getPrimary()];
} else { } else {
return null; return null;
} }
@@ -234,23 +236,25 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, \
* @return Model|null * @return Model|null
* @throws * @throws
*/ */
public static function findOne(int|string|array $param): ?static public static function findOne(int|string|array|null $param): ?static
{ {
$model = static::instance(); if (empty($param)) {
$query = new ActiveQuery($model); return null;
}
$query = new ActiveQuery($model = static::instance());
$query->from($model->getTable())->alias('t1'); $query->from($model->getTable())->alias('t1');
if (is_numeric($param)) { if (is_numeric($param)) {
$query->where([$model->getPrimary() => $param]); $query->where([$model->getPrimary() => +$param]);
} else if (is_array($param)) { } else if (is_array($param)) {
$query->where($param); $query->where($param);
} else { } else {
$query->whereRaw($param); $query->whereRaw($param);
} }
$data = $query->first(); if (($data = $query->first()) === false) {
if ($data === false) {
throw new Exception($model->getLastError()); throw new Exception($model->getLastError());
} else {
return $data;
} }
return $data;
} }
@@ -440,10 +444,13 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, \
if ($lastId === false) { if ($lastId === false) {
return false; return false;
} }
if ($this->hasPrimary()) { if (!$this->hasPrimary()) {
$this->_attributes[$this->getPrimary()] = $lastId; return $this->refresh()->afterSave($this->_attributes, []);
} }
return $this;
$this->_attributes[$this->getPrimary()] = $lastId;
return $this->refresh()->afterSave($this->_attributes, [$this->getPrimary() => $lastId]);
} }
@@ -591,7 +598,7 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, \
*/ */
public function getRelation(): ?Relation public function getRelation(): ?Relation
{ {
return Kiri::getDi()->get(Relation::class); return di(Relation::class);
} }
@@ -653,6 +660,7 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, \
public function refresh(): static public function refresh(): static
{ {
$this->_oldAttributes = $this->_attributes; $this->_oldAttributes = $this->_attributes;
$this->isNewExample = false;
return $this; return $this;
} }
+4 -2
View File
@@ -109,6 +109,7 @@ class Command extends Component
{ {
$client = $this->connection->getConnection(); $client = $this->connection->getConnection();
try { try {
$startTime = microtime(true);
if (($prepare = $client->prepare($this->sql)) === false) { if (($prepare = $client->prepare($this->sql)) === false) {
throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $client->errorInfo()[2]); throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $client->errorInfo()[2]);
} }
@@ -118,7 +119,7 @@ class Command extends Component
$result = $method == 'rowCount' ? $prepare->rowCount() : $prepare->{$method}(PDO::FETCH_ASSOC); $result = $method == 'rowCount' ? $prepare->rowCount() : $prepare->{$method}(PDO::FETCH_ASSOC);
$prepare->closeCursor(); $prepare->closeCursor();
$this->connection->println($this->sql, $this->params); $this->connection->println($startTime, microtime(true), $this->sql, $this->params);
return $result; return $result;
} catch (Throwable $throwable) { } catch (Throwable $throwable) {
@@ -151,6 +152,7 @@ class Command extends Component
{ {
$client = $this->connection->getConnection(); $client = $this->connection->getConnection();
try { try {
$startTime = microtime(true);
if (($prepare = $client->prepare($this->sql)) === false) { if (($prepare = $client->prepare($this->sql)) === false) {
throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $prepare->errorInfo()[2]); throw new Exception('(' . $prepare->errorInfo()[0] . ')' . $prepare->errorInfo()[2]);
} }
@@ -161,7 +163,7 @@ class Command extends Component
$result = $client->lastInsertId(); $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; return $result == 0 ? $prepare->rowCount() : (int)$result;
} catch (Throwable $throwable) { } catch (Throwable $throwable) {
+20 -12
View File
@@ -67,6 +67,14 @@ class Connection extends Component
#[Container(LoggerInterface::class)] #[Container(LoggerInterface::class)]
public StdoutLogger $logger; public StdoutLogger $logger;
/**
* @var EventProvider
*/
#[Container(EventProvider::class)]
public EventProvider $eventProvider;
/** /**
* @param Pool $connections * @param Pool $connections
*/ */
@@ -79,14 +87,16 @@ class Connection extends Component
/** /**
* @param float $startTime
* @param float $endTime
* @param string $sql * @param string $sql
* @param array $params * @param array $params
* @return void * @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)) { if (is_callable($this->_println)) {
call_user_func($this->_println, $sql, $params); call_user_func($this->_println, $startTime, $endTime, $sql, $params);
} }
} }
@@ -97,14 +107,13 @@ class Connection extends Component
*/ */
public function init(): void public function init(): void
{ {
$eventProvider = Kiri::getDi()->get(EventProvider::class); $this->eventProvider->on(BeginTransaction::class, [$this, 'beginTransaction'], 0);
$eventProvider->on(BeginTransaction::class, [$this, 'beginTransaction'], 0); $this->eventProvider->on(Rollback::class, [$this, 'rollback'], 0);
$eventProvider->on(Rollback::class, [$this, 'rollback'], 0); $this->eventProvider->on(Commit::class, [$this, 'commit'], 0);
$eventProvider->on(Commit::class, [$this, 'commit'], 0); $this->eventProvider->on(OnAfterRequest::class, [$this, 'clear']);
$eventProvider->on(OnAfterRequest::class, [$this, 'clear']); $this->eventProvider->on(OnWorkerExit::class, [$this, 'disconnect']);
$eventProvider->on(OnWorkerExit::class, [$this, 'disconnect']); $this->eventProvider->on(OnWorkerStart::class, [$this, 'tick']);
$eventProvider->on(OnWorkerStart::class, [$this, 'tick']); $this->eventProvider->on(OnTaskerStart::class, [$this, 'tick']);
$eventProvider->on(OnTaskerStart::class, [$this, 'tick']);
} }
@@ -113,7 +122,7 @@ class Connection extends Component
*/ */
public function tick(): void 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()));
} }
@@ -348,7 +357,6 @@ class Connection extends Component
public function newConnect(): \PDO public function newConnect(): \PDO
{ {
$pdo = new PDO($this->database, $this->cds, $this->username, $this->password, [ $pdo = new PDO($this->database, $this->cds, $this->username, $this->password, [
// $pdo = new \PDO('mysql:dbname=' . $this->database . ';host=' . $this->cds, $this->username, $this->password, [
\PDO::ATTR_CASE => \PDO::CASE_NATURAL, \PDO::ATTR_CASE => \PDO::CASE_NATURAL,
\PDO::ATTR_ERRMODE => \PDO::ERRMODE_EXCEPTION, \PDO::ATTR_ERRMODE => \PDO::ERRMODE_EXCEPTION,
\PDO::ATTR_ORACLE_NULLS => \PDO::NULL_NATURAL, \PDO::ATTR_ORACLE_NULLS => \PDO::NULL_NATURAL,
+2 -3
View File
@@ -4,8 +4,8 @@ declare(strict_types=1);
namespace Database; namespace Database;
use Database\Traits\HasBase; use Database\Traits\HasBase;
use Exception;
use Kiri; use Kiri;
use Kiri\Di\Context;
/** /**
* Class HasCount * Class HasCount
@@ -20,8 +20,7 @@ class HasCount extends HasBase
*/ */
public function get(): array|ModelInterface|null public function get(): array|ModelInterface|null
{ {
$relation = Kiri::getDi()->get(Relation::class); return di(Relation::class)->get($this->name);
return $relation->get($this->name);
} }
} }
+2 -2
View File
@@ -12,6 +12,7 @@ namespace Database;
use Database\Traits\HasBase; use Database\Traits\HasBase;
use Exception; use Exception;
use Kiri; use Kiri;
use Kiri\Di\Context;
/** /**
* Class HasMany * Class HasMany
@@ -28,7 +29,6 @@ class HasMany extends HasBase
*/ */
public function get(): array|Collection|null public function get(): array|Collection|null
{ {
$relation = Kiri::getDi()->get(Relation::class); return di(Relation::class)->get($this->name);
return $relation->get($this->name);
} }
} }
+2 -2
View File
@@ -12,6 +12,7 @@ namespace Database;
use Database\Traits\HasBase; use Database\Traits\HasBase;
use Exception; use Exception;
use Kiri; use Kiri;
use Kiri\Di\Context;
/** /**
* Class HasOne * Class HasOne
@@ -27,7 +28,6 @@ class HasOne extends HasBase
*/ */
public function get(): array|ModelInterface|null public function get(): array|ModelInterface|null
{ {
$relation = Kiri::getDi()->get(Relation::class); return di(Relation::class)->first($this->name);
return $relation->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 $foreignKey
* @param string $localKey * @param string $localKey
* @return string * @return string
* @throws Exception * @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) { if (($value = $this->{$localKey}) === null) {
throw new Exception("Need join table primary key."); 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)) { if (!$relation->hasIdentification($primaryKey)) {
$relation->bindIdentification($primaryKey, $modelName::query()->where([$foreignKey => $value])); $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 $foreignKey
* @param string $localKey * @param string $localKey
* @return HasOne|ActiveQuery * @return HasOne|ActiveQuery
* @throws Exception * @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)); return new HasOne($this->_hasBase($modelName, $foreignKey, $localKey));
} }
/** /**
* @param ModelInterface|string $modelName * @param string $modelName
* @param string $foreignKey * @param string $foreignKey
* @param string $localKey * @param string $localKey
* @return ActiveQuery|HasCount * @return ActiveQuery|HasCount
* @throws Exception * @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)); return new HasCount($this->_hasBase($modelName, $foreignKey, $localKey));
} }
/** /**
* @param ModelInterface|string $modelName * @param string $modelName
* @param string $foreignKey * @param string $foreignKey
* @param string $localKey * @param string $localKey
* @return ActiveQuery|HasMany * @return ActiveQuery|HasMany
* @throws Exception * @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)); return new HasMany($this->_hasBase($modelName, $foreignKey, $localKey));
} }
/** /**
* @param ModelInterface|string $modelName * @param string $modelName
* @param string $foreignKey * @param string $foreignKey
* @param string $localKey * @param string $localKey
* @return ActiveQuery|HasMany * @return ActiveQuery|HasMany
* @throws Exception * @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) { if (($value = $this->{$localKey}) === null) {
throw new Exception("Need join table primary key."); 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)) { if (!$relation->hasIdentification($primaryKey)) {
$relation->bindIdentification($primaryKey, $modelName::query()->whereIn($foreignKey, $value)); $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 * @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; namespace Database;
use Database\Base\ActiveQueryInterface;
use Kiri\Abstracts\Component; use Kiri\Abstracts\Component;
use Kiri\Di\Context; use Kiri\Di\Context;
use Swoole\Coroutine;
/** /**
* Class Relation * Class Relation
@@ -34,7 +36,7 @@ class Relation extends Component
*/ */
public function hasIdentification(string $identification): bool 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 public function first(string $_identification): mixed
{ {
if (Context::exists($_identification)) { if (!Context::exists($_identification)) {
return Context::get($_identification); Context::set($_identification, $this->_query[$_identification]->first());
} }
$activeModel = $this->_query[$_identification]->first(); return Context::get($_identification);
unset($this->_query[$_identification]);
return Context::set($_identification, $activeModel);
} }
@@ -69,12 +69,10 @@ class Relation extends Component
*/ */
public function count(string $_identification): mixed public function count(string $_identification): mixed
{ {
if (Context::exists($_identification)) { if (!Context::exists($_identification)) {
return Context::get($_identification); Context::set($_identification, $this->_query[$_identification]->count());
} }
$activeModel = $this->_query[$_identification]->count(); return Context::get($_identification);
unset($this->_query[$_identification]);
return Context::set($_identification, $activeModel);
} }
@@ -86,11 +84,9 @@ class Relation extends Component
public function get(string $_identification): mixed public function get(string $_identification): mixed
{ {
if (Context::exists($_identification)) { if (Context::exists($_identification)) {
return Context::get($_identification); Context::set($_identification, $this->_query[$_identification]->get());
} }
$activeModel = $this->_query[$_identification]->get(); return Context::get($_identification);
unset($this->_query[$_identification]);
return Context::set($_identification, $activeModel);
} }
} }
+7 -2
View File
@@ -13,6 +13,7 @@ use Database\ModelInterface;
use Database\Collection; use Database\Collection;
use Database\Relation; use Database\Relation;
use Kiri; use Kiri;
use Kiri\Di\Context;
/** /**
* Class HasBase * Class HasBase
@@ -52,12 +53,16 @@ abstract class HasBase implements \Database\Traits\Relation
* @param string $name * @param string $name
* @param array $arguments * @param array $arguments
* @return $this|mixed * @return $this|mixed
* @throws \Exception
*/ */
public function __call(string $name, array $arguments) public function __call(string $name, array $arguments)
{ {
if ($name !== 'get') { if ($name !== 'get') {
$relation = Kiri::getDi()->get(Relation::class); $query = di(Relation::class)->getQuery($this->name);
$relation->getQuery($this->name)->$name(...$arguments); if (is_null($query)) {
throw new \Exception('Unknown relation key: ' . $this->name);
}
$query->$name(...$arguments);
return $this; return $this;
} else { } else {
return $this->get(); return $this->get();
+13
View File
@@ -0,0 +1,13 @@
<?php
class Users extends \Database\Model
{
public function hasD()
{
return $this->hasOne(static::class, 'id', 'id')->with([]);
}
}