Compare commits

..

27 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
11 changed files with 137 additions and 66 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
*/
@@ -95,7 +137,7 @@ class ActiveQuery extends QueryTrait implements ISqlBuilder
return;
}
if (Context::inCoroutine()) {
Coroutine::create(fn() => $closure($data));
Coroutine::create(fn () => $closure($data));
} else {
call_user_func($closure, $data);
}
+19 -13
View File
@@ -23,6 +23,7 @@ use Database\ModelInterface;
use Database\Relation;
use Database\SqlBuilder;
use Exception;
use Kiri\Di\Context;
use Kiri;
use Kiri\Abstracts\Component;
use ReturnTypeWillChange;
@@ -223,8 +224,8 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, \
*/
public function getPrimaryValue(): ?int
{
if ($this->hasPrimary()) {
return (int)$this->_oldAttributes[$this->getPrimary()] ?? null;
if ($this->hasPrimary() && isset($this->_oldAttributes[$this->getPrimary()])) {
return (int)$this->_oldAttributes[$this->getPrimary()];
} else {
return null;
}
@@ -237,22 +238,23 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, \
*/
public static function findOne(int|string|array|null $param): ?static
{
if (empty($param)) return null;
$model = static::instance();
$query = new ActiveQuery($model);
if (empty($param)) {
return null;
}
$query = new ActiveQuery($model = static::instance());
$query->from($model->getTable())->alias('t1');
if (is_numeric($param)) {
$query->where([$model->getPrimary() => $param]);
$query->where([$model->getPrimary() => +$param]);
} else if (is_array($param)) {
$query->where($param);
} else {
$query->whereRaw($param);
}
$data = $query->first();
if ($data === false) {
if (($data = $query->first()) === false) {
throw new Exception($model->getLastError());
} else {
return $data;
}
return $data;
}
@@ -442,10 +444,13 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, \
if ($lastId === false) {
return false;
}
if ($this->hasPrimary()) {
$this->_attributes[$this->getPrimary()] = $lastId;
if (!$this->hasPrimary()) {
return $this->refresh()->afterSave($this->_attributes, []);
}
return $this;
$this->_attributes[$this->getPrimary()] = $lastId;
return $this->refresh()->afterSave($this->_attributes, [$this->getPrimary() => $lastId]);
}
@@ -593,7 +598,7 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, \
*/
public function getRelation(): ?Relation
{
return Kiri::getDi()->get(Relation::class);
return di(Relation::class);
}
@@ -655,6 +660,7 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, \
public function refresh(): static
{
$this->_oldAttributes = $this->_attributes;
$this->isNewExample = false;
return $this;
}
+4 -2
View File
@@ -109,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]);
}
@@ -118,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) {
@@ -151,6 +152,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] . ')' . $prepare->errorInfo()[2]);
}
@@ -161,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) {
+20 -12
View File
@@ -67,6 +67,14 @@ class Connection extends Component
#[Container(LoggerInterface::class)]
public StdoutLogger $logger;
/**
* @var EventProvider
*/
#[Container(EventProvider::class)]
public EventProvider $eventProvider;
/**
* @param Pool $connections
*/
@@ -79,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);
}
}
@@ -97,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']);
}
@@ -113,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()));
}
@@ -348,7 +357,6 @@ class Connection extends Component
public function newConnect(): \PDO
{
$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_ERRMODE => \PDO::ERRMODE_EXCEPTION,
\PDO::ATTR_ORACLE_NULLS => \PDO::NULL_NATURAL,
+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));
}
+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);
}
}
+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();
+13
View File
@@ -0,0 +1,13 @@
<?php
class Users extends \Database\Model
{
public function hasD()
{
return $this->hasOne(static::class, 'id', 'id')->with([]);
}
}