Yii2 Bootstrap 3
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.

91 lines
2.5 KiB

14 years ago
<?php
/**
* @link http://www.yiiframework.com/
* @copyright Copyright (c) 2008 Yii Software LLC
14 years ago
* @license http://www.yiiframework.com/license/
*/
14 years ago
namespace yii\validators;
use Yii;
11 years ago
use yii\db\ActiveRecordInterface;
14 years ago
14 years ago
/**
11 years ago
* UniqueValidator validates that the attribute value is unique in the corresponding database table.
14 years ago
*
* @author Qiang Xue <qiang.xue@gmail.com>
13 years ago
* @since 2.0
14 years ago
*/
13 years ago
class UniqueValidator extends Validator
14 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.
14 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;
/**
* @inheritdoc
*/
public function init()
{
parent::init();
if ($this->message === null) {
$this->message = Yii::t('yii', '{attribute} "{value}" has already been taken.');
}
}
/**
* @inheritdoc
14 years ago
*/
14 years ago
public function validateAttribute($object, $attribute)
14 years ago
{
$value = $object->$attribute;
if (is_array($value)) {
$this->addError($object, $attribute, Yii::t('yii', '{attribute} is invalid.'));
return;
}
/** @var ActiveRecordInterface $className */
$className = $this->className === null ? get_class($object) : $this->className;
$attributeName = $this->attributeName === null ? $attribute : $this->attributeName;
14 years ago
12 years ago
$query = $className::find();
$query->where([$attributeName => $value]);
13 years ago
11 years ago
if (!$object instanceof ActiveRecordInterface || $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()
/** @var ActiveRecordInterface[] $objects */
$objects = $query->limit(2)->all();
14 years ago
$n = count($objects);
13 years ago
if ($n === 1) {
if (in_array($attributeName, $className::primaryKey())) {
13 years ago
// primary key is modified and not unique
14 years ago
$exists = $object->getOldPrimaryKey() != $object->getPrimaryKey();
13 years ago
} else {
14 years ago
// non-primary key, need to exclude the current record based on PK
$exists = $objects[0]->getPrimaryKey() != $object->getOldPrimaryKey();
14 years ago
}
13 years ago
} else {
14 years ago
$exists = $n > 1;
13 years ago
}
14 years ago
}
13 years ago
if ($exists) {
$this->addError($object, $attribute, $this->message);
14 years ago
}
}
}