Compare commits

..

7 Commits

Author SHA1 Message Date
as2252258 547bb85ba9 1 2021-12-12 23:55:38 +08:00
as2252258 4ddcc87263 1 2021-12-12 23:54:27 +08:00
as2252258 31a6862d62 1 2021-12-12 23:52:35 +08:00
as2252258 6e1d1a300a 1 2021-12-12 06:06:08 +08:00
as2252258 e260e43c17 1 2021-12-12 04:29:35 +08:00
as2252258 76292522e4 1 2021-12-12 04:29:09 +08:00
as2252258 7eb6151111 改名 2021-12-08 11:32:31 +08:00
4 changed files with 1229 additions and 1211 deletions
+284 -253
View File
@@ -11,6 +11,7 @@ namespace Database;
use Database\Traits\QueryTrait;
use Exception;
use JetBrains\PhpStorm\ArrayShape;
use Kiri\Abstracts\Component;
/**
@@ -20,290 +21,320 @@ use Kiri\Abstracts\Component;
class ActiveQuery extends Component implements ISqlBuilder
{
use QueryTrait;
use QueryTrait;
/** @var array */
public array $with = [];
/** @var array */
public array $with = [];
/** @var bool */
public bool $asArray = FALSE;
/** @var bool */
public bool $asArray = FALSE;
/** @var bool */
public bool $useCache = FALSE;
/** @var bool */
public bool $useCache = FALSE;
/**
* @var Connection|null
*/
public ?Connection $db = NULL;
/**
* @var Connection|null
*/
public ?Connection $db = NULL;
/**
* @var array
* 参数绑定
*/
public array $attributes = [];
/**
* @var array
* 参数绑定
*/
public array $attributes = [];
/**
* Comply constructor.
* @param $model
* @param array $config
* @throws
*/
public function __construct($model, array $config = [])
{
$this->modelClass = $model;
/**
* Comply constructor.
* @param $model
* @param array $config
* @throws
*/
public function __construct($model, array $config = [])
{
$this->modelClass = $model;
$this->builder = SqlBuilder::builder($this);
parent::__construct($config);
}
$this->builder = SqlBuilder::builder($this);
parent::__construct($config);
}
/**
* 清除不完整数据
*/
public function clear()
{
$this->db = null;
$this->useCache = false;
$this->with = [];
}
/**
* 清除不完整数据
*/
public function clear()
{
$this->db = NULL;
$this->useCache = FALSE;
$this->with = [];
}
/**
* @param $key
* @param $value
* @return $this
*/
public function addParam($key, $value): static
{
$this->attributes[$key] = $value;
return $this;
}
/**
* @param array $values
* @return $this
*/
public function addParams(array $values): static
{
foreach ($values as $key => $val) {
$this->addParam($key, $val);
}
return $this;
}
/**
* @param $name
* @return $this
*/
public function with($name): static
{
if (empty($name)) {
return $this;
}
if (is_string($name)) {
$name = explode(',', $name);
}
foreach ($name as $val) {
array_push($this->with, $val);
}
return $this;
}
/**
* @param $key
* @param $value
* @return $this
*/
public function addParam($key, $value): static
{
$this->attributes[$key] = $value;
return $this;
}
/**
* @param $sql
* @param array $params
* @return mixed
* @throws Exception
*/
public function execute($sql, array $params = []): Command
{
return $this->modelClass->getConnection()->createCommand($sql, $params);
}
/**
* @param int $size
* @param int $page
* @return array
* @throws Exception
*/
#[ArrayShape(['code' => "int", 'message' => "string", 'size' => "mixed", 'page' => "mixed", 'count' => "int", 'next' => "mixed", 'prev' => "mixed", 'param' => "array"])]
public function pagination(int $size = 20, int $page = 1): array
{
$page = max(1, $page);
$size = max(1, $size);
$offset = ($page - 1) * $size;
$count = $this->count();
$lists = $this->limit($offset, $size)->get()->toArray();
return [
'code' => 0,
'message' => 'ok',
'size' => $size,
'page' => $page,
'count' => $count,
'next' => max($page + 1, 1),
'prev' => max($page - 1, 1),
'param' => $lists,
];
}
/**
* @return ModelInterface|null
* @throws Exception
*/
public function first(): ModelInterface|null
{
$data = $this->execute($this->builder->one())->one();
if (empty($data)) {
return NULL;
}
return $this->populate($data);
}
/**
* @param array $values
* @return $this
*/
public function addParams(array $values): static
{
foreach ($values as $key => $val) {
$this->addParam($key, $val);
}
return $this;
}
/**
* @param $name
* @return $this
*/
public function with($name): static
{
if (empty($name)) {
return $this;
}
if (is_string($name)) {
$name = explode(',', $name);
}
foreach ($name as $val) {
array_push($this->with, $val);
}
return $this;
}
/**
* @return string
* @throws Exception
*/
public function toSql(): string
{
return $this->builder->get();
}
/**
* @param $sql
* @param array $params
* @return mixed
* @throws Exception
*/
public function execute($sql, array $params = []): Command
{
return $this->modelClass->getConnection()->createCommand($sql, $params);
}
/**
* @return array|Collection
*/
public function get(): Collection|array
{
return $this->all();
}
/**
* @return ModelInterface|null
* @throws Exception
*/
public function first(): ModelInterface|null
{
$data = $this->execute($this->builder->one())->one();
if (empty($data)) {
return NULL;
}
return $this->populate($data);
}
/**
* @throws Exception
*/
public function flush(): array|bool|int|string|null
{
return $this->execute($this->builder->truncate())->exec();
}
/**
* @return string
* @throws Exception
*/
public function toSql(): string
{
return $this->builder->get();
}
/**
* @param int $size
* @param callable $callback
* @return Pagination
* @throws Exception
*/
public function page(int $size, callable $callback): Pagination
{
$pagination = new Pagination($this);
$pagination->setOffset(0);
$pagination->setLimit($size);
$pagination->setCallback($callback);
return $pagination;
}
/**
* @param string $field
* @param string $setKey
*
* @return array|null
* @throws Exception
*/
public function column(string $field, string $setKey = ''): ?array
{
return $this->all()->column($field, $setKey);
}
/**
* @return array|Collection
* @throws
*/
public function all(): Collection|array
{
$data = $this->execute($this->builder->all())->all();
if (!empty($this->with)){
$this->getWith($this->modelClass);
}
$collect = new Collection($this, $data, $this->modelClass);
if ($this->asArray) {
return $collect->toArray();
}
return $collect;
}
/**
* @param $data
* @return ModelInterface
* @throws Exception
*/
public function populate($data): ModelInterface
{
return $this->getWith($this->modelClass::populate($data));
}
/**
* @return array|Collection
*/
public function get(): Collection|array
{
return $this->all();
}
/**
* @param ModelInterface $model
* @return ModelInterface
*/
public function getWith(ModelInterface $model): ModelInterface
{
if (empty($this->with) || !is_array($this->with)) {
return $model;
}
return $model->setWith($this->with);
}
/**
* @return int
* @throws Exception
*/
public function count(): int
{
$data = $this->execute($this->builder->count())->one();
if ($data && is_array($data)) {
return (int)array_shift($data);
}
return 0;
}
/**
* @throws Exception
*/
public function flush(): array|bool|int|string|null
{
return $this->execute($this->builder->truncate())->exec();
}
/**
* @param array $data
* @return array|Command|bool|int|string
* @throws Exception
*/
public function batchUpdate(array $data): Command|array|bool|int|string
{
$generate = $this->builder->update($data);
if (is_bool($generate)) {
return $generate;
}
return $this->execute(...$generate)->exec();
}
/**
* @param int $size
* @param callable $callback
* @return Pagination
* @throws Exception
*/
public function page(int $size, callable $callback): Pagination
{
$pagination = new Pagination($this);
$pagination->setOffset(0);
$pagination->setLimit($size);
$pagination->setCallback($callback);
return $pagination;
}
/**
* @param array $data
* @return bool
* @throws Exception
*/
public function batchInsert(array $data): bool
{
[$sql, $params] = $this->builder->insert($data, true);
/**
* @param string $field
* @param string $setKey
*
* @return array|null
* @throws Exception
*/
public function column(string $field, string $setKey = ''): ?array
{
return $this->all()->column($field, $setKey);
}
/**
* @return array|Collection
* @throws
*/
public function all(): Collection|array
{
$data = $this->execute($this->builder->all())->all();
if (!empty($this->with)) {
$this->getWith($this->modelClass);
}
$collect = new Collection($this, $data, $this->modelClass);
if ($this->asArray) {
return $collect->toArray();
}
return $collect;
}
/**
* @param $data
* @return ModelInterface
* @throws Exception
*/
public function populate($data): ModelInterface
{
return $this->getWith($this->modelClass::populate($data));
}
return $this->execute($sql, $params)->exec();
}
/**
* @param ModelInterface $model
* @return ModelInterface
*/
public function getWith(ModelInterface $model): ModelInterface
{
if (empty($this->with) || !is_array($this->with)) {
return $model;
}
return $model->setWith($this->with);
}
/**
* @param $filed
*
* @return null
* @throws Exception
*/
public function value($filed)
{
return $this->first()[$filed] ?? null;
}
/**
* @return bool
* @throws Exception
*/
public function exists(): bool
{
return !empty($this->execute($this->builder->one())->fetchColumn());
}
/**
* @return int
* @throws Exception
*/
public function count(): int
{
$data = $this->execute($this->builder->count())->one();
if ($data && is_array($data)) {
return (int)array_shift($data);
}
return 0;
}
/**
* @param bool $getSql
* @return int|bool|string|null
* @throws Exception
*/
public function delete(bool $getSql = false): int|bool|string|null
{
$sql = $this->builder->delete();
if ($getSql === false) {
return $this->execute($sql)->delete();
}
return $sql;
}
/**
* @param array $data
* @return array|Command|bool|int|string
* @throws Exception
*/
public function batchUpdate(array $data): Command|array|bool|int|string
{
$generate = $this->builder->update($data);
if (is_bool($generate)) {
return $generate;
}
return $this->execute(...$generate)->exec();
}
/**
* @param array $data
* @return bool
* @throws Exception
*/
public function batchInsert(array $data): bool
{
[$sql, $params] = $this->builder->insert($data, TRUE);
return $this->execute($sql, $params)->exec();
}
/**
* @param $filed
*
* @return null
* @throws Exception
*/
public function value($filed)
{
return $this->first()[$filed] ?? NULL;
}
/**
* @return bool
* @throws Exception
*/
public function exists(): bool
{
return !empty($this->execute($this->builder->one())->fetchColumn());
}
/**
* @param bool $getSql
* @return int|bool|string|null
* @throws Exception
*/
public function delete(bool $getSql = FALSE): int|bool|string|null
{
$sql = $this->builder->delete();
if ($getSql === FALSE) {
return $this->execute($sql)->delete();
}
return $sql;
}
}
+939 -938
View File
File diff suppressed because it is too large Load Diff
+6 -12
View File
@@ -75,24 +75,18 @@ class Connection extends Component
public Schema $_schema;
/**
* @var EventProvider
*/
#[Inject(EventProvider::class)]
public EventProvider $eventProvider;
/**
* execute by __construct
* @throws Exception
*/
public function init()
{
$this->eventProvider->on(OnWorkerStop::class, [$this, 'clear_connection'], 0);
$this->eventProvider->on(OnWorkerExit::class, [$this, 'clear_connection'], 0);
$this->eventProvider->on(BeginTransaction::class, [$this, 'beginTransaction'], 0);
$this->eventProvider->on(Rollback::class, [$this, 'rollback'], 0);
$this->eventProvider->on(Commit::class, [$this, 'commit'], 0);
$eventProvider = Kiri::getDi()->get(EventProvider::class);
$eventProvider->on(OnWorkerStop::class, [$this, 'clear_connection'], 0);
$eventProvider->on(OnWorkerExit::class, [$this, 'clear_connection'], 0);
$eventProvider->on(BeginTransaction::class, [$this, 'beginTransaction'], 0);
$eventProvider->on(Rollback::class, [$this, 'rollback'], 0);
$eventProvider->on(Commit::class, [$this, 'commit'], 0);
if (Db::transactionsActive()) {
$this->beginTransaction();
-8
View File
@@ -11,7 +11,6 @@ use Kiri\Application;
use Kiri\Events\EventProvider;
use Kiri\Exception\ConfigException;
use Kiri\Kiri;
use Note\Inject;
use Server\Events\OnWorkerStart;
/**
@@ -24,13 +23,6 @@ class DatabasesProviders extends Providers
private array $_pooLength = ['min' => 0, 'max' => 1];
/**
* @var EventProvider
*/
#[Inject(EventProvider::class)]
public EventProvider $eventProvider;
/**
* @param Application $application
* @throws Exception