Merge pull request #240 from MAXakaWIZARD/pr-locale-for-new-objects

Set current and default locale on prePersist event
KnpLabs · Sep 11, 2015 · f2663e4 · f2663e4
2 parents daa1ed7 + 61119b2
commit f2663e4
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/ORM/Translatable/TranslatableSubscriber.php b/src/ORM/Translatable/TranslatableSubscriber.php
@@ -298,6 +298,16 @@ private function isTranslation(ClassMetadata $classMetadata)
     }
 
     public function postLoad(LifecycleEventArgs $eventArgs)
+    {
+        $this->setLocales($eventArgs);
+    }
+
+    public function prePersist(LifecycleEventArgs $eventArgs)
+    {
+        $this->setLocales($eventArgs);
+    }
+
+    private function setLocales(LifecycleEventArgs $eventArgs)
     {
         $em            = $eventArgs->getEntityManager();
         $entity        = $eventArgs->getEntity();
@@ -340,6 +350,7 @@ public function getSubscribedEvents()
         return [
             Events::loadClassMetadata,
             Events::postLoad,
+            Events::prePersist,
         ];
     }
 }