MappingFactory in Mapper requires its own method so we can override i…

…t when doing framework integrations
nilportugues · Oct 16, 2015 · 58e19a0 · 58e19a0
1 parent a3ae4e6
commit 58e19a0
Showing 1 changed file with 16 additions and 8 deletions.
diff --git a/src/Mapping/Mapper.php b/src/Mapping/Mapper.php
@@ -33,11 +33,7 @@ public function __construct(array $mappings = null)
     {
         if (is_array($mappings)) {
             foreach ($mappings as $mappedClass) {
-                if (is_string($mappedClass) && class_exists($mappedClass, true)) {
-                    $mapping = MappingFactory::fromClass($mappedClass);
-                } else {
-                    $mapping = MappingFactory::fromArray($mappedClass);
-                }
+                $mapping = $this->buildMapping($mappedClass);
 
                 if (false === empty($this->aliasMap[$mapping->getClassAlias()])) {
                     throw new MappingException(
@@ -52,11 +48,15 @@ public function __construct(array $mappings = null)
     }
 
     /**
-     * @param array $array
+     * @param string|array $mappedClass
+     *
+     * @return array
      */
-    public function setClassMap(array $array)
+    protected function buildMapping($mappedClass)
     {
-        $this->classMap = $array;
+        return (is_string($mappedClass) && class_exists($mappedClass, true)) ?
+            MappingFactory::fromClass($mappedClass) :
+            MappingFactory::fromArray($mappedClass);
     }
 
     /**
@@ -66,4 +66,12 @@ public function getClassMap()
     {
         return $this->classMap;
     }
+
+    /**
+     * @param array $array
+     */
+    public function setClassMap(array $array)
+    {
+        $this->classMap = $array;
+    }
 }