190 lines
6.5 KiB
PHP
190 lines
6.5 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\DBAL\Types\Type;
|
|
use Doctrine\ORM\Mapping\ClassMetadata;
|
|
use Doctrine\ORM\Query;
|
|
|
|
class SimpleObjectHydrator extends AbstractHydrator
|
|
{
|
|
/**
|
|
* @var ClassMetadata
|
|
*/
|
|
private $class;
|
|
|
|
/**
|
|
* @var array
|
|
*/
|
|
private $declaringClasses = array();
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function hydrateAllData()
|
|
{
|
|
$result = array();
|
|
$cache = array();
|
|
|
|
while ($row = $this->_stmt->fetch(PDO::FETCH_ASSOC)) {
|
|
$this->hydrateRowData($row, $cache, $result);
|
|
}
|
|
|
|
$this->_em->getUnitOfWork()->triggerEagerLoads();
|
|
|
|
return $result;
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function prepare()
|
|
{
|
|
if (count($this->_rsm->aliasMap) !== 1) {
|
|
throw new \RuntimeException("Cannot use SimpleObjectHydrator with a ResultSetMapping that contains more than one object result.");
|
|
}
|
|
|
|
if ($this->_rsm->scalarMappings) {
|
|
throw new \RuntimeException("Cannot use SimpleObjectHydrator with a ResultSetMapping that contains scalar mappings.");
|
|
}
|
|
|
|
$this->class = $this->_em->getClassMetadata(reset($this->_rsm->aliasMap));
|
|
|
|
// We only need to add declaring classes if we have inheritance.
|
|
if ($this->class->inheritanceType === ClassMetadata::INHERITANCE_TYPE_NONE) {
|
|
return;
|
|
}
|
|
|
|
foreach ($this->_rsm->declaringClasses as $column => $class) {
|
|
$this->declaringClasses[$column] = $this->_em->getClassMetadata($class);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function hydrateRowData(array $sqlResult, array &$cache, array &$result)
|
|
{
|
|
$entityName = $this->class->name;
|
|
$data = array();
|
|
|
|
// We need to find the correct entity class name if we have inheritance in resultset
|
|
if ($this->class->inheritanceType !== ClassMetadata::INHERITANCE_TYPE_NONE) {
|
|
$discrColumnName = $this->_platform->getSQLResultCasing($this->class->discriminatorColumn['name']);
|
|
|
|
if ( ! isset($sqlResult[$discrColumnName])) {
|
|
throw HydrationException::missingDiscriminatorColumn($entityName, $discrColumnName, key($this->_rsm->aliasMap));
|
|
}
|
|
|
|
if ($sqlResult[$discrColumnName] === '') {
|
|
throw HydrationException::emptyDiscriminatorValue(key($this->_rsm->aliasMap));
|
|
}
|
|
|
|
$entityName = $this->class->discriminatorMap[$sqlResult[$discrColumnName]];
|
|
|
|
unset($sqlResult[$discrColumnName]);
|
|
}
|
|
|
|
foreach ($sqlResult as $column => $value) {
|
|
// Hydrate column information if not yet present
|
|
if ( ! isset($cache[$column])) {
|
|
if (($info = $this->hydrateColumnInfo($entityName, $column)) === null) {
|
|
continue;
|
|
}
|
|
|
|
$cache[$column] = $info;
|
|
}
|
|
|
|
// Convert field to a valid PHP value
|
|
if (isset($cache[$column]['type'])) {
|
|
$value = Type::getType($cache[$column]['type'])->convertToPHPValue($value, $this->_platform);
|
|
}
|
|
|
|
// Prevent overwrite in case of inherit classes using same property name (See AbstractHydrator)
|
|
if (isset($cache[$column]) && ( ! isset($data[$cache[$column]['name']]) || $value !== null)) {
|
|
$data[$cache[$column]['name']] = $value;
|
|
}
|
|
}
|
|
|
|
if (isset($this->_hints[Query::HINT_REFRESH_ENTITY])) {
|
|
$this->registerManaged($this->class, $this->_hints[Query::HINT_REFRESH_ENTITY], $data);
|
|
}
|
|
|
|
$uow = $this->_em->getUnitOfWork();
|
|
$entity = $uow->createEntity($entityName, $data, $this->_hints);
|
|
|
|
$result[] = $entity;
|
|
}
|
|
|
|
/**
|
|
* Retrieve column information form ResultSetMapping.
|
|
*
|
|
* @param string $entityName
|
|
* @param string $column
|
|
*
|
|
* @return array
|
|
*/
|
|
protected function hydrateColumnInfo($entityName, $column)
|
|
{
|
|
|
|
if (isset($this->_rsm->fieldMappings[$column])) {
|
|
$name = $this->_rsm->fieldMappings[$column];
|
|
$class = isset($this->declaringClasses[$column])
|
|
? $this->declaringClasses[$column]
|
|
: $this->class;
|
|
|
|
// If class is not part of the inheritance, ignore
|
|
if ( ! ($class->name === $entityName || is_subclass_of($entityName, $class->name))) {
|
|
return null;
|
|
}
|
|
|
|
return array(
|
|
'name' => $name,
|
|
'type' => $class->fieldMappings[$name]['type']
|
|
);
|
|
}
|
|
|
|
if (isset($this->_rsm->relationMap[$column])) {
|
|
$class = isset($this->_rsm->relationMap[$column])
|
|
? $this->_rsm->relationMap[$column]
|
|
: $this->class;
|
|
|
|
// If class is not self referencing, ignore
|
|
if ( ! ($class === $entityName || is_subclass_of($entityName, $class))) {
|
|
return null;
|
|
}
|
|
|
|
// TODO: Decide what to do with associations. It seems original code is incomplete.
|
|
// One solution is to load the association, but it might require extra efforts.
|
|
return array('name' => $column);
|
|
}
|
|
|
|
if (isset($this->_rsm->metaMappings[$column])) {
|
|
return array(
|
|
'name' => $this->_rsm->metaMappings[$column],
|
|
'type' => (isset($this->_rsm->typeMappings[$column]) ? $this->_rsm->typeMappings[$column] : null)
|
|
);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
}
|