327 lines
11 KiB
PHP
327 lines
11 KiB
PHP
<?php
|
|
/*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*
|
|
* This software consists of voluntary contributions made by many individuals
|
|
* and is licensed under the MIT license. For more information, see
|
|
* <http://www.doctrine-project.org>.
|
|
*/
|
|
|
|
namespace Doctrine\ORM\Internal\Hydration;
|
|
|
|
use PDO;
|
|
use Doctrine\ORM\Mapping\ClassMetadata;
|
|
|
|
/**
|
|
* The ArrayHydrator produces a nested array "graph" that is often (not always)
|
|
* interchangeable with the corresponding object graph for read-only access.
|
|
*
|
|
* @since 2.0
|
|
* @author Roman Borschel <roman@code-factory.org>
|
|
* @author Guilherme Blanco <guilhermeblanoc@hotmail.com>
|
|
*/
|
|
class ArrayHydrator extends AbstractHydrator
|
|
{
|
|
/**
|
|
* @var array
|
|
*/
|
|
private $_ce = array();
|
|
|
|
/**
|
|
* @var array
|
|
*/
|
|
private $_rootAliases = array();
|
|
|
|
/**
|
|
* @var bool
|
|
*/
|
|
private $_isSimpleQuery = false;
|
|
|
|
/**
|
|
* @var array
|
|
*/
|
|
private $_identifierMap = array();
|
|
|
|
/**
|
|
* @var array
|
|
*/
|
|
private $_resultPointers = array();
|
|
|
|
/**
|
|
* @var array
|
|
*/
|
|
private $_idTemplate = array();
|
|
|
|
/**
|
|
* @var int
|
|
*/
|
|
private $_resultCounter = 0;
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function prepare()
|
|
{
|
|
$this->_isSimpleQuery = count($this->_rsm->aliasMap) <= 1;
|
|
$this->_identifierMap = array();
|
|
$this->_resultPointers = array();
|
|
$this->_idTemplate = array();
|
|
$this->_resultCounter = 0;
|
|
|
|
foreach ($this->_rsm->aliasMap as $dqlAlias => $className) {
|
|
$this->_identifierMap[$dqlAlias] = array();
|
|
$this->_resultPointers[$dqlAlias] = array();
|
|
$this->_idTemplate[$dqlAlias] = '';
|
|
}
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function hydrateAllData()
|
|
{
|
|
$result = array();
|
|
$cache = array();
|
|
|
|
while ($data = $this->_stmt->fetch(PDO::FETCH_ASSOC)) {
|
|
$this->hydrateRowData($data, $cache, $result);
|
|
}
|
|
|
|
return $result;
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function hydrateRowData(array $row, array &$cache, array &$result)
|
|
{
|
|
// 1) Initialize
|
|
$id = $this->_idTemplate; // initialize the id-memory
|
|
$nonemptyComponents = array();
|
|
$rowData = $this->gatherRowData($row, $cache, $id, $nonemptyComponents);
|
|
|
|
// Extract scalar values. They're appended at the end.
|
|
if (isset($rowData['scalars'])) {
|
|
$scalars = $rowData['scalars'];
|
|
|
|
unset($rowData['scalars']);
|
|
|
|
if (empty($rowData)) {
|
|
++$this->_resultCounter;
|
|
}
|
|
}
|
|
|
|
// 2) Now hydrate the data found in the current row.
|
|
foreach ($rowData as $dqlAlias => $data) {
|
|
$index = false;
|
|
|
|
if (isset($this->_rsm->parentAliasMap[$dqlAlias])) {
|
|
// It's a joined result
|
|
|
|
$parent = $this->_rsm->parentAliasMap[$dqlAlias];
|
|
$path = $parent . '.' . $dqlAlias;
|
|
|
|
// missing parent data, skipping as RIGHT JOIN hydration is not supported.
|
|
if ( ! isset($nonemptyComponents[$parent]) ) {
|
|
continue;
|
|
}
|
|
|
|
// Get a reference to the right element in the result tree.
|
|
// This element will get the associated element attached.
|
|
if ($this->_rsm->isMixed && isset($this->_rootAliases[$parent])) {
|
|
$first = reset($this->_resultPointers);
|
|
// TODO: Exception if $key === null ?
|
|
$baseElement =& $this->_resultPointers[$parent][key($first)];
|
|
} else if (isset($this->_resultPointers[$parent])) {
|
|
$baseElement =& $this->_resultPointers[$parent];
|
|
} else {
|
|
unset($this->_resultPointers[$dqlAlias]); // Ticket #1228
|
|
|
|
continue;
|
|
}
|
|
|
|
$relationAlias = $this->_rsm->relationMap[$dqlAlias];
|
|
$relation = $this->getClassMetadata($this->_rsm->aliasMap[$parent])->associationMappings[$relationAlias];
|
|
|
|
// Check the type of the relation (many or single-valued)
|
|
if ( ! ($relation['type'] & ClassMetadata::TO_ONE)) {
|
|
$oneToOne = false;
|
|
|
|
if (isset($nonemptyComponents[$dqlAlias])) {
|
|
if ( ! isset($baseElement[$relationAlias])) {
|
|
$baseElement[$relationAlias] = array();
|
|
}
|
|
|
|
$indexExists = isset($this->_identifierMap[$path][$id[$parent]][$id[$dqlAlias]]);
|
|
$index = $indexExists ? $this->_identifierMap[$path][$id[$parent]][$id[$dqlAlias]] : false;
|
|
$indexIsValid = $index !== false ? isset($baseElement[$relationAlias][$index]) : false;
|
|
|
|
if ( ! $indexExists || ! $indexIsValid) {
|
|
$element = $data;
|
|
|
|
if (isset($this->_rsm->indexByMap[$dqlAlias])) {
|
|
$baseElement[$relationAlias][$row[$this->_rsm->indexByMap[$dqlAlias]]] = $element;
|
|
} else {
|
|
$baseElement[$relationAlias][] = $element;
|
|
}
|
|
|
|
end($baseElement[$relationAlias]);
|
|
|
|
$this->_identifierMap[$path][$id[$parent]][$id[$dqlAlias]] = key($baseElement[$relationAlias]);
|
|
}
|
|
} else if ( ! isset($baseElement[$relationAlias])) {
|
|
$baseElement[$relationAlias] = array();
|
|
}
|
|
} else {
|
|
$oneToOne = true;
|
|
|
|
if (
|
|
( ! isset($nonemptyComponents[$dqlAlias])) &&
|
|
( ! isset($baseElement[$relationAlias]))
|
|
) {
|
|
$baseElement[$relationAlias] = null;
|
|
} else if (
|
|
( ! isset($baseElement[$relationAlias])) ||
|
|
( ! isset($this->_identifierMap[$path][$id[$parent]][$id[$dqlAlias]]))
|
|
) {
|
|
$baseElement[$relationAlias] = $data;
|
|
}
|
|
}
|
|
|
|
$coll =& $baseElement[$relationAlias];
|
|
|
|
if ($coll !== null) {
|
|
$this->updateResultPointer($coll, $index, $dqlAlias, $oneToOne);
|
|
}
|
|
} else {
|
|
// It's a root result element
|
|
|
|
$this->_rootAliases[$dqlAlias] = true; // Mark as root
|
|
$entityKey = $this->_rsm->entityMappings[$dqlAlias] ?: 0;
|
|
|
|
// if this row has a NULL value for the root result id then make it a null result.
|
|
if ( ! isset($nonemptyComponents[$dqlAlias]) ) {
|
|
$result[] = $this->_rsm->isMixed
|
|
? array($entityKey => null)
|
|
: null;
|
|
|
|
$resultKey = $this->_resultCounter;
|
|
++$this->_resultCounter;
|
|
|
|
continue;
|
|
}
|
|
|
|
// Check for an existing element
|
|
if ($this->_isSimpleQuery || ! isset($this->_identifierMap[$dqlAlias][$id[$dqlAlias]])) {
|
|
$element = $this->_rsm->isMixed
|
|
? array($entityKey => $rowData[$dqlAlias])
|
|
: $rowData[$dqlAlias];
|
|
|
|
if (isset($this->_rsm->indexByMap[$dqlAlias])) {
|
|
$resultKey = $row[$this->_rsm->indexByMap[$dqlAlias]];
|
|
$result[$resultKey] = $element;
|
|
} else {
|
|
$resultKey = $this->_resultCounter;
|
|
$result[] = $element;
|
|
|
|
++$this->_resultCounter;
|
|
}
|
|
|
|
$this->_identifierMap[$dqlAlias][$id[$dqlAlias]] = $resultKey;
|
|
} else {
|
|
$index = $this->_identifierMap[$dqlAlias][$id[$dqlAlias]];
|
|
$resultKey = $index;
|
|
|
|
/*if ($this->_rsm->isMixed) {
|
|
$result[] =& $result[$index];
|
|
++$this->_resultCounter;
|
|
}*/
|
|
}
|
|
|
|
$this->updateResultPointer($result, $index, $dqlAlias, false);
|
|
}
|
|
}
|
|
|
|
// Append scalar values to mixed result sets
|
|
if (isset($scalars)) {
|
|
if ( ! isset($resultKey) ) {
|
|
// this only ever happens when no object is fetched (scalar result only)
|
|
$resultKey = isset($this->_rsm->indexByMap['scalars'])
|
|
? $row[$this->_rsm->indexByMap['scalars']]
|
|
: $this->_resultCounter - 1;
|
|
}
|
|
|
|
foreach ($scalars as $name => $value) {
|
|
$result[$resultKey][$name] = $value;
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Updates the result pointer for an Entity. The result pointers point to the
|
|
* last seen instance of each Entity type. This is used for graph construction.
|
|
*
|
|
* @param array $coll The element.
|
|
* @param boolean|integer $index Index of the element in the collection.
|
|
* @param string $dqlAlias
|
|
* @param boolean $oneToOne Whether it is a single-valued association or not.
|
|
*
|
|
* @return void
|
|
*/
|
|
private function updateResultPointer(array &$coll, $index, $dqlAlias, $oneToOne)
|
|
{
|
|
if ($coll === null) {
|
|
unset($this->_resultPointers[$dqlAlias]); // Ticket #1228
|
|
|
|
return;
|
|
}
|
|
|
|
if ($oneToOne) {
|
|
$this->_resultPointers[$dqlAlias] =& $coll;
|
|
|
|
return;
|
|
}
|
|
|
|
if ($index !== false) {
|
|
$this->_resultPointers[$dqlAlias] =& $coll[$index];
|
|
|
|
return;
|
|
}
|
|
|
|
if ( ! $coll) {
|
|
return;
|
|
}
|
|
|
|
end($coll);
|
|
$this->_resultPointers[$dqlAlias] =& $coll[key($coll)];
|
|
|
|
return;
|
|
}
|
|
|
|
/**
|
|
* Retrieve ClassMetadata associated to entity class name.
|
|
*
|
|
* @param string $className
|
|
*
|
|
* @return \Doctrine\ORM\Mapping\ClassMetadata
|
|
*/
|
|
private function getClassMetadata($className)
|
|
{
|
|
if ( ! isset($this->_ce[$className])) {
|
|
$this->_ce[$className] = $this->_em->getClassMetadata($className);
|
|
}
|
|
|
|
return $this->_ce[$className];
|
|
}
|
|
}
|