From dc925cc9c5461ca4160c685cd40a4a769bc168eb Mon Sep 17 00:00:00 2001 From: Stefan Kleff <s.kleff@goalio.de> Date: Thu, 10 Jan 2013 15:19:44 +0100 Subject: [PATCH] fixed indentation Restored old way of injection to just inject it during a refresh Added injection for initialized proxies --- lib/Doctrine/ORM/UnitOfWork.php | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/lib/Doctrine/ORM/UnitOfWork.php b/lib/Doctrine/ORM/UnitOfWork.php index 6e7eeb129..356ea02b2 100644 --- a/lib/Doctrine/ORM/UnitOfWork.php +++ b/lib/Doctrine/ORM/UnitOfWork.php @@ -2478,6 +2478,11 @@ class UnitOfWork implements PropertyChangedListener $entity->addPropertyChangedListener($this); } + // inject ObjectManager into just loaded proxies. + if ($overrideLocalValues && $entity instanceof ObjectManagerAware) { + $entity->injectObjectManager($this->em, $class); + } + } else { $overrideLocalValues = isset($hints[Query::HINT_REFRESH]); @@ -2485,12 +2490,12 @@ class UnitOfWork implements PropertyChangedListener if(isset($hints[Query::HINT_REFRESH_ENTITY])) { $overrideLocalValues = $hints[Query::HINT_REFRESH_ENTITY] === $entity; } + + // inject ObjectManager upon refresh. + if ($overrideLocalValues && $entity instanceof ObjectManagerAware) { + $entity->injectObjectManager($this->em, $class); + } } - - // inject ObjectManager into just loaded proxies. - if ($overrideLocalValues && $entity instanceof ObjectManagerAware) { - $entity->injectObjectManager($this->em, $class); - } if ($overrideLocalValues) { $this->originalEntityData[$oid] = $data;