123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183 |
- <?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 LGPL. For more information, see
- * <http://www.doctrine-project.org>.
- */
- namespace Doctrine\ORM\Internal\Hydration;
- use \PDO,
- Doctrine\DBAL\Types\Type,
- Doctrine\ORM\Mapping\ClassMetadata,
- Doctrine\ORM\Event\LifecycleEventArgs,
- Doctrine\ORM\Events,
- 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 ($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]['field'])) {
- $type = Type::getType($cache[$column]['class']->fieldMappings[$cache[$column]['name']]['type']);
- $value = $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)
- {
- switch (true) {
- case (isset($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(
- 'class' => $class,
- 'name' => $this->_rsm->fieldMappings[$column],
- 'field' => true,
- );
- case (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);
- default:
- return array(
- 'name' => $this->_rsm->metaMappings[$column]
- );
- }
- }
- }
|