Merge pull request #23 from wearemarketing/feature/solve-uow-issues

change how AutoRoute entity is updated
wearemarketing · Nov 5, 2019 · ff1ea0f · ff1ea0f
2 parents d6e2e35 + c19dc87
commit ff1ea0f
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/src/Doctrine/Orm/AutoRouteListener.php b/src/Doctrine/Orm/AutoRouteListener.php
@@ -89,7 +89,9 @@ private function parseUpdateEntities(array $inEntities)
     private function isAutoRouteable($document)
     {
         try {
-            return (boolean)$this->getMetadataFactory()->getMetadataForClass(get_class($document));
+            $metadataFactory = $this->getMetadataFactory();
+            $classMetadata = $metadataFactory->getMetadataForClass(get_class($document));
+            return (boolean)$classMetadata;
         } catch (ClassNotMappedException $e) {
             return false;
         }
@@ -140,9 +142,15 @@ public function onFlush(OnFlushEventArgs $eventArgs)
 
                 //set persistence order to allow set in the route the contentEntity PK
                 if ($new) {
+
+                    $fromHash = $this->getEntityMetadata($manager, $entity)->name;
+
+                    $commitOrderCalculator->addNode($fromHash, $entity);
+
                     $commitOrderCalculator->addDependency(
-                        $this->getEntityMetadata($manager, $entity),
-                        $this->getEntityMetadata($manager, $autoRoute)
+                        $fromHash,
+                        $this->getEntityMetadata($manager, $autoRoute)->name,
+                        1
                     );
                 }
 
@@ -189,7 +197,8 @@ public function postPersist(LifecycleEventArgs $eventArgs)
                     $autoRoute->setContentId($id);
                     $this->replaceIdOnNameField($autoRoute, $id, 'name');
                     $this->replaceIdOnNameField($autoRoute, $id, 'canonicalName');
-                    $unitOfWork->recomputeSingleEntityChangeSet($this->getEntityMetadata($manager, $autoRoute), $autoRoute);
+
+                    $unitOfWork->computeChangeSet($this->getEntityMetadata($manager, $autoRoute), $autoRoute);
                 }
             }
         }