Fixed code duplicating

doctrine · Dec 19, 2017 · d9142d2 · d9142d2
1 parent e5ebbf7
commit d9142d2
Showing 1 changed file with 1 addition and 10 deletions.
diff --git a/lib/Doctrine/ODM/MongoDB/SchemaManager.php b/lib/Doctrine/ODM/MongoDB/SchemaManager.php
@@ -615,16 +615,7 @@ private function runShardCollectionCommand($documentName)
             try {
                 $mapping = $class->getFieldMappingByDbFieldName($key);
                 if (isset($mapping['association']) && $mapping['association'] === ClassMetadata::REFERENCE_ONE) {
-                    switch ($mapping['storeAs']) {
-                        case ClassMetadataInfo::REFERENCE_STORE_AS_REF:
-                            $key .= '.id';
-                            break;
-
-                        case ClassMetadataInfo::REFERENCE_STORE_AS_DB_REF:
-                        case ClassMetadataInfo::REFERENCE_STORE_AS_DB_REF_WITH_DB:
-                            $key .= '.$id';
-                            break;
-                    }
+                    $key = ClassMetadataInfo::getReferenceFieldName($mapping['storeAs'], $key);
                 }
             } catch (MappingException $exception) {
                 // Ignore hitting unmapped fields