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.

385 lines
10 KiB

13 years ago
<?php
/**
13 years ago
* ActiveFinder class file.
13 years ago
*
* @author Qiang Xue <qiang.xue@gmail.com>
* @link http://www.yiiframework.com/
* @copyright Copyright &copy; 2008-2012 Yii Software LLC
* @license http://www.yiiframework.com/license/
*/
namespace yii\db\ar;
13 years ago
use yii\base\Object;
13 years ago
use yii\base\VectorIterator;
13 years ago
use yii\db\dao\Query;
13 years ago
use yii\db\Exception;
13 years ago
13 years ago
/**
* ActiveFinder.php is ...
13 years ago
* todo: base limited with has_many, bySQL, lazy loading
* todo: quoting column names in 'on' clause
13 years ago
*
13 years ago
* @property integer $count
*
13 years ago
* @author Qiang Xue <qiang.xue@gmail.com>
* @since 2.0
*/
13 years ago
class ActiveFinder extends \yii\base\Object
13 years ago
{
13 years ago
/**
13 years ago
* @var \yii\db\dao\Connection
13 years ago
*/
13 years ago
public $connection;
13 years ago
13 years ago
public function __construct($connection)
13 years ago
{
13 years ago
$this->connection = $connection;
13 years ago
}
/**
13 years ago
* @param ActiveQuery $query
13 years ago
*/
13 years ago
public function findRecords($query)
13 years ago
{
13 years ago
if (!empty($query->with)) {
return $this->findRecordsWithRelations($query);
}
13 years ago
if ($query->sql !== null) {
$sql = $query->sql;
13 years ago
} else {
13 years ago
if ($query->from === null) {
$modelClass = $query->modelClass;
$tableName = $modelClass::tableName();
if ($query->tableAlias !== null) {
$tableName .= ' ' . $query->tableAlias;
}
$query->from = array($tableName);
}
13 years ago
$this->applyScopes($query);
$sql = $this->connection->getQueryBuilder()->build($query);
if (strpos($sql, '@.') !== false) {
if ($query->tableAlias !== null) {
$alias = $this->connection->quoteTableName($query->tableAlias) . '.';
} else {
$class = $query->modelClass;
$alias = $this->connection->quoteTableName($class::tableName()) . '.';
}
$sql = str_replace('@.', $alias, $sql);
13 years ago
}
13 years ago
}
13 years ago
$command = $this->connection->createCommand($sql, $query->params);
13 years ago
13 years ago
$rows = $command->queryAll();
13 years ago
$records = array();
if ($query->asArray) {
if ($query->indexBy === null) {
return $rows;
}
foreach ($rows as $row) {
$records[$row[$query->indexBy]] = $row;
}
13 years ago
} else {
13 years ago
$class = $query->modelClass;
if ($query->indexBy === null) {
foreach ($rows as $row) {
13 years ago
$records[] = $class::create($row);
13 years ago
}
} else {
foreach ($rows as $row) {
13 years ago
$records[$row[$query->indexBy]] = $class::create($row);
13 years ago
}
}
13 years ago
}
13 years ago
return $records;
13 years ago
}
13 years ago
13 years ago
public function findRelatedRecords($record, $relation, $params)
13 years ago
{
13 years ago
}
13 years ago
private $_joinCount;
private $_tableAliases;
/**
* @param ActiveQuery $query
* @return array
*/
13 years ago
public function findRecordsWithRelations($query)
13 years ago
{
13 years ago
$this->_joinCount = 0;
$this->_tableAliases = array();
$joinTree = new JoinElement($this->_joinCount++, $query, null, null);
$this->buildJoinTree($joinTree, $query->with);
$this->initJoinTree($joinTree);
13 years ago
13 years ago
$q = new Query;
$this->buildJoinQuery($joinTree, $q);
$rows = $q->createCommand($this->connection)->queryAll();
foreach ($rows as $row) {
$joinTree->createRecord($row);
13 years ago
}
13 years ago
return $query->indexBy !== null ? $joinTree->records : array_values($joinTree->records);
13 years ago
}
13 years ago
protected function applyScopes($query)
{
13 years ago
$class = $query->modelClass;
$class::defaultScope($query);
13 years ago
if (is_array($query->scopes)) {
13 years ago
$scopes = $class::scopes();
foreach ($query->scopes as $name => $params) {
if (is_integer($name)) {
$name = $params;
13 years ago
$params = array();
}
13 years ago
if (isset($scopes[$name])) {
array_unshift($params, $query);
call_user_func_array($scopes[$name], $params);
} else {
13 years ago
throw new Exception("$class has no scope named '$name'.");
}
13 years ago
}
}
}
13 years ago
/**
* @param JoinElement $parent
* @param array|string $with
* @param array $config
* @return null|JoinElement
* @throws \yii\db\Exception
*/
13 years ago
protected function buildJoinTree($parent, $with, $config = array())
13 years ago
{
if (is_array($with)) {
foreach ($with as $name => $value) {
13 years ago
if (is_array($value)) {
13 years ago
$this->buildJoinTree($parent, $name, $value);
13 years ago
} else {
$this->buildJoinTree($parent, $value);
13 years ago
}
}
return null;
}
if (($pos = strrpos($with, '.')) !== false) {
13 years ago
$parent = $this->buildJoinTree($parent, substr($with, 0, $pos));
13 years ago
$with = substr($with, $pos + 1);
}
if (isset($parent->children[$with])) {
$child = $parent->children[$with];
13 years ago
$child->joinOnly = false;
13 years ago
} else {
13 years ago
$modelClass = $parent->query->modelClass;
13 years ago
$relations = $modelClass::getMetaData()->relations;
if (!isset($relations[$with])) {
throw new Exception("$modelClass has no relation named '$with'.");
}
$relation = clone $relations[$with];
13 years ago
if ($relation->via !== null && isset($relations[$relation->via])) {
$parent2 = $this->buildJoinTree($parent, $relation->via);
13 years ago
$relation->via = null;
13 years ago
if ($parent2->joinOnly === null) {
$parent2->joinOnly = true;
}
13 years ago
$child = new JoinElement($this->_joinCount++, $relation, $parent2, $parent);
13 years ago
} else {
13 years ago
$child = new JoinElement($this->_joinCount++, $relation, $parent, $parent);
13 years ago
}
}
foreach ($config as $name => $value) {
13 years ago
$child->query->$name = $value;
13 years ago
}
return $child;
}
13 years ago
13 years ago
/**
* @param JoinElement $element
*/
13 years ago
protected function initJoinTree($element)
13 years ago
{
13 years ago
if ($element->query->tableAlias !== null) {
$alias = $element->query->tableAlias;
} elseif ($element->query instanceof ActiveRelation) {
$alias = $element->query->name;
} else {
$alias = 't';
13 years ago
}
13 years ago
$count = 0;
while (isset($this->_tableAliases[$alias])) {
$alias = 't' . $count++;
}
$this->_tableAliases[$alias] = true;
$element->query->tableAlias = $alias;
13 years ago
$this->applyScopes($element->query);
13 years ago
foreach ($element->children as $child) {
13 years ago
$this->initJoinTree($child, $count);
13 years ago
}
}
13 years ago
/**
* @param JoinElement $element
13 years ago
* @param \yii\db\dao\Query $query
13 years ago
*/
protected function buildJoinQuery($element, $query)
{
13 years ago
if ($element->parent) {
$prefixes = array(
'@.' => $element->query->tableAlias . '.',
'?.' => $element->parent->query->tableAlias . '.',
);
$quotedPrefixes = array(
'@.' => $this->connection->quoteTableName($element->query->tableAlias, true) . '.',
'?.' => $this->connection->quoteTableName($element->parent->query->tableAlias, true) . '.',
);
} else {
$prefixes = array(
'@.' => $element->query->tableAlias . '.',
);
$quotedPrefixes = array(
'@.' => $this->connection->quoteTableName($element->query->tableAlias, true) . '.',
);
}
$qb = $this->connection->getQueryBuilder();
13 years ago
foreach ($this->buildSelect($element, $element->query->select) as $column) {
$query->select[] = strtr($column, $prefixes);
}
13 years ago
if ($element->query instanceof ActiveQuery) {
if ($element->query->from === null) {
$modelClass = $element->query->modelClass;
$tableName = $modelClass::tableName();
if ($element->query->tableAlias !== null) {
$tableName .= ' ' . $element->query->tableAlias;
}
$query->from = array($tableName);
} else {
$query->from = $element->query->from;
}
13 years ago
}
13 years ago
if (($where = $qb->buildCondition($element->query->where)) !== '') {
$query->andWhere(strtr($where, $quotedPrefixes));
13 years ago
}
13 years ago
if (($having = $qb->buildCondition($element->query->having)) !== '') {
$query->andHaving(strtr($having, $quotedPrefixes));
13 years ago
}
13 years ago
if ($element->query instanceof ActiveRelation) {
if ($element->query->via !== null) {
$query->join[] = strtr($element->query->via, $quotedPrefixes);
}
if ($element->query->joinType === null) {
$joinType = $element->query->select === false ? 'INNER JOIN' : 'LEFT JOIN';
} else {
$joinType = $element->query->joinType;
}
$modelClass = $element->query->modelClass;
$tableName = $this->connection->quoteTableName($modelClass::tableName());
$tableAlias = $this->connection->quoteTableName($element->query->tableAlias);
$join = "$joinType $tableName $tableAlias";
if ($element->query->on !== null) {
$join .= ' ON ' . strtr($qb->buildCondition($element->query->on), $quotedPrefixes);
}
$query->join[] = $join;
13 years ago
}
if ($element->query->join !== null) {
13 years ago
if (is_array($element->query->join)) {
foreach ($element->query->join as $join) {
if (is_array($join) && isset($join[2])) {
$join[2] = strtr($join[2], $quotedPrefixes);
}
$query->join[] = $join;
}
} else {
$query->join[] = strtr($element->query->join, $quotedPrefixes);
}
13 years ago
}
if ($element->query->orderBy !== null) {
if (!is_array($element->query->orderBy)) {
$element->query->orderBy = preg_split('/\s*,\s*/', trim($element->query->orderBy), -1, PREG_SPLIT_NO_EMPTY);
}
foreach ($element->query->orderBy as $orderBy) {
$query->orderBy[] = strtr($orderBy, $prefixes);
}
}
if ($element->query->groupBy !== null) {
if (!is_array($element->query->groupBy)) {
$element->query->groupBy = preg_split('/\s*,\s*/', trim($element->query->groupBy), -1, PREG_SPLIT_NO_EMPTY);
}
foreach ($element->query->groupBy as $groupBy) {
$query->groupBy[] = strtr($groupBy, $prefixes);
}
}
if ($element->query->params !== null) {
$query->addParams($element->query->params);
}
foreach ($element->children as $child) {
13 years ago
$this->buildJoinQuery($child, $query);
13 years ago
}
}
protected function buildSelect($element, $select)
{
if ($select === false) {
return array();
}
$modelClass = $element->query->modelClass;
$table = $modelClass::getMetaData()->table;
$columns = array();
$columnCount = 0;
$prefix = $element->query->tableAlias;
if (empty($select) || $select === '*') {
foreach ($table->columns as $column) {
13 years ago
$alias = "c{$element->id}_" . ($columnCount++);
13 years ago
$columns[] = "$prefix.{$column->name} AS $alias";
$element->columnAliases[$alias] = $column->name;
if ($column->isPrimaryKey) {
$element->pkAlias[$column->name] = $alias;
}
}
} else {
if (is_string($select)) {
$select = explode(',', $select);
}
foreach ($table->primaryKey as $column) {
13 years ago
$alias = "c{$element->id}_" . ($columnCount++);
13 years ago
$columns[] = "$prefix.$column AS $alias";
$element->pkAlias[$column] = $alias;
}
foreach ($select as $column) {
$column = trim($column);
if (preg_match('/^(.*?)\s+AS\s+(\w+)$/im', $column, $matches)) {
// if the column is already aliased
$element->columnAliases[$matches[2]] = $matches[2];
$columns[] = $column;
} elseif (!isset($element->pkAlias[$column])) {
13 years ago
$alias = "c{$element->id}_" . ($columnCount++);
13 years ago
$columns[] = "$prefix.$column AS $alias";
$element->columnAliases[$alias] = $column;
}
}
}
return $columns;
}
13 years ago
}