kuvalda-db/ActiveRecord.php

122 lines
3.6 KiB
PHP

<?php
namespace dominion\db;
use Yii;
/**
*
*/
class ActiveRecord extends \yii\db\ActiveRecord
{
protected static $tableSchemaCache = [];
/**
* этот класс позволяет ускорить работу за счет хранения структуры бд
*/
public static function getTableSchema()
{
if(!isset(self::$tableSchemaCache[static::tableName()]))
{
self::$tableSchemaCache[static::tableName()] = parent::getTableSchema();
}
return self::$tableSchemaCache[static::tableName()];
}
/**
* этот класс позволяет ускорить работу за счет отказа от получения списка атрибутов из схемы бд
* /
public function attributes()
{
return array_keys($this->attributeLabels());
}*/
public function init()
{
}
public function insertUpdate($runValidation = true, $attributes = null)
{
if ($runValidation && !$this->validate($attributes)) {
Yii::info('Model not inserted due to validation error.', __METHOD__);
return false;
}
if (!$this->isTransactional(self::OP_INSERT)) {
return $this->insertUpdateInternal($attributes);
}
$transaction = static::getDb()->beginTransaction();
try {
$result = $this->insertUpdateInternal($attributes);
if ($result === false) {
$transaction->rollBack();
} else {
$transaction->commit();
}
return $result;
} catch (\Exception $e) {
$transaction->rollBack();
throw $e;
} catch (\Throwable $e) {
$transaction->rollBack();
throw $e;
}
}
protected function insertUpdateInternal($attributes = null)
{
if (!$this->beforeSave(true)) {
return false;
}
$values = $this->getDirtyAttributes($attributes);
if (($primaryKeys = $this->schemaInsertUpdate(static::tableName(), $values)) === false) {
return false;
}
foreach ($primaryKeys as $name => $value) {
$id = static::getTableSchema()->columns[$name]->phpTypecast($value);
$this->setAttribute($name, $id);
$values[$name] = $id;
}
$changedAttributes = array_fill_keys(array_keys($values), null);
$this->setOldAttributes($values);
$this->afterSave(true, $changedAttributes);
return true;
}
public function schemaInsertUpdate($table, $columns)
{
$db = static::getDb();
foreach ($this->attributes() as $itemColumn) {
$column = $db->getSchema()->quoteColumnName($itemColumn);
$onDuplicateKeyValues[] = $column . ' = VALUES(' . $column . ')';
}
$command = $db->createCommand()->insert($table, $columns);
$sql = $command->getRawSql();
$sql .= ' ON DUPLICATE KEY UPDATE ' . implode(', ', $onDuplicateKeyValues);;
if (!$db->createCommand($sql)->execute()) {
return false;
}
$tableSchema = $this->getTableSchema($table);
$result = [];
foreach ($tableSchema->primaryKey as $name) {
if ($tableSchema->columns[$name]->autoIncrement) {
$result[$name] = $this->getLastInsertID($tableSchema->sequenceName);
break;
}
$result[$name] = isset($columns[$name]) ? $columns[$name] : $tableSchema->columns[$name]->defaultValue;
}
return $result;
}
}