php/yii2
1
0
Fork 0
mirror of https://github.com/yiisoft/yii2.git synced 2025-11-02 21:41:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
Qiang Xue
2012-01-28 11:49:21 -05:00
parent 8a5a9ce913
commit f62d062269
2 changed files with 82 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

163
framework/db/ar/ActiveQuery.php
View File

@ -29,25 +29,26 @@ class ActiveQuery extends BaseQuery implements \IteratorAggregate, \ArrayAccess,
public $indexBy; public $indexBy;
public $asArray; public $asArray;
private $_count; public $records;
private $_sql; public $sql;
private $_countSql;
private $_records;
public function all() public function all($refresh = false)
{ {
return $this->performQuery(); if ($this->records === null || $refresh) {
$this->records = $this->performQuery();
}
return $this->records;
} }
public function one() public function one($refresh = false)
{ {
$this->limit = 1; if ($this->records === null || $refresh) {
$records = $this->performQuery(); $this->limit = 1;
if (isset($records[0])) { $this->records = $this->performQuery();
$this->_count = 1; }
return $records[0]; if (isset($this->records[0])) {
return $this->records[0];
} else { } else {
$this->_count = 0;
return null; return null;
} }
} }
@ -76,45 +77,15 @@ class ActiveQuery extends BaseQuery implements \IteratorAggregate, \ArrayAccess,
return $this; return $this;
} }
protected function performQuery() /**
* Returns the database connection used by this query.
* This method returns the connection used by the [[modelClass]].
* @return \yii\db\dao\Connection the database connection used by this query
*/
public function getDbConnection()
{ {
$class = $this->modelClass; $class = $this->modelClass;
$db = $class::getDbConnection(); return $class::getDbConnection();
$this->_sql = $this->getSql($db);
$command = $db->createCommand($this->_sql);
$command->bindValues($this->params);
$rows = $command->queryAll();
if ($this->_asArray) {
$records = $rows;
} else {
$records = array();
foreach ($rows as $row) {
$records[] = $class::populateRecord($row);
}
}
$this->_count = count($records);
return $records;
}
//
// public function getSql($connection = null)
// {
//
// }
public function setSql($value)
{
$this->_sql = $value;
}
public function getCountSql()
{
}
public function getOneSql()
{
} }
/** /**
@ -123,23 +94,7 @@ class ActiveQuery extends BaseQuery implements \IteratorAggregate, \ArrayAccess,
*/ */
public function getCount() public function getCount()
{ {
if ($this->_count !== null) { return $this->count();
return $this->_count;
} else {
return $this->_count = $this->performCountQuery();
}
}
protected function performCountQuery()
{
$select = $this->select;
$this->select = 'COUNT(*)';
$class = $this->modelClass;
$command = $this->createCommand($class::getDbConnection());
$this->_countSql = $command->getSql();
$count = $command->queryScalar();
$this->select = $select;
return $count;
} }
/** /**
@ -155,7 +110,7 @@ class ActiveQuery extends BaseQuery implements \IteratorAggregate, \ArrayAccess,
*/ */
public function cache($duration, $dependency = null, $queryCount = 1) public function cache($duration, $dependency = null, $queryCount = 1)
{ {
$this->connection->cache($duration, $dependency, $queryCount); $this->getDbConnection()->cache($duration, $dependency, $queryCount);
return $this; return $this;
} }
@ -167,19 +122,30 @@ class ActiveQuery extends BaseQuery implements \IteratorAggregate, \ArrayAccess,
*/ */
public function getIterator() public function getIterator()
{ {
$records = $this->performQuery(); if ($this->records === null) {
return new VectorIterator($records); $this->records = $this->performQuery();
}
return new VectorIterator($this->records);
} }
/** /**
* Returns the number of items in the vector. * Returns the number of items in the vector.
* This method is required by the SPL `Countable` interface. * This method is required by the SPL `Countable` interface.
* It will be implicitly called when you use `count($vector)`. * It will be implicitly called when you use `count($vector)`.
* @param boolean $bySql whether to get the count by performing a SQL COUNT query.
* If this is false, it will count the number of records brought back by this query.
* @return integer number of items in the vector. * @return integer number of items in the vector.
*/ */
public function count() public function count($bySql = false)
{ {
return $this->getCount(); if ($bySql) {
return $this->performCountQuery();
} else {
if ($this->records === null) {
$this->records = $this->performQuery();
}
return count($this->records);
}
} }
/** /**
@ -191,10 +157,10 @@ class ActiveQuery extends BaseQuery implements \IteratorAggregate, \ArrayAccess,
*/ */
public function offsetExists($offset) public function offsetExists($offset)
{ {
if ($this->_records === null) { if ($this->records === null) {
$this->_records = $this->performQuery(); $this->records = $this->performQuery();
} }
return isset($this->_records[$offset]); return isset($this->records[$offset]);
} }
/** /**
@ -208,10 +174,10 @@ class ActiveQuery extends BaseQuery implements \IteratorAggregate, \ArrayAccess,
*/ */
public function offsetGet($offset) public function offsetGet($offset)
{ {
if ($this->_records === null) { if ($this->records === null) {
$this->_records = $this->performQuery(); $this->records = $this->performQuery();
} }
return isset($this->_records[$offset]) ? $this->_records[$offset] : null; return isset($this->records[$offset]) ? $this->records[$offset] : null;
} }
/** /**
@ -227,10 +193,10 @@ class ActiveQuery extends BaseQuery implements \IteratorAggregate, \ArrayAccess,
*/ */
public function offsetSet($offset, $item) public function offsetSet($offset, $item)
{ {
if ($this->_records === null) { if ($this->records === null) {
$this->_records = $this->performQuery(); $this->records = $this->performQuery();
} }
$this->_records[$offset] = $item; $this->records[$offset] = $item;
} }
/** /**
@ -243,9 +209,38 @@ class ActiveQuery extends BaseQuery implements \IteratorAggregate, \ArrayAccess,
*/ */
public function offsetUnset($offset) public function offsetUnset($offset)
{ {
if ($this->_records === null) { if ($this->records === null) {
$this->_records = $this->performQuery(); $this->records = $this->performQuery();
} }
unset($this->_records[$offset]); unset($this->records[$offset]);
}
protected function performQuery()
{
$db = $this->getDbConnection();
$this->sql = $this->getSql($db);
$command = $db->createCommand($this->sql);
$command->bindValues($this->params);
$rows = $command->queryAll();
if ($this->asArray) {
$records = $rows;
} else {
$records = array();
$class = $this->modelClass;
foreach ($rows as $row) {
$records[] = $class::populateData($row);
}
}
return $records;
}
protected function performCountQuery()
{
$this->select = 'COUNT(*)';
$class = $this->modelClass;
$command = $this->createCommand($class::getDbConnection());
$this->sql = $command->getSql();
$count = $command->queryScalar();
return $count;
} }
} }

6
framework/db/ar/ActiveRecord.php
View File

@ -308,7 +308,7 @@ abstract class ActiveRecord extends \yii\base\Model
*/ */
public function __construct($scenario = 'insert') public function __construct($scenario = 'insert')
{ {
if ($scenario === null) // internally used by populateRecord() and model() if ($scenario === null) // internally used by populateData() and model()
{ {
return; return;
} }
@ -1186,7 +1186,7 @@ abstract class ActiveRecord extends \yii\base\Model
* @return ActiveRecord the newly created active record. The class of the object is the same as the model class. * @return ActiveRecord the newly created active record. The class of the object is the same as the model class.
* Null is returned if the input data is false. * Null is returned if the input data is false.
*/ */
public static function populateRecord($row) public static function populateData($row)
{ {
$record = static::instantiate($row); $record = static::instantiate($row);
$record->setScenario('update'); $record->setScenario('update');
@ -1204,7 +1204,7 @@ abstract class ActiveRecord extends \yii\base\Model
/** /**
* Creates an active record instance. * Creates an active record instance.
* This method is called by {@link populateRecord} and {@link populateRecords}. * This method is called by {@link populateData}.
* You may override this method if the instance being created * You may override this method if the instance being created
* depends the attributes that are to be populated to the record. * depends the attributes that are to be populated to the record.
* For example, by creating a record based on the value of a column, * For example, by creating a record based on the value of a column,