Qiang Xue
13 years ago
13 changed files with 419 additions and 174 deletions
@ -0,0 +1,176 @@
|
||||
<?php |
||||
/** |
||||
* ActiveQuery class file. |
||||
* |
||||
* @author Qiang Xue <qiang.xue@gmail.com> |
||||
* @link http://www.yiiframework.com/ |
||||
* @copyright Copyright © 2008-2012 Yii Software LLC |
||||
* @license http://www.yiiframework.com/license/ |
||||
*/ |
||||
|
||||
namespace yii\db\ar; |
||||
|
||||
use yii\base\VectorIterator; |
||||
use yii\db\dao\Query; |
||||
use yii\db\Exception; |
||||
|
||||
|
||||
class JoinElement extends \yii\base\Object |
||||
{ |
||||
/** |
||||
* @var ActiveRelation |
||||
*/ |
||||
public $relation; |
||||
/** |
||||
* @var JoinElement |
||||
*/ |
||||
public $parent; |
||||
/** |
||||
* @var JoinElement[] |
||||
*/ |
||||
public $children = array(); |
||||
|
||||
public $columnAliases = array(); // alias => original name |
||||
public $pkAlias = array(); // original name => alias |
||||
|
||||
public $records; |
||||
public $relatedRecords; |
||||
|
||||
public function __construct($parent, $relation) |
||||
{ |
||||
if ($parent !== null) { |
||||
$this->parent = $parent; |
||||
$this->relation = $relation; |
||||
$parent->children[$relation->name] = $this; |
||||
} |
||||
} |
||||
|
||||
public function populateData($row) |
||||
{ |
||||
$pk = array(); |
||||
foreach ($this->pkAlias as $alias) { |
||||
if (isset($row[$alias])) { |
||||
$pk[] = $row[$alias]; |
||||
} else { |
||||
return null; |
||||
} |
||||
} |
||||
$pk = count($pk) === 1 ? $pk[0] : serialize($pk); |
||||
|
||||
// create active record |
||||
if (isset($this->records[$pk])) { |
||||
$record = $this->records[$pk]; |
||||
} else { |
||||
$attributes = array(); |
||||
foreach ($row as $alias => $value) { |
||||
if (isset($this->columnAliases[$alias])) { |
||||
$attributes[$this->columnAliases[$alias]] = $value; |
||||
} |
||||
} |
||||
$modelClass = $this->relation->modelClass; |
||||
$record = $modelClass::populateRecord($attributes); |
||||
foreach ($this->children as $child) { |
||||
if ($child->relation->select !== false) { |
||||
$record->initRelation($child->relation); |
||||
} |
||||
} |
||||
$this->records[$pk] = $record; |
||||
} |
||||
|
||||
// populate child records |
||||
foreach ($this->children as $child) { |
||||
if ($child->relation->select === false) { |
||||
continue; |
||||
} |
||||
$childRecord = $child->populateData($row); |
||||
if ($childRecord === null) { |
||||
continue; |
||||
} |
||||
if ($child->relation->hasMany) { |
||||
$fpk = serialize($childRecord->getPrimaryKey()); |
||||
if (isset($this->relatedRecords[$pk][$child->relation->name][$fpk])) { |
||||
continue; |
||||
} |
||||
$this->relatedRecords[$pk][$child->relation->name][$fpk] = true; |
||||
} |
||||
$record->addRelatedRecord($child->relation, $childRecord); |
||||
} |
||||
|
||||
return $record; |
||||
} |
||||
|
||||
public function buildQuery($query) |
||||
{ |
||||
$tokens = array( |
||||
'@.' => $this->relation->tableAlias, |
||||
'?.' => $this->parent->relation->tableAlias, |
||||
); |
||||
foreach ($this->buildSelect() as $column) { |
||||
$query->select[] = strtr($column, $tokens); |
||||
} |
||||
|
||||
if ($this->relation->where !== null) { |
||||
$query->where[] = strtr($this->relation->where, $tokens); |
||||
} |
||||
|
||||
if ($this->relation->having !== null) { |
||||
$query->having[] = strtr($this->relation->having, $tokens); |
||||
} |
||||
|
||||
/* |
||||
* joinType; |
||||
on; |
||||
via; |
||||
orderby |
||||
groupby |
||||
join |
||||
params |
||||
*/ |
||||
|
||||
foreach ($this->children as $child) { |
||||
$child->buildQuery($query); |
||||
} |
||||
} |
||||
|
||||
public function buildSelect() |
||||
{ |
||||
$modelClass = $this->relation->modelClass; |
||||
$tableSchema = $modelClass::getMetaData()->table; |
||||
$select = $this->relation->select; |
||||
$columns = array(); |
||||
$columnCount = 0; |
||||
if (empty($select) || $select === '*') { |
||||
foreach ($tableSchema->columns as $column) { |
||||
$alias = $this->tableAlias . '_' . ($columnCount++); |
||||
$columns[] = "{$column->name} AS $alias"; |
||||
$this->columnAliases[$alias] = $column->name; |
||||
if ($column->isPrimaryKey) { |
||||
$this->pkAlias[$column->name] = $alias; |
||||
} |
||||
} |
||||
} else { |
||||
if (is_string($select)) { |
||||
$select = explode(',', $select); |
||||
} |
||||
foreach ($tableSchema->primaryKey as $column) { |
||||
$alias = $this->tableAlias . '_' . ($columnCount++); |
||||
$columns[] = "$column AS $alias"; |
||||
$this->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 |
||||
$this->columnAliases[$matches[2]] = $matches[2]; |
||||
$columns[] = $column; |
||||
} elseif (!isset($this->pkAlias[$column])) { |
||||
$alias = $this->tableAlias . '_' . ($columnCount++); |
||||
$columns[] = "$column AS $alias"; |
||||
$this->columnAliases[$alias] = $column; |
||||
} |
||||
} |
||||
} |
||||
|
||||
return $columns; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue