Error202
/
yii2-bootstrap
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

AR WIP

tags/2.0.0-beta
Qiang Xue 12 years ago
parent
36f7302515
commit
58490ae4ff
2 changed files with 134 additions and 74 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 126
      framework/db/ar/ActiveQuery.php
  2. 82
      framework/db/ar/ActiveRecord.php

126
framework/db/ar/ActiveQuery.php
Unescape View File

@ -10,6 +10,9 @@
namespace yii\db\ar; namespace yii\db\ar;
use yii\db\dao\Connection;
use yii\db\dao\Command;
use yii\db\dao\QueryBuilder;
use yii\db\dao\BaseQuery; use yii\db\dao\BaseQuery;
use yii\base\VectorIterator; use yii\base\VectorIterator;
use yii\db\dao\Expression; use yii\db\dao\Expression;
@ -24,10 +27,10 @@ class ActiveQuery extends BaseQuery
/** /**
* @var array list of relations that this query should be performed with * @var array list of relations that this query should be performed with
*/ */
public $with = array(); public $with;
/** /**
* @var string the name of the column that the result should be indexed by. * @var string the name of the column by which the query result should be indexed.
* This is only useful when the query result is returned as an array. * This is only used when the query result is returned as an array when calling [[all()]].
*/ */
public $index; public $index;
/** /**
@ -38,7 +41,7 @@ class ActiveQuery extends BaseQuery
/** /**
* @var array list of scopes that should be applied to this query * @var array list of scopes that should be applied to this query
*/ */
public $scopes = array(); public $scopes;
/** /**
* @var string the SQL statement to be executed for retrieving AR records. * @var string the SQL statement to be executed for retrieving AR records.
* This is set by [[ActiveRecord::findBySql()]]. * This is set by [[ActiveRecord::findBySql()]].
@ -61,19 +64,30 @@ class ActiveQuery extends BaseQuery
*/ */
public function all() public function all()
{ {
return $this->find(); $command = $this->createCommand();
$rows = $command->queryAll();
$models = $this->createModels($rows);
if (empty($this->with)) {
return $models;
}
} }
/** /**
* Executes query and returns a single row of result. * Executes query and returns a single row of result.
* @return null|array|ActiveRecord the single row of query result. Depending on the setting of [[asArray]], * @return ActiveRecord|array|boolean a single row of query result. Depending on the setting of [[asArray]],
* the query result may be either an array or an ActiveRecord object. Null will be returned * the query result may be either an array or an ActiveRecord object. False will be returned
* if the query results in nothing. * if the query results in nothing.
*/ */
public function one() public function one()
{ {
$records = $this->find(); $command = $this->createCommand();
return isset($records[0]) ? $records[0] : null; $row = $command->queryRow();
if ($row === false || $this->asArray) {
return $row;
} else {
$class = $this->modelClass;
return $class::create($row);
}
} }
/** /**
@ -84,7 +98,7 @@ class ActiveQuery extends BaseQuery
*/ */
public function value() public function value()
{ {
return $this->createFinder()->find($this, true); return $this->createCommand()->queryScalar();
} }
/** /**
@ -97,14 +111,38 @@ class ActiveQuery extends BaseQuery
} }
/** /**
* Returns the database connection used by this query. * Creates a DB command that can be used to execute this query.
* This method returns the connection used by the [[modelClass]]. * @param Connection $db the database connection used to generate the SQL statement.
* @return \yii\db\dao\Connection the database connection used by this query * If this parameter is not given, the `db` application component will be used.
* @return Command the created DB command instance.
*/ */
public function getDbConnection() public function createCommand($db = null)
{ {
$class = $this->modelClass; $modelClass = $this->modelClass;
return $class::getDbConnection(); if ($db === null) {
$db = $modelClass::getDbConnection();
}
if ($this->sql === null) {
if ($this->from === null) {
$tableName = $modelClass::tableName();
$this->from = array($tableName);
}
if (!empty($this->scopes)) {
foreach ($this->scopes as $name => $config) {
if (is_integer($name)) {
$modelClass::$config($this);
} else {
array_unshift($config, $this);
call_user_func_array(array($modelClass, $name), $config);
}
}
}
/** @var $qb QueryBuilder */
$qb = $db->getQueryBuilder();
return $db->createCommand($qb->build($this), $this->params);
} else {
return $db->createCommand($this->sql, $this->params);
}
} }
public function asArray($value = true) public function asArray($value = true)
@ -135,45 +173,24 @@ class ActiveQuery extends BaseQuery
return $this; return $this;
} }
/**
* Sets the parameters about query caching.
* This is a shortcut method to {@link CDbConnection::cache()}.
* It changes the query caching parameter of the {@link dbConnection} instance.
* @param integer $duration the number of seconds that query results may remain valid in cache.
* If this is 0, the caching will be disabled.
* @param \yii\caching\Dependency $dependency the dependency that will be used when saving the query results into cache.
* @param integer $queryCount number of SQL queries that need to be cached after calling this method. Defaults to 1,
* meaning that the next SQL query will be cached.
* @return ActiveRecord the active record instance itself.
*/
public function cache($duration, $dependency = null, $queryCount = 1)
{
$this->getDbConnection()->cache($duration, $dependency, $queryCount);
return $this;
}
protected function find() protected function find()
{ {
$modelClass = $this->modelClass; $modelClass = $this->modelClass;
/** /**
* @var ActiveRecord $model
*/
$model = $modelClass::model();
/**
* @var \yii\db\dao\Connection $db * @var \yii\db\dao\Connection $db
*/ */
$db = $model->getDbConnection(); $db = $modelClass::getDbConnection();
if ($this->sql === null) { if ($this->sql === null) {
if ($this->from === null) { if ($this->from === null) {
$tableName = $model->getTableSchema()->name; $tableName = $modelClass::getTableSchema()->name;
$this->from = array($tableName); $this->from = array($tableName);
} }
foreach ($this->scopes as $name => $config) { foreach ($this->scopes as $name => $config) {
if (is_integer($name)) { if (is_integer($name)) {
$model->$config($this); $modelClass::$config($this);
} else { } else {
array_unshift($config, $this); array_unshift($config, $this);
call_user_func_array(array($model, $name), $config); call_user_func_array(array($modelClass, $name), $config);
} }
} }
$this->sql = $db->getQueryBuilder()->build($this); $this->sql = $db->getQueryBuilder()->build($this);
@ -204,4 +221,31 @@ class ActiveQuery extends BaseQuery
return $records; return $records;
} }
protected function createModels($rows)
{
$models = array();
if ($this->asArray) {
if ($this->index === null) {
return $rows;
}
foreach ($rows as $row) {
$models[$row[$this->index]] = $row;
}
} else {
/** @var $class ActiveRecord */
$class = $this->modelClass;
if ($this->index === null) {
foreach ($rows as $row) {
$models[] = $class::create($row);
}
} else {
foreach ($rows as $row) {
$model = $class::create($row);
$models[$model->{$this->index}] = $model;
}
}
}
return $models;
}
} }

82
framework/db/ar/ActiveRecord.php
Unescape View File

@ -299,10 +299,15 @@ abstract class ActiveRecord extends Model
} elseif (isset($this->getTableSchema()->columns[$name])) { } elseif (isset($this->getTableSchema()->columns[$name])) {
return null; return null;
} elseif (method_exists($this, $name)) { } elseif (method_exists($this, $name)) {
// lazy loading related records // related records
/** @var $query ActiveRelation */ if (isset($this->_related[$name]) || array_key_exists($name, $this->_related)) {
$query = $this->$name(); return $this->_related[$name];
return $this->_attributes[$name] = $query->multiple ? $query->all() : $query->one(); } else {
// lazy loading related records
/** @var $relation ActiveRelation */
$relation = $this->$name();
return $this->_related[$name] = $relation->multiple ? $relation->all() : $relation->one();
}
} else { } else {
return parent::__get($name); return parent::__get($name);
} }
@ -316,8 +321,10 @@ abstract class ActiveRecord extends Model
*/ */
public function __set($name, $value) public function __set($name, $value)
{ {
if (isset($this->getTableSchema()->columns[$name]) || method_exists($this, $name)) { if (isset($this->getTableSchema()->columns[$name])) {
$this->_attributes[$name] = $value; $this->_attributes[$name] = $value;
} elseif (method_exists($this, $name)) {
$this->_related[$name] = $value;
} else { } else {
parent::__set($name, $value); parent::__set($name, $value);
} }
@ -334,8 +341,7 @@ abstract class ActiveRecord extends Model
{ {
if (isset($this->_attributes[$name]) || isset($this->_related[$name])) { if (isset($this->_attributes[$name]) || isset($this->_related[$name])) {
return true; return true;
} } elseif (isset($this->getTableSchema()->columns[$name]) || method_exists($this, $name)) {
if (isset($this->getTableSchema()->columns[$name]) || method_exists($this, $name)) {
return false; return false;
} else { } else {
return parent::__isset($name); return parent::__isset($name);
@ -359,36 +365,46 @@ abstract class ActiveRecord extends Model
} }
} }
public function hasOne($class, $link) /**
{ * Declares a `has-one` relation.
return new HasOneRelation(array( * The declaration is returned in terms of an [[ActiveRelation]] instance
'modelClass' => $class, * through which the related record can be queried and retrieved back.
'parentClass' => get_class($this), * @param string $class the class name of the related record
'parentRecords' => array($this), * @param array $link the primary-foreign key constraint. The keys of the array refer to
'link' => $link, * the columns in the table associated with the `$class` model, while the values of the
)); * array refer to the corresponding columns in the table associated with this AR class.
} * @param array $properties additional property values that should be used to
* initialize the newly created relation object.
public function hasMany($class, $link) * @return ActiveRelation the relation object.
*/
public function hasOne($class, $link, $properties = array())
{ {
return new HasManyRelation(array( $properties['primaryModel'] = $this;
'modelClass' => $class, $properties['modelClass'] = $class;
'parentClass' => get_class($this), $properties['link'] = $link;
'parentRecords' => array($this), $properties['multiple'] = false;
'link' => $link, return new ActiveRelation($properties);
));
} }
public function manyMany($class, $leftLink, $joinTable, $rightLink) /**
* Declares a `has-many` relation.
* The declaration is returned in terms of an [[ActiveRelation]] instance
* through which the related record can be queried and retrieved back.
* @param string $class the class name of the related record
* @param array $link the primary-foreign key constraint. The keys of the array refer to
* the columns in the table associated with the `$class` model, while the values of the
* array refer to the corresponding columns in the table associated with this AR class.
* @param array $properties additional property values that should be used to
* initialize the newly created relation object.
* @return ActiveRelation the relation object.
*/
public function hasMany($class, $link, $properties = array())
{ {
return new ManyManyRelation(array( $properties['primaryModel'] = $this;
'modelClass' => $class, $properties['modelClass'] = $class;
'parentClass' => get_class($this), $properties['link'] = $link;
'parentRecords' => array($this), $properties['multiple'] = true;
'leftLink' => $leftLink, return new ActiveRelation($properties);
'joinTable' => $joinTable,
'rightLink' => $rightLink,
));
} }
/** /**

Write Preview
Loading…
Cancel
Save