882 lines
16 KiB
PHP
882 lines
16 KiB
PHP
<?php
|
|
/**
|
|
* Created by PhpStorm.
|
|
* User: whwyy
|
|
* Date: 2018/3/30 0030
|
|
* Time: 14:39
|
|
*/
|
|
declare(strict_types=1);
|
|
|
|
namespace Database\Base;
|
|
|
|
|
|
use Annotation\Model\Get;
|
|
use ArrayAccess;
|
|
use HttpServer\Http\Context;
|
|
use ReflectionException;
|
|
use Snowflake\Abstracts\Component;
|
|
use Database\ActiveQuery;
|
|
use Database\ActiveRecord;
|
|
use Database\Connection;
|
|
use Database\HasMany;
|
|
use Database\HasOne;
|
|
use Database\Mysql\Columns;
|
|
use Database\Relation;
|
|
use Exception;
|
|
use Snowflake\Exception\ComponentException;
|
|
use Snowflake\Exception\NotFindClassException;
|
|
use validator\Validator;
|
|
use Database\IOrm;
|
|
use Snowflake\Snowflake;
|
|
|
|
/**
|
|
* Class BOrm
|
|
*
|
|
* @package Snowflake\Abstracts
|
|
*
|
|
* @property bool $isCreate
|
|
* @method rules()
|
|
* @method static tableName()
|
|
*/
|
|
abstract class BaseActiveRecord extends Component implements IOrm, ArrayAccess
|
|
{
|
|
|
|
/** @var array */
|
|
protected array $_attributes = [];
|
|
|
|
/** @var array */
|
|
protected array $_oldAttributes = [];
|
|
|
|
/** @var array */
|
|
protected array $_relate = [];
|
|
|
|
/** @var null|string */
|
|
protected ?string $primary = NULL;
|
|
|
|
|
|
private array $_annotations = [];
|
|
|
|
/**
|
|
* @var bool
|
|
*/
|
|
protected bool $isNewExample = TRUE;
|
|
|
|
protected array $actions = [];
|
|
|
|
protected ?Relation $_relation;
|
|
|
|
/**
|
|
* @throws Exception
|
|
*/
|
|
public function init()
|
|
{
|
|
if (!Context::hasContext(Relation::class)) {
|
|
$relation = Snowflake::createObject(Relation::class);
|
|
$this->_relation = Context::setContext(Relation::class, $relation);
|
|
} else {
|
|
$this->_relation = Context::getContext(Relation::class);
|
|
}
|
|
$this->parseAnnotation();
|
|
}
|
|
|
|
|
|
/**
|
|
* @throws ComponentException
|
|
*/
|
|
private function parseAnnotation(): void
|
|
{
|
|
$attributes = Snowflake::app()->getAttributes();
|
|
$annotations = $attributes->getByClass(static::class);
|
|
|
|
$response = [];
|
|
foreach ($annotations as $annotation) {
|
|
if (!isset($annotation['attributes'])) {
|
|
continue;
|
|
}
|
|
foreach ($annotation['attributes'] as $attribute) {
|
|
if (!($attribute instanceof Get)) {
|
|
continue;
|
|
}
|
|
$response[$attribute->name] = $annotation['handler'];
|
|
}
|
|
}
|
|
$this->_annotations = $response;
|
|
}
|
|
|
|
|
|
/**
|
|
* @param string $column
|
|
* @param int $value
|
|
* @return void
|
|
* @throws Exception
|
|
*/
|
|
public function incrBy(string $column, int $value)
|
|
{
|
|
throw new Exception('Undefined function incrBy in ' . get_called_class());
|
|
}
|
|
|
|
/**
|
|
* @param string $column
|
|
* @param int $value
|
|
* @return void
|
|
* @throws Exception
|
|
*/
|
|
public function decrBy(string $column, int $value)
|
|
{
|
|
throw new Exception('Undefined function decrBy in ' . get_called_class());
|
|
}
|
|
|
|
/**
|
|
* @return array
|
|
*/
|
|
public function getActions(): array
|
|
{
|
|
return $this->actions;
|
|
}
|
|
|
|
/**
|
|
* @return bool
|
|
*/
|
|
public function getIsCreate(): bool
|
|
{
|
|
return $this->isNewExample === TRUE;
|
|
}
|
|
|
|
/**
|
|
* @param bool $bool
|
|
* @return $this
|
|
*/
|
|
public function setIsCreate($bool = FALSE): static
|
|
{
|
|
$this->isNewExample = $bool;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* @return mixed
|
|
*
|
|
* get last exception or other error
|
|
* @throws ComponentException
|
|
*/
|
|
public function getLastError(): mixed
|
|
{
|
|
return Snowflake::app()->getLogger()->getLastError('mysql');
|
|
}
|
|
|
|
/**
|
|
* @return bool
|
|
* @throws Exception
|
|
*/
|
|
public function hasPrimary(): bool
|
|
{
|
|
if ($this->primary !== NULL) {
|
|
return true;
|
|
}
|
|
$primary = static::getColumns()->getPrimaryKeys();
|
|
if (!empty($primary)) {
|
|
return $this->primary = is_array($primary) ? current($primary) : $primary;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* @throws Exception
|
|
*/
|
|
public function hasAutoIncrement(): bool
|
|
{
|
|
$autoIncrement = $this->getAutoIncrement();
|
|
return $autoIncrement !== null;
|
|
}
|
|
|
|
/**
|
|
* @throws Exception
|
|
*/
|
|
public function getAutoIncrement(): int|string|null
|
|
{
|
|
return static::getColumns()->getAutoIncrement();
|
|
}
|
|
|
|
/**
|
|
* @return null|string
|
|
* @throws Exception
|
|
*/
|
|
public function getPrimary(): ?string
|
|
{
|
|
if (!$this->hasPrimary()) {
|
|
return null;
|
|
}
|
|
return $this->primary;
|
|
}
|
|
|
|
/**
|
|
* @return int|null
|
|
* @throws Exception
|
|
*/
|
|
public function getPrimaryValue(): ?int
|
|
{
|
|
if (!$this->hasPrimary()) {
|
|
return null;
|
|
}
|
|
return $this->getAttribute($this->primary);
|
|
}
|
|
|
|
/**
|
|
* @param $param
|
|
* @param null $db
|
|
* @return $this
|
|
* @throws NotFindClassException
|
|
* @throws ReflectionException
|
|
* @throws Exception
|
|
*/
|
|
public static function findOne($param, $db = NULL): static
|
|
{
|
|
if (is_numeric($param)) {
|
|
$primary = static::getColumns()->getPrimaryKeys();
|
|
if (empty($primary)) {
|
|
throw new Exception('Primary key cannot be empty.');
|
|
}
|
|
if (is_array($primary)) {
|
|
$primary = current($primary);
|
|
}
|
|
$param = [$primary => $param];
|
|
}
|
|
return static::find()->where($param)->first();
|
|
}
|
|
|
|
/**
|
|
* @param null $field
|
|
* @return ActiveRecord|null
|
|
* @throws Exception
|
|
* @throws Exception
|
|
*/
|
|
public static function max($field = null): ?ActiveRecord
|
|
{
|
|
$columns = static::getColumns();
|
|
if (empty($field)) {
|
|
$field = $columns->getFirstPrimary();
|
|
}
|
|
$columns = $columns->get_fields();
|
|
if (!isset($columns[$field])) {
|
|
return null;
|
|
}
|
|
$first = static::find()->max($field)->first();
|
|
if (empty($first)) {
|
|
return null;
|
|
}
|
|
return $first[$field];
|
|
}
|
|
|
|
/**
|
|
* @return ActiveQuery
|
|
* @throws ReflectionException
|
|
* @throws NotFindClassException
|
|
*/
|
|
public static function find(): ActiveQuery
|
|
{
|
|
return Snowflake::createObject(ActiveQuery::class, [get_called_class()]);
|
|
}
|
|
|
|
/**
|
|
* @param null $condition
|
|
* @param array $attributes
|
|
*
|
|
* @param bool $if_condition_is_null
|
|
* @return bool
|
|
* @throws Exception
|
|
*/
|
|
public static function deleteByCondition($condition = NULL, $attributes = [], $if_condition_is_null = false): bool
|
|
{
|
|
if (empty($condition)) {
|
|
if (!$if_condition_is_null) {
|
|
return false;
|
|
}
|
|
return static::find()->delete();
|
|
}
|
|
$model = static::find()->ifNotWhere($if_condition_is_null)->where($condition);
|
|
if (!empty($attributes)) {
|
|
$model->bindParams($attributes);
|
|
}
|
|
return $model->delete();
|
|
}
|
|
|
|
|
|
/**
|
|
* @return array
|
|
*/
|
|
public function getAttributes(): array
|
|
{
|
|
return $this->_attributes;
|
|
}
|
|
|
|
/**
|
|
* @return array
|
|
*/
|
|
public function getOldAttributes(): array
|
|
{
|
|
return $this->_oldAttributes;
|
|
}
|
|
|
|
/**
|
|
* @param $name
|
|
* @param $value
|
|
* @return mixed
|
|
*/
|
|
public function setAttribute($name, $value): mixed
|
|
{
|
|
return $this->_attributes[$name] = $value;
|
|
}
|
|
|
|
/**
|
|
* @param $name
|
|
* @param $value
|
|
* @return mixed
|
|
*/
|
|
public function setOldAttribute($name, $value): mixed
|
|
{
|
|
return $this->_oldAttributes[$name] = $value;
|
|
}
|
|
|
|
/**
|
|
* @param array $param
|
|
* @return $this
|
|
* @throws Exception
|
|
*/
|
|
public function setAttributes(array $param): static
|
|
{
|
|
if (empty($param)) {
|
|
return $this;
|
|
}
|
|
foreach ($param as $key => $val) {
|
|
if (!$this->has($key)) {
|
|
$this->setAttribute($key, $val);
|
|
} else {
|
|
$this->$key = $val;
|
|
}
|
|
}
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* @param $param
|
|
* @return $this
|
|
*/
|
|
public function setOldAttributes($param): static
|
|
{
|
|
if (empty($param) || !is_array($param)) {
|
|
return $this;
|
|
}
|
|
foreach ($param as $key => $val) {
|
|
$this->setOldAttribute($key, $val);
|
|
}
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* @return bool
|
|
* @throws Exception
|
|
*/
|
|
public function beforeSave()
|
|
{
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* @param $attributes
|
|
* @param $param
|
|
* @return $this|bool
|
|
* @throws Exception
|
|
*/
|
|
private function insert($param, $attributes): bool|static
|
|
{
|
|
if (empty($param)) {
|
|
return FALSE;
|
|
}
|
|
$dbConnection = static::getDb();
|
|
$change = $dbConnection->getSchema()->getChange();
|
|
$sqlBuilder = $change->insert(static::getTable(), $attributes, $param);
|
|
|
|
$trance = $dbConnection->beginTransaction();
|
|
try {
|
|
$commandExec = $dbConnection->createCommand($sqlBuilder, $param);
|
|
if (($lastId = $commandExec->save(true, $this->hasAutoIncrement())) === false) {
|
|
throw new Exception('保存失败.' . $sqlBuilder);
|
|
}
|
|
$trance->commit();
|
|
$this->setPrimary($lastId, $param);
|
|
$this->afterSave($attributes, $param);
|
|
$lastId = $this->refresh();
|
|
} catch (\Throwable $exception) {
|
|
$trance->rollback();
|
|
$lastId = $this->addError($exception->getMessage(), 'mysql');
|
|
}
|
|
return $lastId;
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $lastId
|
|
* @param $param
|
|
* @return mixed
|
|
* @throws Exception
|
|
*/
|
|
private function setPrimary($lastId, $param): mixed
|
|
{
|
|
if ($this->hasAutoIncrement()) {
|
|
return $this->setAttribute($this->getAutoIncrement(), (int)$lastId);
|
|
}
|
|
if (!$this->hasPrimary()) {
|
|
return $this;
|
|
}
|
|
$primary = $this->getPrimary();
|
|
if (!isset($param[$primary]) || empty($param[$primary])) {
|
|
$this->setAttribute($primary, (int)$lastId);
|
|
}
|
|
return $this->setAttributes($param);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $param
|
|
* @param $condition
|
|
* @param $attributes
|
|
* @return $this|bool
|
|
* @throws Exception
|
|
*/
|
|
private function update($attributes, $condition, $param): bool|static
|
|
{
|
|
if (empty($param)) {
|
|
return true;
|
|
}
|
|
$command = static::getDb();
|
|
$change = $command->getSchema()->getChange();
|
|
$sql = $change->update(static::getTable(), $attributes, $condition, $param);
|
|
|
|
$trance = $command->beginTransaction();
|
|
if (!($command = $command->createCommand($sql, $param)->save(false, $this->hasAutoIncrement()))) {
|
|
$trance->rollback();
|
|
$result = false;
|
|
} else {
|
|
$trance->commit();
|
|
$result = $this->refresh();
|
|
$this->afterSave($attributes, $param);
|
|
}
|
|
return $result;
|
|
}
|
|
|
|
/**
|
|
* @param null $data
|
|
* @return bool|$this
|
|
* @throws Exception
|
|
*/
|
|
public function save($data = NULL): static|bool
|
|
{
|
|
if (is_array($data)) {
|
|
$this->_attributes = array_merge($this->_attributes, $data);
|
|
}
|
|
if (!$this->validator($this->rules()) || !$this->beforeSave()) {
|
|
return false;
|
|
}
|
|
|
|
static::getDb()->enablingTransactions();
|
|
[$attributes, $condition, $param] = $this->filtration_and_separation();
|
|
if (($primary = $this->getPrimary()) !== null) {
|
|
$condition = [$primary => $this->getPrimaryValue()];
|
|
}
|
|
|
|
if (!$this->getIsCreate()) {
|
|
return $this->update($param, $condition, $attributes);
|
|
}
|
|
return $this->insert($attributes, $param);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param array $rule
|
|
* @return bool
|
|
* @throws Exception
|
|
*/
|
|
public function validator(array $rule): bool
|
|
{
|
|
if (empty($rule)) return true;
|
|
$validate = $this->resolve($rule);
|
|
if (!$validate->validation()) {
|
|
return $this->addError($validate->getError(), 'mysql');
|
|
} else {
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param $rule
|
|
* @return Validator
|
|
* @throws Exception
|
|
*/
|
|
private function resolve($rule): Validator
|
|
{
|
|
$validate = Validator::getInstance();
|
|
$validate->setParams($this->_attributes);
|
|
$validate->setModel($this);
|
|
foreach ($rule as $Key => $val) {
|
|
$field = array_shift($val);
|
|
if (empty($val)) {
|
|
continue;
|
|
}
|
|
$validate->make($field, $val);
|
|
}
|
|
return $validate;
|
|
}
|
|
|
|
/**
|
|
* @param string $name
|
|
* @return null
|
|
* @throws Exception
|
|
*/
|
|
public function getAttribute(string $name)
|
|
{
|
|
$method = 'get' . ucfirst($name) . 'Attribute';
|
|
|
|
if (method_exists($this, $method)) {
|
|
return $this->$method($this->_attributes[$name]);
|
|
}
|
|
return $this->_attributes[$name] ?? null;
|
|
}
|
|
|
|
|
|
/**
|
|
* @return array
|
|
* @throws Exception
|
|
*/
|
|
private function filtration_and_separation(): array
|
|
{
|
|
$_tmp = [];
|
|
$condition = [];
|
|
$columns = static::getColumns();
|
|
foreach ($this->_attributes as $key => $val) {
|
|
if ($val === NULL) continue;
|
|
$oldValue = $this->_oldAttributes[$key] ?? null;
|
|
if ($val !== $oldValue) {
|
|
$_tmp[$key] = $columns->fieldFormat($key, $val);
|
|
} else {
|
|
$condition[$key] = $val;
|
|
}
|
|
}
|
|
return [$_tmp, $condition, array_keys($_tmp)];
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $name
|
|
* @param $value
|
|
*/
|
|
public function setRelate($name, $value)
|
|
{
|
|
$this->_relate[$name] = $value;
|
|
}
|
|
|
|
/**
|
|
* @param array $relates
|
|
*/
|
|
public function setRelates(array $relates)
|
|
{
|
|
if (empty($relates)) {
|
|
return;
|
|
}
|
|
foreach ($relates as $key => $val) {
|
|
$this->setRelate($key, $val);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @return array
|
|
*/
|
|
public function getRelates(): array
|
|
{
|
|
return $this->_relate;
|
|
}
|
|
|
|
|
|
/**
|
|
* @return Relation
|
|
*/
|
|
public function getRelation()
|
|
{
|
|
return $this->_relation;
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $name
|
|
* @return mixed
|
|
*/
|
|
public function getRelate($name): mixed
|
|
{
|
|
if (!isset($this->_relate[$name])) {
|
|
return NULL;
|
|
}
|
|
return $this->_relate[$name];
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $attribute
|
|
* @return bool
|
|
* @throws Exception
|
|
*/
|
|
public function has($attribute): bool
|
|
{
|
|
$format = static::getColumns()->format();
|
|
|
|
return array_key_exists($attribute, $format);
|
|
}
|
|
|
|
/**ƒ
|
|
* @return string
|
|
* @throws Exception
|
|
*/
|
|
public static function getTable(): string
|
|
{
|
|
$tablePrefix = static::getDb()->tablePrefix;
|
|
|
|
$table = static::tableName();
|
|
|
|
if (str_starts_with($table, $tablePrefix)) {
|
|
return $table;
|
|
}
|
|
|
|
if (empty($table)) {
|
|
$class = preg_replace('/model\\\/', '', get_called_class());
|
|
$table = lcfirst($class);
|
|
}
|
|
|
|
$table = trim($table, '{{%}}');
|
|
if ($tablePrefix) {
|
|
$table = $tablePrefix . $table;
|
|
}
|
|
return $table;
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $attributes
|
|
* @param $changeAttributes
|
|
* @throws Exception
|
|
*/
|
|
public function afterSave($attributes, $changeAttributes): void
|
|
{
|
|
}
|
|
|
|
/**
|
|
* @return Connection
|
|
* @throws Exception
|
|
*/
|
|
public static function getDb(): Connection
|
|
{
|
|
return static::setDatabaseConnect('db');
|
|
}
|
|
|
|
/**
|
|
* @return static
|
|
*/
|
|
public function refresh(): static
|
|
{
|
|
$this->_oldAttributes = $this->_attributes;
|
|
return $this;
|
|
}
|
|
|
|
/**
|
|
* @param $name
|
|
* @param $value
|
|
* @throws Exception
|
|
*/
|
|
public function __set($name, $value)
|
|
{
|
|
if (!$this->has($name)) {
|
|
parent::__set($name, $value);
|
|
} else {
|
|
$sets = 'set' . ucfirst($name) . 'Attribute';
|
|
if (method_exists($this, $sets)) {
|
|
$value = $this->$sets($value);
|
|
}
|
|
$this->_attributes[$name] = $value;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param $name
|
|
* @return mixed
|
|
* @throws Exception
|
|
*/
|
|
public function __get($name): mixed
|
|
{
|
|
$value = $this->_attributes[$name] ?? null;
|
|
if ($this->hasAnnotation($name)) {
|
|
return call_user_func($this->_annotations[$name], $value);
|
|
}
|
|
if (array_key_exists($name, $this->_attributes)) {
|
|
return static::getColumns()->_decode($name, $value);
|
|
}
|
|
if (isset($this->_relate[$name])) {
|
|
$gets = $this->{$this->_relate[$name]}();
|
|
}
|
|
if (isset($gets)) {
|
|
return $this->resolveClass($gets);
|
|
}
|
|
return parent::__get($name);
|
|
}
|
|
|
|
|
|
/**
|
|
* @return array
|
|
*/
|
|
protected function getAnnotation(): array
|
|
{
|
|
return $this->_annotations;
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $name
|
|
* @return bool
|
|
*/
|
|
protected function hasAnnotation($name): bool
|
|
{
|
|
return isset($this->_annotations[$name]);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $item
|
|
* @param $data
|
|
* @return array
|
|
*/
|
|
protected function resolveAttributes($item, $data): array
|
|
{
|
|
return call_user_func($item, $data);
|
|
}
|
|
|
|
/**
|
|
* @param $name
|
|
* @return bool
|
|
*/
|
|
public function __isset($name): bool
|
|
{
|
|
return isset($this->_attributes[$name]);
|
|
}
|
|
|
|
/**
|
|
* @param $call
|
|
* @return mixed
|
|
* @throws Exception
|
|
*/
|
|
private function resolveClass($call): mixed
|
|
{
|
|
if ($call instanceof HasOne) {
|
|
return $call->get();
|
|
} else if ($call instanceof HasMany) {
|
|
return $call->get();
|
|
} else {
|
|
return $call;
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* @param mixed $offset
|
|
* @return bool
|
|
* @throws Exception
|
|
*/
|
|
public function offsetExists(mixed $offset): bool
|
|
{
|
|
return $this->has($offset);
|
|
}
|
|
|
|
/**
|
|
* @param mixed $offset
|
|
* @return mixed
|
|
* @throws Exception
|
|
*/
|
|
public function offsetGet(mixed $offset): mixed
|
|
{
|
|
return $this->__get($offset);
|
|
}
|
|
|
|
/**
|
|
* @param mixed $offset
|
|
* @param mixed $value
|
|
* @throws Exception
|
|
*/
|
|
public function offsetSet(mixed $offset, mixed $value)
|
|
{
|
|
$this->__set($offset, $value);
|
|
}
|
|
|
|
/**
|
|
* @param mixed $offset
|
|
* @throws Exception
|
|
*/
|
|
public function offsetUnset(mixed $offset)
|
|
{
|
|
if (!$this->has($offset)) {
|
|
return;
|
|
}
|
|
unset($this->_attributes[$offset]);
|
|
unset($this->_oldAttributes[$offset]);
|
|
if (isset($this->_relate)) {
|
|
unset($this->_relate[$offset]);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @return array
|
|
*/
|
|
public function unset(): array
|
|
{
|
|
$fields = func_get_args();
|
|
$fields = array_shift($fields);
|
|
if (!is_array($fields)) {
|
|
$fields = explode(',', $fields);
|
|
}
|
|
|
|
$array = array_combine($fields, $fields);
|
|
|
|
return array_diff_assoc($array, $this->_attributes);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $dbName
|
|
* @return mixed
|
|
* @throws Exception
|
|
*/
|
|
public static function setDatabaseConnect($dbName): Connection
|
|
{
|
|
return Snowflake::app()->db->get($dbName);
|
|
}
|
|
|
|
/**
|
|
* @return Columns
|
|
* @throws Exception
|
|
*/
|
|
public static function getColumns(): Columns
|
|
{
|
|
return static::getDb()->getSchema()
|
|
->getColumns()
|
|
->table(static::getTable());
|
|
}
|
|
|
|
/**
|
|
* @param array $data
|
|
* @return static
|
|
* @throws
|
|
*/
|
|
public static function populate(array $data): static
|
|
{
|
|
$model = new static();
|
|
$model->setAttributes($data);
|
|
$model->setIsCreate(false);
|
|
$model->refresh();
|
|
return $model;
|
|
}
|
|
|
|
}
|