You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

102 lines
3.0 KiB

13 years ago
<?php
/**
* @link http://www.yiiframework.com/
* @copyright Copyright (c) 2008 Yii Software LLC
13 years ago
* @license http://www.yiiframework.com/license/
*/
13 years ago
namespace yii\validators;
use Yii;
use yii\base\InvalidConfigException;
use yii\db\ActiveRecord;
13 years ago
13 years ago
/**
* CUniqueValidator validates that the attribute value is unique in the corresponding database table.
*
* @author Qiang Xue <qiang.xue@gmail.com>
13 years ago
* @since 2.0
13 years ago
*/
13 years ago
class UniqueValidator extends Validator
13 years ago
{
/**
* @var string the ActiveRecord class name or alias of the class
13 years ago
* that should be used to look for the attribute value being validated.
* Defaults to null, meaning using the ActiveRecord class of the attribute being validated.
13 years ago
* @see attributeName
*/
public $className;
/**
* @var string the ActiveRecord class attribute name that should be
* used to look for the attribute value being validated. Defaults to null,
* meaning using the name of the attribute being validated.
*/
public $attributeName;
/**
* Initializes the validator.
*/
public function init()
{
parent::init();
if ($this->message === null) {
$this->message = Yii::t('yii', '{attribute} "{value}" has already been taken.');
}
}
/**
13 years ago
* Validates the attribute of the object.
* If there is any error, the error message is added to the object.
12 years ago
* @param \yii\db\ActiveRecord $object the object being validated
13 years ago
* @param string $attribute the attribute being validated
* @throws InvalidConfigException if table doesn't have column specified
13 years ago
*/
13 years ago
public function validateAttribute($object, $attribute)
13 years ago
{
$value = $object->$attribute;
if (is_array($value)) {
$this->addError($object, $attribute, Yii::t('yii', '{attribute} is invalid.'));
return;
}
12 years ago
/** @var $className \yii\db\ActiveRecord */
$className = $this->className === null ? get_class($object) : Yii::import($this->className);
$attributeName = $this->attributeName === null ? $attribute : $this->attributeName;
13 years ago
12 years ago
$table = $className::getTableSchema();
13 years ago
if (($column = $table->getColumn($attributeName)) === null) {
throw new InvalidConfigException("Table '{$table->name}' does not have a column named '$attributeName'.");
13 years ago
}
13 years ago
12 years ago
$query = $className::find();
12 years ago
$query->where(array($column->name => $value));
13 years ago
if (!$object instanceof ActiveRecord || $object->getIsNewRecord()) {
// if current $object isn't in the database yet then it's OK just to call exists()
12 years ago
$exists = $query->exists();
13 years ago
} else {
// if current $object is in the database already we can't use exists()
12 years ago
$query->limit(2);
$objects = $query->all();
13 years ago
$n = count($objects);
13 years ago
if ($n === 1) {
if ($column->isPrimaryKey) {
// primary key is modified and not unique
13 years ago
$exists = $object->getOldPrimaryKey() != $object->getPrimaryKey();
13 years ago
} else {
13 years ago
// non-primary key, need to exclude the current record based on PK
$exists = array_shift($objects)->getPrimaryKey() != $object->getOldPrimaryKey();
}
13 years ago
} else {
13 years ago
$exists = $n > 1;
13 years ago
}
13 years ago
}
13 years ago
if ($exists) {
$this->addError($object, $attribute, $this->message);
13 years ago
}
}
}