Compare commits

...

9 Commits

Author SHA1 Message Date
as2252258 92f0c1a973 eee 2024-01-10 18:12:39 +08:00
as2252258 11ad7217c2 eee 2024-01-10 18:08:41 +08:00
as2252258 b647716f1c eee 2024-01-10 18:07:40 +08:00
as2252258 d8b2d1ca2c eee 2024-01-10 17:59:05 +08:00
as2252258 a439c616b4 eee 2024-01-10 17:48:54 +08:00
as2252258 d036e440cd eee 2024-01-10 17:43:10 +08:00
as2252258 88fb563c99 eee 2024-01-10 17:35:55 +08:00
as2252258 1066aa03e8 eee 2023-12-26 17:41:03 +08:00
as2252258 609e5d4749 eee 2023-12-26 17:40:48 +08:00
5 changed files with 72 additions and 24 deletions
+1
View File
@@ -89,6 +89,7 @@ class ActiveQuery extends QueryTrait implements ISqlBuilder
*/ */
public function chunk(int $size, Closure $closure): void public function chunk(int $size, Closure $closure): void
{ {
if ($this->offset === -1) $this->offset = 0;
$data = $this->offset($this->offset)->limit($size)->get(); $data = $this->offset($this->offset)->limit($size)->get();
if (!$data || $data->isEmpty()) { if (!$data || $data->isEmpty()) {
return; return;
+6 -3
View File
@@ -476,7 +476,7 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, \
*/ */
public function save(): static|bool public function save(): static|bool
{ {
if (!$this->validator($this->rules()) || !$this->beforeSave($this)) { if (!$this->validator($this->rules(), $this->_attributes) || !$this->beforeSave($this)) {
return FALSE; return FALSE;
} }
if (!$this->isNewExample) { if (!$this->isNewExample) {
@@ -503,6 +503,9 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, \
$oldPrams[$key] = $this->_oldAttributes[$key]; $oldPrams[$key] = $this->_oldAttributes[$key];
} }
} }
if ($this->hasPrimary()) {
$condition = [$this->getPrimary() => $this->getPrimaryValue()];
}
return [$oldPrams, $condition, $params]; return [$oldPrams, $condition, $params];
} }
@@ -537,13 +540,13 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, \
* @return bool * @return bool
* @throws * @throws
*/ */
public function validator(array $rule): bool public function validator(array $rule, array $params): bool
{ {
if (count($rule) < 1 || $this->skipValidate) { if (count($rule) < 1 || $this->skipValidate) {
return TRUE; return TRUE;
} }
$validate = $this->resolve($rule); $validate = $this->resolve($rule);
if (!$validate->validation($this)) { if (!$validate->validation($params)) {
return \Kiri::getLogger()->failure($validate->getError() . PHP_EOL, 'mysql'); return \Kiri::getLogger()->failure($validate->getError() . PHP_EOL, 'mysql');
} else { } else {
return TRUE; return TRUE;
+30 -2
View File
@@ -88,9 +88,32 @@ class Command extends Component
* @return mixed * @return mixed
* @throws * @throws
*/ */
public function rowCount(): int public function rowCount(): mixed
{ {
return $this->search('rowCount'); $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]);
}
$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);
}
} }
@@ -126,6 +149,8 @@ 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);
return $result; return $result;
} catch (Throwable $throwable) { } catch (Throwable $throwable) {
if ($this->isRefresh($throwable)) return $this->search($method); if ($this->isRefresh($throwable)) return $this->search($method);
@@ -157,6 +182,7 @@ class Command extends Component
{ {
$client = $this->connection->getConnection(); $client = $this->connection->getConnection();
try { try {
$this->connection->println($this->sql, $this->params);
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]);
} }
@@ -167,6 +193,8 @@ class Command extends Component
$result = $client->lastInsertId(); $result = $client->lastInsertId();
$this->connection->println($this->sql, $this->params);
return $result == 0 ? $prepare->rowCount() : (int)$result; return $result == 0 ? $prepare->rowCount() : (int)$result;
} catch (Throwable $throwable) { } catch (Throwable $throwable) {
if ($this->isRefresh($throwable)) return $this->_prepare(); if ($this->isRefresh($throwable)) return $this->_prepare();
+34 -18
View File
@@ -37,24 +37,25 @@ use Swoole\Timer;
class Connection extends Component class Connection extends Component
{ {
public string $id = 'db'; public string $id = 'db';
public string $cds = ''; public string $cds = '';
public string $password = ''; public string $password = '';
public string $username = ''; public string $username = '';
public string $charset = 'utf-8'; public string $charset = 'utf-8';
public string $tablePrefix = ''; public string $tablePrefix = '';
public string $database = ''; public string $database = '';
public int $timeout = 30; public int $timeout = 30;
public int $waite_time = 3; public int $waite_time = 3;
public int $tick_time = 60; public int $tick_time = 60;
public int $idle_count = 3; public int $idle_count = 3;
public int $idle_time = 60; public int $idle_time = 60;
public array $pool = ['max' => 10, 'min' => 1]; public array $pool = ['max' => 10, 'min' => 1];
private int $storey = 0; private int $storey = 0;
protected int $timerId = -1; protected int $timerId = -1;
public bool $enableCache = false; public bool $enableCache = false;
public string $cacheDriver = 'redis'; public string $cacheDriver = 'redis';
public array $attributes = []; public array $attributes = [];
protected \Closure $_println;
/** /**
@@ -69,6 +70,21 @@ class Connection extends Component
public function __construct(public Pool $connections) public function __construct(public Pool $connections)
{ {
parent::__construct(); parent::__construct();
$this->_println = \config('databases.logger', null);
}
/**
* @param string $sql
* @param array $params
* @return void
*/
public function println(string $sql, array $params = []): void
{
if (is_callable($this->_println)) {
call_user_func($this->_println, $sql, $params);
}
} }
+1 -1
View File
@@ -191,7 +191,7 @@ class Model extends Base\Model
*/ */
public function update(array $params): static|bool public function update(array $params): static|bool
{ {
if (!$this->validator($this->rules()) || !$this->beforeSave($this)) { if (!$this->validator($this->rules(), $params) || !$this->beforeSave($this)) {
return FALSE; return FALSE;
} }
return $this->updateInternal(...$this->arrayIntersect($params)); return $this->updateInternal(...$this->arrayIntersect($params));