FileMaster

Edit File: DefaultEntityHydrator.php

<?php
declare (strict_types=1);
namespace MailPoetVendor\Doctrine\ORM\Cache;
if (!defined('ABSPATH')) exit;
use MailPoetVendor\Doctrine\ORM\EntityManagerInterface;
use MailPoetVendor\Doctrine\ORM\Mapping\ClassMetadata;
use MailPoetVendor\Doctrine\ORM\Proxy\DefaultProxyClassNameResolver;
use MailPoetVendor\Doctrine\ORM\Query;
use MailPoetVendor\Doctrine\ORM\UnitOfWork;
use MailPoetVendor\Doctrine\ORM\Utility\IdentifierFlattener;
use function array_merge;
use function assert;
use function is_array;
use function is_object;
use function reset;
class DefaultEntityHydrator implements EntityHydrator
{
 private $em;
 private $uow;
 private $identifierFlattener;
 private static $hints = [Query::HINT_CACHE_ENABLED => \true];
 public function __construct(EntityManagerInterface $em)
 {
 $this->em = $em;
 $this->uow = $em->getUnitOfWork();
 $this->identifierFlattener = new IdentifierFlattener($em->getUnitOfWork(), $em->getMetadataFactory());
 }
 public function buildCacheEntry(ClassMetadata $metadata, EntityCacheKey $key, $entity)
 {
 $data = $this->uow->getOriginalEntityData($entity);
 $data = array_merge($data, $metadata->getIdentifierValues($entity));
 // why update has no identifier values ?
 if ($metadata->requiresFetchAfterChange) {
 if ($metadata->isVersioned) {
 assert($metadata->versionField !== null);
 $data[$metadata->versionField] = $metadata->getFieldValue($entity, $metadata->versionField);
 }
 foreach ($metadata->fieldMappings as $name => $fieldMapping) {
 if (isset($fieldMapping['generated'])) {
 $data[$name] = $metadata->getFieldValue($entity, $name);
 }
 }
 }
 foreach ($metadata->associationMappings as $name => $assoc) {
 if (!isset($data[$name])) {
 continue;
 }
 if (!($assoc['type'] & ClassMetadata::TO_ONE)) {
 unset($data[$name]);
 continue;
 }
 if (!isset($assoc['cache'])) {
 $targetClassMetadata = $this->em->getClassMetadata($assoc['targetEntity']);
 $owningAssociation = !$assoc['isOwningSide'] ? $targetClassMetadata->associationMappings[$assoc['mappedBy']] : $assoc;
 $associationIds = $this->identifierFlattener->flattenIdentifier($targetClassMetadata, $targetClassMetadata->getIdentifierValues($data[$name]));
 unset($data[$name]);
 foreach ($associationIds as $fieldName => $fieldValue) {
 if (isset($targetClassMetadata->fieldMappings[$fieldName])) {
 $fieldMapping = $targetClassMetadata->fieldMappings[$fieldName];
 $data[$owningAssociation['targetToSourceKeyColumns'][$fieldMapping['columnName']]] = $fieldValue;
 continue;
 }
 $targetAssoc = $targetClassMetadata->associationMappings[$fieldName];
 foreach ($assoc['targetToSourceKeyColumns'] as $referencedColumn => $localColumn) {
 if (isset($targetAssoc['sourceToTargetKeyColumns'][$referencedColumn])) {
 $data[$localColumn] = $fieldValue;
 }
 }
 }
 continue;
 }
 if (!isset($assoc['id'])) {
 $targetClass = DefaultProxyClassNameResolver::getClass($data[$name]);
 $targetId = $this->uow->getEntityIdentifier($data[$name]);
 $data[$name] = new AssociationCacheEntry($targetClass, $targetId);
 continue;
 }
 // handle association identifier
 $targetId = is_object($data[$name]) && $this->uow->isInIdentityMap($data[$name]) ? $this->uow->getEntityIdentifier($data[$name]) : $data[$name];
 // @TODO - fix it !
 // handle UnitOfWork#createEntity hash generation
 if (!is_array($targetId)) {
 $data[reset($assoc['joinColumnFieldNames'])] = $targetId;
 $targetEntity = $this->em->getClassMetadata($assoc['targetEntity']);
 $targetId = [$targetEntity->identifier[0] => $targetId];
 }
 $data[$name] = new AssociationCacheEntry($assoc['targetEntity'], $targetId);
 }
 return new EntityCacheEntry($metadata->name, $data);
 }
 public function loadCacheEntry(ClassMetadata $metadata, EntityCacheKey $key, EntityCacheEntry $entry, $entity = null)
 {
 $data = $entry->data;
 $hints = self::$hints;
 if ($entity !== null) {
 $hints[Query::HINT_REFRESH] = \true;
 $hints[Query::HINT_REFRESH_ENTITY] = $entity;
 }
 foreach ($metadata->associationMappings as $name => $assoc) {
 if (!isset($assoc['cache']) || !isset($data[$name])) {
 continue;
 }
 $assocClass = $data[$name]->class;
 $assocId = $data[$name]->identifier;
 $isEagerLoad = $assoc['fetch'] === ClassMetadata::FETCH_EAGER || $assoc['type'] === ClassMetadata::ONE_TO_ONE && !$assoc['isOwningSide'];
 if (!$isEagerLoad) {
 $data[$name] = $this->em->getReference($assocClass, $assocId);
 continue;
 }
 $assocMetadata = $this->em->getClassMetadata($assoc['targetEntity']);
 $assocKey = new EntityCacheKey($assocMetadata->rootEntityName, $assocId);
 $assocPersister = $this->uow->getEntityPersister($assoc['targetEntity']);
 $assocRegion = $assocPersister->getCacheRegion();
 $assocEntry = $assocRegion->get($assocKey);
 if ($assocEntry === null) {
 return null;
 }
 $data[$name] = $this->uow->createEntity($assocEntry->class, $assocEntry->resolveAssociationEntries($this->em), $hints);
 }
 if ($entity !== null) {
 $this->uow->registerManaged($entity, $key->identifier, $data);
 }
 $result = $this->uow->createEntity($entry->class, $data, $hints);
 $this->uow->hydrationComplete();
 return $result;
 }
}

Back