Compare commits

...

66 Commits

Author SHA1 Message Date
as2252258 5756573d8d 改名 2021-11-24 16:56:16 +08:00
as2252258 eb75e69d60 改名 2021-11-24 16:48:25 +08:00
as2252258 6745036672 改名 2021-11-13 04:08:25 +08:00
as2252258 3e75bdc089 1 2021-11-11 03:45:09 +08:00
as2252258 70f3714b15 1 2021-11-11 03:39:57 +08:00
as2252258 d7f982b44f 1 2021-11-11 03:39:22 +08:00
as2252258 368a102409 1 2021-11-11 03:36:45 +08:00
as2252258 1f42c353b9 1 2021-11-11 03:34:53 +08:00
as2252258 50f6deeae7 1 2021-11-11 03:33:01 +08:00
as2252258 68dbaba012 1 2021-11-11 03:31:28 +08:00
as2252258 cb83878684 1 2021-11-11 03:29:02 +08:00
as2252258 1b4c7ea957 1 2021-11-11 03:26:59 +08:00
as2252258 a58b5217a8 1 2021-11-11 03:25:30 +08:00
as2252258 f4084b0b06 1 2021-11-11 03:23:05 +08:00
as2252258 2f4e6b23e0 1 2021-11-11 03:21:06 +08:00
as2252258 c13d4101b9 1 2021-11-11 03:14:52 +08:00
as2252258 07a40f4449 1 2021-11-11 03:11:41 +08:00
as2252258 229c0fa523 1 2021-11-11 03:10:01 +08:00
as2252258 8c3d3b2ce0 1 2021-11-11 03:03:40 +08:00
as2252258 8b164c7611 1 2021-11-11 03:00:50 +08:00
as2252258 8e9047de8a 改名 2021-11-10 15:15:30 +08:00
as2252258 ec8f0849ee 改名 2021-11-10 11:51:38 +08:00
as2252258 3038c332c0 改名 2021-11-10 11:48:33 +08:00
as2252258 98e8abfb7c 改名 2021-11-10 11:45:36 +08:00
as2252258 d3bb5e851f 改名 2021-11-10 11:33:15 +08:00
as2252258 d60799d5a4 改名 2021-11-10 11:29:13 +08:00
as2252258 0c32559716 改名 2021-11-10 11:27:38 +08:00
as2252258 ea4f7ba11a 改名 2021-11-10 11:23:42 +08:00
as2252258 a167169207 改名 2021-11-10 11:22:26 +08:00
as2252258 b5f347b07a 改名 2021-11-10 11:21:20 +08:00
as2252258 a113d5f64b 改名 2021-11-10 11:18:12 +08:00
as2252258 c3dbaa8828 改名 2021-11-10 11:16:13 +08:00
as2252258 b0dea25640 改名 2021-11-10 11:06:15 +08:00
as2252258 6dbc18c7d9 改名 2021-11-10 11:06:01 +08:00
as2252258 f22a58e121 改名 2021-11-10 11:03:35 +08:00
as2252258 d45725603d 改名 2021-11-10 11:02:07 +08:00
as2252258 f49d7cfa4b 改名 2021-11-10 10:29:19 +08:00
as2252258 e5a723cdb2 改名 2021-11-10 10:27:19 +08:00
as2252258 d56b28a4aa 改名 2021-11-10 10:18:45 +08:00
as2252258 b3d1f64241 1 2021-11-10 02:42:52 +08:00
as2252258 6b78d130fb 1 2021-11-10 02:41:39 +08:00
as2252258 ee44cfbca9 1 2021-11-10 02:36:49 +08:00
as2252258 3d588f3f33 1 2021-11-10 02:36:39 +08:00
as2252258 d903bba60f 1 2021-11-10 02:35:18 +08:00
as2252258 2f3be8cabc 1 2021-11-10 02:33:54 +08:00
as2252258 fcfc2213c2 1 2021-11-10 02:32:35 +08:00
as2252258 f76755407c 1 2021-11-10 02:31:34 +08:00
as2252258 2f64375ca2 1 2021-11-10 02:31:12 +08:00
as2252258 606cf7a722 1 2021-11-10 02:28:08 +08:00
as2252258 7400b27cb9 1 2021-11-10 02:23:14 +08:00
as2252258 50f2db7a2d 改名 2021-11-09 16:40:32 +08:00
as2252258 a538c39837 改名 2021-11-09 14:37:45 +08:00
as2252258 8c98ce8c7f 改名 2021-11-09 13:58:08 +08:00
as2252258 f8718fa4de 改名 2021-11-09 13:55:32 +08:00
as2252258 ffae51bccf 改名 2021-11-09 11:17:03 +08:00
as2252258 714df3dda4 改名 2021-11-09 11:15:43 +08:00
as2252258 f3a7789184 改名 2021-11-09 11:14:51 +08:00
as2252258 e186ebd54c 改名 2021-11-09 10:59:29 +08:00
as2252258 4d7c9fdd17 改名 2021-11-09 10:58:57 +08:00
as2252258 2ea8e8276b 改名 2021-11-09 10:50:07 +08:00
as2252258 d2bdb4355f 改名 2021-11-09 10:49:12 +08:00
as2252258 f7e88e6c18 改名 2021-11-09 10:48:27 +08:00
as2252258 67b961ce6e 改名 2021-11-09 10:47:50 +08:00
as2252258 81d0e09c65 改名 2021-11-09 10:43:29 +08:00
as2252258 022f3936c4 改名 2021-11-09 10:35:57 +08:00
as2252258 cf468b8a24 改名 2021-11-09 10:33:55 +08:00
15 changed files with 330 additions and 317 deletions
+3 -4
View File
@@ -237,7 +237,6 @@ class ActiveQuery extends Component implements ISqlBuilder
*/
public function count(): int
{
$this->select = ['COUNT(*)'];
$data = $this->execute($this->builder->count())->one();
if ($data && is_array($data)) {
return (int)array_shift($data);
@@ -270,7 +269,7 @@ class ActiveQuery extends Component implements ISqlBuilder
[$sql, $params] = $this->builder->insert($data, true);
return $this->execute($sql, $params)->exec(null, true);
return $this->execute($sql, $params)->exec();
}
/**
@@ -296,10 +295,10 @@ class ActiveQuery extends Component implements ISqlBuilder
/**
* @param bool $getSql
* @return string|bool
* @return int|bool|string|null
* @throws Exception
*/
public function delete(bool $getSql = false): string|bool
public function delete(bool $getSql = false): int|bool|string|null
{
$sql = $this->builder->delete();
if ($getSql === false) {
+1 -1
View File
@@ -134,7 +134,7 @@ abstract class AbstractCollection extends Component implements \IteratorAggregat
return NULL;
}
if (!($this->_item[$offset] instanceof ModelInterface)) {
return $this->model->setAttributes($this->_item[$offset]);
return $this->model->populates($this->_item[$offset]);
}
return $this->_item[$offset];
}
+1 -1
View File
@@ -56,7 +56,7 @@ class CollectionIterator extends \ArrayIterator
*/
protected function newModel($current): ModelInterface
{
return $this->model->setAttributes($current);
return $this->model->populates($current);
}
+49 -32
View File
@@ -13,12 +13,10 @@ defined('SAVE_FAIL') or define('SAVE_FAIL', 3227);
defined('FIND_OR_CREATE_MESSAGE') or define('FIND_OR_CREATE_MESSAGE', 'Create a new model, but the data cannot be empty.');
use Annotation\Inject;
use ArrayAccess;
use Closure;
use Database\ActiveQuery;
use Database\Connection;
use Database\HasCount;
use Database\HasMany;
use Database\HasOne;
use Database\ModelInterface;
@@ -98,12 +96,6 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, T
protected string $connection = 'db';
/**
* @var array
*/
protected array $rules = [];
/**
* @var array
*/
@@ -120,16 +112,26 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, T
/**
* @param string $name
* @param mixed $value
* @return array
*/
private function _setter(string $name, mixed $value): void
public function rules(): array
{
return [];
}
/**
* @param string $name
* @param mixed $value
* @return mixed
*/
private function _setter(string $name, mixed $value): mixed
{
$method = di(Setter::class)->getSetter(static::class, $name);
if (!empty($method)) {
$value = $this->{$method}($value);
}
$this->_attributes[$name] = $value;
return $value;
}
@@ -453,13 +455,13 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, T
if (!$if_condition_is_null) {
return false;
}
return static::query()->delete();
return (bool)static::query()->delete();
}
$model = static::query()->ifNotWhere($if_condition_is_null)->where($condition);
if (!empty($attributes)) {
$model->bindParams($attributes);
}
return $model->delete();
return (bool)$model->delete();
}
@@ -487,7 +489,7 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, T
*/
public function setAttribute($name, $value): mixed
{
return $this->_attributes[$name] = $value;
return $this->_attributes[$name] = $this->_setter($name, $value);
}
/**
@@ -497,7 +499,7 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, T
*/
public function setOldAttribute($name, $value): mixed
{
return $this->_oldAttributes[$name] = $value;
return $this->_oldAttributes[$name] = $this->_setter($name, $value);
}
/**
@@ -510,7 +512,9 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, T
if (empty($param)) {
return $this;
}
$this->_attributes = array_merge($this->_attributes, $param);
foreach ($param as $key => $attribute) {
$this->setAttribute($key, $attribute);
}
return $this;
}
@@ -523,10 +527,10 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, T
if (empty($param) || !is_array($param)) {
return $this;
}
foreach ($param as $key => $val) {
$this->setOldAttribute($key, $val);
}
return $this;
foreach ($param as $key => $attribute) {
$this->setOldAttribute($key, $attribute);
}
return $this;
}
/**
@@ -539,10 +543,10 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, T
{
[$sql, $param] = SqlBuilder::builder(static::query())->insert($param);
$dbConnection = $this->getConnection()->createCommand($sql, $param);
if (!($lastId = (int)$dbConnection->save(true, $this))) {
throw new Exception('保存失败.');
}
$lastId = $this->setPrimary($lastId, $param);
$lastId = $dbConnection->save(true);
$lastId = $this->setPrimary((int)$lastId, $param);
$this->refresh()->afterSave($attributes, $param);
@@ -571,7 +575,7 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, T
if (!isset($param[$primary]) || empty($param[$primary])) {
$this->setAttribute($primary, (int)$lastId);
}
return $this->setAttributes($param);
return $this;
}
@@ -608,10 +612,10 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, T
*/
public function save($data = NULL): static|bool
{
if (!is_null($data)) {
if (!is_null($data)) {
$this->_attributes = merge($this->_attributes, $data);
}
if (!$this->validator($this->rules) || !$this->beforeSave($this)) {
if (!$this->validator($this->rules()) || !$this->beforeSave($this)) {
return false;
}
[$change, $condition, $fields] = $this->separation();
@@ -622,6 +626,19 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, T
}
/**
* @param $value
* @return $this
*/
public function populates($value): static
{
$this->_attributes = $value;
$this->_oldAttributes = $value;
$this->setIsCreate(FALSE);
return $this;
}
/**
* @param array|null $rule
* @return bool
@@ -632,7 +649,7 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, T
if (empty($rule)) return true;
$validate = $this->resolve($rule);
if (!$validate->validation()) {
return $this->addError($validate->getError(), 'mysql');
return $this->addError('$validate->getError()', 'mysql');
} else {
return TRUE;
}
@@ -805,8 +822,8 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, T
throw new Exception('You need add static method `tableName` and return table name.');
}
$table = trim($this->table, '{{%}}');
if (!empty($tablePrefix) && !str_starts_with($this->table, $tablePrefix)) {
$table = $tablePrefix . $this->table;
if (!empty($tablePrefix) && !str_starts_with($table, $tablePrefix)) {
$table = $tablePrefix . $table;
}
return '`' . $connection->database . '`.' . $table;
}
@@ -853,7 +870,7 @@ abstract class Model extends Component implements ModelInterface, ArrayAccess, T
if (method_exists($this, 'set' . ucfirst($name))) {
$this->{'set' . ucfirst($name)}($value);
} else {
$this->_setter($name, $value);
$this->_attributes[$name] = $this->_setter($name, $value);
}
}
+5 -25
View File
@@ -60,7 +60,7 @@ class Command extends Component
*/
public function save(bool $isInsert = TRUE, mixed $hasAutoIncrement = null): int|bool|array|string|null
{
return $this->execute(static::EXECUTE, $isInsert, $hasAutoIncrement);
return $this->execute(static::EXECUTE);
}
@@ -111,17 +111,15 @@ class Command extends Component
/**
* @param $type
* @param null $isInsert
* @param bool|null $hasAutoIncrement
* @return int|bool|array|string|null
* @throws Exception
*/
private function execute($type, $isInsert = null, mixed $hasAutoIncrement = null): int|bool|array|string|null
private function execute($type): int|bool|array|string|null
{
try {
$time = microtime(true);
if ($type === static::EXECUTE) {
$result = $this->insert_or_change($isInsert, $hasAutoIncrement);
$result = $this->db->getConnect($this->sql)->execute($this->sql,$this->params);
} else {
$result = $this->search($type);
}
@@ -158,22 +156,6 @@ class Command extends Component
}
/**
* @param $isInsert
* @param $hasAutoIncrement
* @return bool|int
* @throws Exception
*/
private function insert_or_change($isInsert, $hasAutoIncrement): bool|int
{
$pdo = $this->db->getConnect($this->sql);
$result = $pdo->execute($this->sql, $isInsert, $this->params);
if ($hasAutoIncrement && $result == 0) {
return false;
}
return $result;
}
/**
* @return int|bool|array|string|null
* @throws Exception
@@ -184,14 +166,12 @@ class Command extends Component
}
/**
* @param null $scope
* @param bool $insert
* @return int|bool|array|string|null
* @throws Exception
*/
public function exec($scope = null, bool $insert = false): int|bool|array|string|null
public function exec(): int|bool|array|string|null
{
return $this->execute(static::EXECUTE, $insert, $scope);
return $this->execute(static::EXECUTE);
}
/**
+2 -3
View File
@@ -20,9 +20,8 @@ class HashCondition extends Condition
return '';
}
foreach ($this->value as $key => $value) {
if ($value === null) {
continue;
}
if (is_null($value)) continue;
$array[] = sprintf("%s = '%s'", $key, addslashes($value));
}
return implode(' AND ', $array);
+12 -1
View File
@@ -108,6 +108,17 @@ class Connection extends Component
}
/**
* @param $config
* @return $this
*/
public function configure($config): static
{
Kiri::configure($this, $config);
return $this;
}
/**
* @throws Exception
*/
@@ -141,7 +152,7 @@ class Connection extends Component
* @return mixed
* @throws ReflectionException
* @throws NotFindClassException
* @throws \Exception
* @throws Exception
*/
public function getSchema(): Schema
{
+7 -10
View File
@@ -6,13 +6,13 @@ namespace Database;
use Annotation\Inject;
use Exception;
use Server\Events\OnWorkerStart;
use Kiri\Abstracts\Config;
use Kiri\Abstracts\Providers;
use Kiri\Application;
use Kiri\Events\EventProvider;
use Kiri\Exception\ConfigException;
use Kiri\Kiri;
use Server\Events\OnWorkerStart;
/**
* Class DatabasesProviders
@@ -53,11 +53,9 @@ class DatabasesProviders extends Providers
*/
public function get($name): Connection
{
$application = Kiri::app();
if (!$application->has('databases.' . $name)) {
$application->set('databases.' . $name, $this->_settings($this->getConfig($name)));
}
return $application->get('databases.' . $name);
$config = $this->_settings($this->getConfig($name));
return Kiri::getDi()->get(Connection::class)->configure($config);
}
@@ -71,11 +69,10 @@ class DatabasesProviders extends Providers
if (empty($databases)) {
return;
}
$application = Kiri::app();
foreach ($databases as $name => $database) {
$connection = Kiri::getDi()->get(Connection::class);
foreach ($databases as $database) {
/** @var Connection $connection */
$application->set('databases.' . $name, $this->_settings($database));
$application->get('databases.' . $name)->fill();
$connection->configure($database)->fill();
}
}
+4 -4
View File
@@ -334,7 +334,7 @@ class Model extends Base\Model
*/
public function hasOne(string $modelName, $foreignKey, $localKey): HasOne|ActiveQuery
{
if (($value = $this->getAttribute($localKey)) === null) {
if (($value = $this->{$localKey}) === null) {
throw new Exception("Need join table primary key.");
}
@@ -353,7 +353,7 @@ class Model extends Base\Model
*/
public function hasCount($modelName, $foreignKey, $localKey): ActiveQuery|HasCount
{
if (($value = $this->getAttribute($localKey)) === null) {
if (($value = $this->{$localKey}) === null) {
throw new Exception("Need join table primary key.");
}
@@ -372,7 +372,7 @@ class Model extends Base\Model
*/
public function hasMany($modelName, $foreignKey, $localKey): ActiveQuery|HasMany
{
if (($value = $this->getAttribute($localKey)) === null) {
if (($value = $this->{$localKey}) === null) {
throw new Exception("Need join table primary key.");
}
@@ -390,7 +390,7 @@ class Model extends Base\Model
*/
public function hasIn($modelName, $foreignKey, $localKey): ActiveQuery|HasMany
{
if (($value = $this->getAttribute($localKey)) === null) {
if (($value = $this->{$localKey}) === null) {
throw new Exception("Need join table primary key.");
}
+5 -4
View File
@@ -32,10 +32,11 @@ interface ModelInterface
public static function find(string|int $param): mixed;
/**
* @return string
*/
public static function className(): string;
/**
* @param array $data
* @return static
*/
public static function populate(array $data): static;
/**
+1 -1
View File
@@ -316,7 +316,7 @@ class Columns extends Component
/**
* @param $column
* @param array $column
* @param $table
* @return array
*/
+222 -224
View File
@@ -3,8 +3,8 @@
namespace Database\Mysql;
use Exception;
use Kiri\Context;
use Kiri\Abstracts\Logger;
use Kiri\Context;
use Kiri\Kiri;
use Kiri\Pool\StopHeartbeatCheck;
use PDOStatement;
@@ -16,122 +16,122 @@ use Swoole\Timer;
class PDO implements StopHeartbeatCheck
{
const DB_ERROR_MESSAGE = 'The system is busy, please try again later.';
const DB_ERROR_MESSAGE = 'The system is busy, please try again later.';
private ?\PDO $pdo = null;
private ?\PDO $pdo = null;
private int $_transaction = 0;
private int $_transaction = 0;
private int $_timer = -1;
private int $_timer = -1;
private int $_last = 0;
private int $_last = 0;
/**
* @param string $dbname
* @param string $cds
* @param string $username
* @param string $password
* @param string $chatset
* @throws
*/
public function __construct(public string $dbname, public string $cds,
public string $username, public string $password, public string $chatset = 'utf8mb4')
{
}
/**
* @param string $dbname
* @param string $cds
* @param string $username
* @param string $password
* @param string $chatset
* @throws
*/
public function __construct(public string $dbname, public string $cds,
public string $username, public string $password, public string $chatset = 'utf8mb4')
{
}
public function init()
{
$this->heartbeat_check();
}
public function init()
{
$this->heartbeat_check();
}
/**
* @return bool
*/
public function inTransaction(): bool
{
return $this->_transaction > 0;
}
/**
* @return bool
*/
public function inTransaction(): bool
{
return $this->_transaction > 0;
}
/**
*
*/
public function heartbeat_check(): void
{
if (env('state', 'start') == 'exit') {
return;
}
if ($this->_timer === -1 && Context::inCoroutine()) {
$this->_timer = Timer::tick(1000, function () {
try {
if (env('state', 'start') == 'exit') {
Kiri::getDi()->get(Logger::class)->critical('timer end');
$this->stopHeartbeatCheck();
}
if (time() - $this->_last > 10 * 60) {
$this->stopHeartbeatCheck();
$this->pdo = null;
}
} catch (\Throwable $throwable) {
error($throwable);
}
});
}
}
/**
*
*/
public function heartbeat_check(): void
{
if (env('state', 'start') == 'exit') {
return;
}
if ($this->_timer === -1 && Context::inCoroutine()) {
$this->_timer = Timer::tick(1000, function () {
try {
if (env('state', 'start') == 'exit') {
Kiri::getDi()->get(Logger::class)->critical('timer end');
$this->stopHeartbeatCheck();
}
if (time() - $this->_last > 10 * 60) {
$this->stopHeartbeatCheck();
$this->pdo = null;
}
} catch (\Throwable $throwable) {
error($throwable);
}
});
}
}
/**
*
*/
public function stopHeartbeatCheck(): void
{
if ($this->_timer > -1) {
Timer::clear($this->_timer);
}
$this->_timer = -1;
}
/**
*
*/
public function stopHeartbeatCheck(): void
{
if ($this->_timer > -1) {
Timer::clear($this->_timer);
}
$this->_timer = -1;
}
/**
*
*/
public function beginTransaction()
{
if ($this->_transaction == 0) {
$this->_pdo()->beginTransaction();
}
$this->_transaction++;
}
/**
*
*/
public function beginTransaction()
{
if ($this->_transaction == 0) {
$this->_pdo()->beginTransaction();
}
$this->_transaction++;
}
/**
*
*/
public function commit()
{
if ($this->_transaction == 0) {
$this->_pdo()->commit();
}
$this->_transaction--;
}
/**
*
*/
public function commit()
{
if ($this->_transaction == 0) {
$this->_pdo()->commit();
}
$this->_transaction--;
}
/**
*
*/
public function rollback()
{
if ($this->_transaction == 0) {
$this->_pdo()->rollBack();
}
$this->_transaction--;
}
/**
*
*/
public function rollback()
{
if ($this->_transaction == 0) {
$this->_pdo()->rollBack();
}
$this->_transaction--;
}
/**
@@ -140,14 +140,14 @@ class PDO implements StopHeartbeatCheck
* @return bool|array|null
* @throws Exception
*/
public function fetchAll(string $sql, array $params = []): bool|null|array
{
$pdo = $this->queryPrev($sql, $params);
public function fetchAll(string $sql, array $params = []): bool|null|array
{
$pdo = $this->queryPrev($sql, $params);
$result = $pdo->fetchAll(\PDO::FETCH_ASSOC);
$pdo->closeCursor();
return $result;
}
$result = $pdo->fetchAll(\PDO::FETCH_ASSOC);
$pdo->closeCursor();
return $result;
}
/**
@@ -156,146 +156,144 @@ class PDO implements StopHeartbeatCheck
* @return bool|array|null
* @throws Exception
*/
public function fetch(string $sql, array $params = []): bool|null|array
{
$pdo = $this->queryPrev($sql, $params);
public function fetch(string $sql, array $params = []): bool|null|array
{
$pdo = $this->queryPrev($sql, $params);
$result = $pdo->fetch(\PDO::FETCH_ASSOC);
$pdo->closeCursor();
return $result;
}
$result = $pdo->fetch(\PDO::FETCH_ASSOC);
$pdo->closeCursor();
return $result;
}
/**
* @param string $sql
* @param array $params
* @return bool|array|null
* @throws \Exception
*/
public function fetchColumn(string $sql, array $params = []): bool|null|array
{
$pdo = $this->queryPrev($sql, $params);
/**
* @param string $sql
* @param array $params
* @return bool|array|null
* @throws \Exception
*/
public function fetchColumn(string $sql, array $params = []): bool|null|array
{
$pdo = $this->queryPrev($sql, $params);
$result = $pdo->fetchColumn(\PDO::FETCH_ASSOC);
$pdo->closeCursor();
return $result;
}
$result = $pdo->fetchColumn(\PDO::FETCH_ASSOC);
$pdo->closeCursor();
return $result;
}
/**
* @param string $sql
* @param array $params
* @return int
* @throws Exception
*/
public function count(string $sql, array $params = []): int
{
$pdo = $this->queryPrev($sql, $params);
/**
* @param string $sql
* @param array $params
* @return int
* @throws Exception
*/
public function count(string $sql, array $params = []): int
{
$pdo = $this->queryPrev($sql, $params);
$result = $pdo->rowCount();
$pdo->closeCursor();
return $result;
}
$result = $pdo->rowCount();
$pdo->closeCursor();
return $result;
}
/**
* @param string $sql
* @param array $params
* @return PDOStatement
* @throws Exception
*/
private function queryPrev(string $sql, array $params = []): PDOStatement
{
$this->_last = time();
try {
if (($statement = $this->_pdo()->query($sql)) === false) {
throw new Exception($this->_pdo()->errorInfo()[1]);
}
return $this->bindValue($statement, $params);
} catch (\Throwable $throwable) {
/**
* @param string $sql
* @param array $params
* @return PDOStatement
* @throws Exception
*/
private function queryPrev(string $sql, array $params = []): PDOStatement
{
$this->_last = time();
try {
if (($statement = $this->_pdo()->query($sql)) === false) {
throw new Exception($this->_pdo()->errorInfo()[1]);
}
return $this->bindValue($statement, $params);
} catch (\PDOException | \Throwable $throwable) {
if (str_contains($throwable->getMessage(), 'MySQL server has gone away')) {
$this->pdo = null;
if (str_contains($throwable->getMessage(), 'MySQL server has gone away')) {
$this->pdo = null;
return $this->queryPrev($sql, $params);
}
throw new Exception($throwable->getMessage());
}
}
return $this->queryPrev($sql, $params);
}
throw new Exception($throwable->getMessage());
}
}
/**
* @param PDOStatement $statement
* @param array $params
* @return PDOStatement
*/
private function bindValue(PDOStatement $statement, array $params = []): PDOStatement
{
if (empty($params)) return $statement;
foreach ($params as $key => $param) {
$statement->bindValue($key, $param);
}
return $statement;
}
/**
* @param PDOStatement $statement
* @param array $params
* @return PDOStatement
*/
private function bindValue(PDOStatement $statement, array $params = []): PDOStatement
{
if (empty($params)) return $statement;
foreach ($params as $key => $param) {
$statement->bindValue($key, $param);
}
return $statement;
}
/**
* @param string $sql
* @param $isInsert
* @param array $params
* @return int
* @throws Exception
*/
public function execute(string $sql, $isInsert, array $params = []): int
{
$this->_last = time();
if (!(($prepare = $this->_pdo()->prepare($sql)) instanceof PDOStatement)) {
throw new Exception($prepare->errorInfo()[2] ?? static::DB_ERROR_MESSAGE);
}
if ($prepare->execute($params) === false) {
throw new Exception($prepare->errorInfo()[2] ?? static::DB_ERROR_MESSAGE);
}
$result = 1;
if ($isInsert) {
$result = (int)$this->_pdo()->lastInsertId();
}
$prepare->closeCursor();
return $result;
}
/**
* @param string $sql
* @param array $params
* @return int
* @throws Exception
*/
public function execute(string $sql, array $params = []): int
{
$this->_last = time();
$pdo = $this->_pdo();
if (!(($prepare = $pdo->prepare($sql)) instanceof PDOStatement)) {
throw new Exception($prepare->errorInfo()[2] ?? static::DB_ERROR_MESSAGE);
}
if ($prepare->execute($params) === false) {
throw new Exception($prepare->errorInfo()[2] ?? static::DB_ERROR_MESSAGE);
}
$result = (int)$pdo->lastInsertId();
$prepare->closeCursor();
if ($result == 0) {
return true;
}
return $result;
}
/**
* @return \PDO
*/
public function _pdo(): \PDO
{
if ($this->_timer === -1) {
$this->heartbeat_check();
}
if (!($this->pdo instanceof \PDO)) {
$this->pdo = $this->newClient();
}
return $this->pdo;
}
/**
* @return \PDO
*/
public function _pdo(): \PDO
{
if ($this->_timer === -1) {
$this->heartbeat_check();
}
if (!($this->pdo instanceof \PDO)) {
$this->pdo = $this->newClient();
}
return $this->pdo;
}
/**
* @return \PDO
*/
private function newClient(): \PDO
{
$link = new \PDO('mysql:dbname=' . $this->dbname . ';host=' . $this->cds, $this->username, $this->password, [
\PDO::ATTR_EMULATE_PREPARES => false,
\PDO::ATTR_CASE => \PDO::CASE_NATURAL,
\PDO::ATTR_TIMEOUT => 60,
\PDO::MYSQL_ATTR_USE_BUFFERED_QUERY => true,
\PDO::MYSQL_ATTR_INIT_COMMAND => 'SET NAMES ' . $this->chatset
]);
$link->setAttribute(\PDO::ATTR_ERRMODE, \PDO::ERRMODE_EXCEPTION);
$link->setAttribute(\PDO::ATTR_STRINGIFY_FETCHES, false);
$link->setAttribute(\PDO::ATTR_ORACLE_NULLS, \PDO::NULL_EMPTY_STRING);
return $link;
}
/**
* @return \PDO
*/
private function newClient(): \PDO
{
$link = new \PDO('mysql:dbname=' . $this->dbname . ';host=' . $this->cds, $this->username, $this->password, [
\PDO::ATTR_EMULATE_PREPARES => false,
\PDO::ATTR_CASE => \PDO::CASE_NATURAL,
\PDO::ATTR_TIMEOUT => 60,
\PDO::MYSQL_ATTR_USE_BUFFERED_QUERY => true,
\PDO::MYSQL_ATTR_INIT_COMMAND => 'SET NAMES ' . $this->chatset
]);
$link->setAttribute(\PDO::ATTR_ERRMODE, \PDO::ERRMODE_EXCEPTION);
$link->setAttribute(\PDO::ATTR_STRINGIFY_FETCHES, false);
$link->setAttribute(\PDO::ATTR_ORACLE_NULLS, \PDO::NULL_EMPTY_STRING);
return $link;
}
}
+7 -5
View File
@@ -240,7 +240,7 @@ class SqlBuilder extends Component
if (empty($this->query->from) && !empty($this->query->modelClass)) {
$this->query->from($this->query->getTable());
}
return $this->_prefix();
return $this->_prefix(false, true);
}
@@ -256,14 +256,15 @@ class SqlBuilder extends Component
/**
* @param bool $hasOrder
* @param bool $isCount
* @return string
* @throws Exception
*/
private function _prefix(bool $hasOrder = false): string
private function _prefix(bool $hasOrder = false, bool $isCount = false): string
{
$select = '';
if (!empty($this->query->from)) {
$select = $this->_selectPrefix();
$select = $this->_selectPrefix($isCount);
}
$select = $this->_wherePrefix($select);
if (!empty($this->query->attributes) && is_array($this->query->attributes)) {
@@ -302,12 +303,13 @@ class SqlBuilder extends Component
/**
* @param bool $isCount
* @return string
* @throws Exception
*/
private function _selectPrefix(): string
private function _selectPrefix(bool $isCount): string
{
$select = $this->builderSelect($this->query->select) . ' FROM ' . $this->tableName();
$select = $this->builderSelect($this->query->select, $isCount) . ' FROM ' . $this->tableName();
if (!empty($this->query->alias)) {
$select .= $this->builderAlias($this->query->alias);
}
+9 -1
View File
@@ -61,10 +61,14 @@ trait Builder
/**
* @param null $select
* @param bool $isCount
* @return string
*/
#[Pure] private function builderSelect($select = NULL): string
#[Pure] private function builderSelect($select = NULL, bool $isCount = false): string
{
if ($isCount) {
return "SELECT COUNT(*)";
}
if (empty($select)) {
return "SELECT *";
}
@@ -129,6 +133,8 @@ trait Builder
if (empty($where)) return '';
if (is_string($where)) return $where;
foreach ($where as $key => $value) {
if (is_null($value)) continue;
$_value = $this->resolveCondition($key, $value, $_tmp);
if (empty($_value)) continue;
@@ -210,6 +216,8 @@ trait Builder
{
$_array = [];
foreach ($condition as $key => $value) {
if (is_null($value)) continue;
$value = is_numeric($value) ? $value : '\'' . $value . '\'';
if (!is_numeric($key)) {
$_array[] = sprintf('%s = %s', $key, $value);
+2 -1
View File
@@ -522,7 +522,7 @@ trait QueryTrait
$conditionArray = $this->sprintf($conditionArray, $value, $opera);
}
$this->where = ['(' . implode(' AND ', $this->where) . ') OR (' . $conditionArray . ')'];
$this->where = ['((' . implode(' AND ', $this->where) . ') OR (' . $conditionArray . '))'];
return $this;
}
@@ -923,6 +923,7 @@ trait QueryTrait
private function addArray(array $array): static
{
foreach ($array as $key => $value) {
if (is_null($value)) continue;
if (is_numeric($key)) {
[$column, $opera, $value] = $this->opera(...$value);