Add internal API method

Warxcell · Aug 2, 2017 · 7b95a4c · 7b95a4c
1 parent 7bbcb6c
commit 7b95a4c
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/src/EventSubscriber/CurrentTranslationLoader.php b/src/EventSubscriber/CurrentTranslationLoader.php
@@ -79,8 +79,8 @@ public function initializeTranslation(Translatable $entity, $languageOrLocale)
     public function getTranslation(Translatable $entity, $languageOrLocale)
     {
         $translations = $entity->getTranslations();
-        $translationsFiltered = array_filter(
-            $translations,
+
+        $translation = $translations->filter(
             function (Translation $item) use ($languageOrLocale) {
                 $translationLanguage = $item->getLanguage();
                 if ($languageOrLocale instanceof Language) {
@@ -89,12 +89,13 @@ function (Translation $item) use ($languageOrLocale) {
                     return $translationLanguage->getLocale() === $languageOrLocale;
                 }
             }
-        );
-        if (count($translationsFiltered) > 0) {
-            return reset($translationsFiltered);
+        )->first();
+        if ($translation === false) {
+            return null;
+
         }
 
-        return null;
+        return $translation;
     }
 
     public function initializeCurrentTranslation(Translatable $entity)