Merge pull request #2239 from doctrine/cherry-pick-2234

Assert array value is an instance of the key
doctrine · Nov 3, 2020 · 1409465 · 1409465
2 parents f3f1d1d + a1be178
commit 1409465
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/Doctrine/ODM/MongoDB/Mapping/Driver/AnnotationDriver.php b/lib/Doctrine/ODM/MongoDB/Mapping/Driver/AnnotationDriver.php
@@ -9,6 +9,7 @@
 use Doctrine\ODM\MongoDB\Events;
 use Doctrine\ODM\MongoDB\Mapping\Annotations as ODM;
 use Doctrine\ODM\MongoDB\Mapping\Annotations\AbstractIndex;
+use Doctrine\ODM\MongoDB\Mapping\Annotations\ShardKey;
 use Doctrine\ODM\MongoDB\Mapping\ClassMetadata;
 use Doctrine\ODM\MongoDB\Mapping\MappingException;
 use Doctrine\Persistence\Mapping\Driver\AnnotationDriver as AbstractAnnotationDriver;
@@ -204,8 +205,9 @@ public function loadMetadataForClass($className, \Doctrine\Persistence\Mapping\C
         }
 
         // Set shard key after all fields to ensure we mapped all its keys
-        if (isset($classAnnotations['Doctrine\ODM\MongoDB\Mapping\Annotations\ShardKey'])) {
-            $this->setShardKey($class, $classAnnotations['Doctrine\ODM\MongoDB\Mapping\Annotations\ShardKey']);
+        if (isset($classAnnotations[ShardKey::class])) {
+            assert($classAnnotations[ShardKey::class] instanceof ShardKey);
+            $this->setShardKey($class, $classAnnotations[ShardKey::class]);
         }
 
         /** @var ReflectionMethod $method */