[OpenApi] ObjectReference support mongo db

mpoiriert · Jan 16, 2024 · 31c26d7 · 31c26d7
1 parent fba9a9d
commit 31c26d7
Showing 1 changed file with 14 additions and 7 deletions.
diff --git a/packages/open-api/Serializer/Handler/ObjectReferenceHandler.php b/packages/open-api/Serializer/Handler/ObjectReferenceHandler.php
@@ -3,6 +3,7 @@
 namespace Draw\Component\OpenApi\Serializer\Handler;
 
 use Doctrine\Persistence\ManagerRegistry;
+use Doctrine\Persistence\ObjectManager;
 use JMS\Serializer\Context;
 use JMS\Serializer\DeserializationContext;
 use JMS\Serializer\GraphNavigatorInterface;
@@ -30,8 +31,10 @@ public static function getSubscribingMethods(): array
         ];
     }
 
-    public function __construct(private ManagerRegistry $managerRegistry)
-    {
+    public function __construct(
+        private ?ManagerRegistry $ormManagerRegistry,
+        private ?ManagerRegistry $odmManagerRegistry
+    ) {
     }
 
     public function serializeObjectReference(
@@ -45,8 +48,7 @@ public function serializeObjectReference(
         }
 
         $class = $type['params'][0]['name'];
-        $identifiers = $this->managerRegistry
-            ->getManagerForClass($class)
+        $identifiers = $this->getManagerForClass($class)
             ->getMetadataFactory()
             ->getMetadataFor($class)
             ->getIdentifierValues($value);
@@ -64,9 +66,14 @@ public function deserializeObjectReference(
             return null;
         }
 
-        $repository = $this->managerRegistry
-            ->getRepository($type['params'][0]['name']);
+        return $this->getManagerForClass($type['params'][0]['name'])
+            ->find($type['params'][0]['name'], $value);
+    }
 
-        return $repository->find($value);
+    private function getManagerForClass(string $class): ObjectManager
+    {
+        return $this->ormManagerRegistry?->getManagerForClass($class)
+            ?? $this->odmManagerRegistry?->getManagerForClass($class)
+            ?? throw new \RuntimeException('No object manager found for class ' . $class);
     }
 }