use exist route object for route generation (#166)

sulu · May 3, 2017 · e862d96 · e862d96
1 parent 74b8090
commit e862d96
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/Document/Subscriber/RoutableSubscriber.php b/Document/Subscriber/RoutableSubscriber.php
@@ -235,7 +235,12 @@ public function handlePublish(PublishEvent $event)
      */
     private function createOrUpdatePageRoute(RoutablePageBehavior $document, $locale)
     {
-        $route = $this->routeRepository->findByEntity($document->getClass(), $document->getUuid(), $locale);
+        $route = $document->getRoute();
+
+        if (!$route) {
+            $route = $this->routeRepository->findByEntity($document->getClass(), $document->getUuid(), $locale);
+        }
+
         if ($route) {
             $document->setRoute($route);
 
@@ -255,7 +260,12 @@ private function createOrUpdatePageRoute(RoutablePageBehavior $document, $locale
      */
     private function createOrUpdateRoute(RoutableBehavior $document, $locale)
     {
-        $route = $this->routeRepository->findByEntity($document->getClass(), $document->getUuid(), $locale);
+        $route = $document->getRoute();
+
+        if (!$route) {
+            $route = $this->routeRepository->findByEntity($document->getClass(), $document->getUuid(), $locale);
+        }
+
         if ($route) {
             $document->setRoute($route);