From 2f58aaa3a14c7cafe94a9f651a072aed06f614dc Mon Sep 17 00:00:00 2001
From: Jordi Sala Morales <jordism91@gmail.com>
Date: Sat, 1 Apr 2023 21:35:25 +0200
Subject: [PATCH] Catch MongoDBExceptions to throw ModelManagerExceptions
 (#811)

---
 src/Model/ModelManager.php | 70 +++++++++++++++++++++++++++++---------
 1 file changed, 53 insertions(+), 17 deletions(-)

diff --git a/src/Model/ModelManager.php b/src/Model/ModelManager.php
index bd5e023d..a423107c 100755
--- a/src/Model/ModelManager.php
+++ b/src/Model/ModelManager.php
@@ -16,10 +16,13 @@
 use Doctrine\ODM\MongoDB\DocumentManager;
 use Doctrine\ODM\MongoDB\Iterator\Iterator;
 use Doctrine\ODM\MongoDB\Mapping\ClassMetadata as MongoDBClassMetadata;
+use Doctrine\ODM\MongoDB\MongoDBException;
 use Doctrine\ODM\MongoDB\Query\Builder;
 use Doctrine\ODM\MongoDB\Repository\DocumentRepository;
 use Doctrine\Persistence\Mapping\MappingException;
+use MongoDB\Driver\Exception\Exception;
 use Sonata\AdminBundle\Datagrid\ProxyQueryInterface as BaseProxyQueryInterface;
+use Sonata\AdminBundle\Exception\ModelManagerException;
 use Sonata\AdminBundle\Model\ModelManagerInterface;
 use Sonata\AdminBundle\Model\ProxyResolverInterface;
 use Sonata\DoctrineMongoDBAdminBundle\Datagrid\ProxyQuery;
@@ -44,23 +47,47 @@ public function __construct(
 
     public function create(object $object): void
     {
-        $documentManager = $this->getDocumentManager($object);
-        $documentManager->persist($object);
-        $documentManager->flush();
+        try {
+            $documentManager = $this->getDocumentManager($object);
+            $documentManager->persist($object);
+            $documentManager->flush();
+        } catch (Exception|MongoDBException $exception) {
+            throw new ModelManagerException(
+                sprintf('Failed to create object: %s', $this->getRealClass($object)),
+                $exception->getCode(),
+                $exception
+            );
+        }
     }
 
     public function update(object $object): void
     {
-        $documentManager = $this->getDocumentManager($object);
-        $documentManager->persist($object);
-        $documentManager->flush();
+        try {
+            $documentManager = $this->getDocumentManager($object);
+            $documentManager->persist($object);
+            $documentManager->flush();
+        } catch (Exception|MongoDBException $exception) {
+            throw new ModelManagerException(
+                sprintf('Failed to update object: %s', $this->getRealClass($object)),
+                $exception->getCode(),
+                $exception
+            );
+        }
     }
 
     public function delete(object $object): void
     {
-        $documentManager = $this->getDocumentManager($object);
-        $documentManager->remove($object);
-        $documentManager->flush();
+        try {
+            $documentManager = $this->getDocumentManager($object);
+            $documentManager->remove($object);
+            $documentManager->flush();
+        } catch (Exception|MongoDBException $exception) {
+            throw new ModelManagerException(
+                sprintf('Failed to delete object: %s', $this->getRealClass($object)),
+                $exception->getCode(),
+                $exception
+            );
+        }
     }
 
     /**
@@ -207,17 +234,26 @@ public function batchDelete(string $class, BaseProxyQueryInterface $query): void
         \assert($iterator instanceof Iterator);
 
         $i = 0;
-        foreach ($iterator as $object) {
-            $documentManager->remove($object);
 
-            if (0 === (++$i % 20)) {
-                $documentManager->flush();
-                $documentManager->clear();
+        try {
+            foreach ($iterator as $object) {
+                $documentManager->remove($object);
+
+                if (0 === (++$i % 20)) {
+                    $documentManager->flush();
+                    $documentManager->clear();
+                }
             }
-        }
 
-        $documentManager->flush();
-        $documentManager->clear();
+            $documentManager->flush();
+            $documentManager->clear();
+        } catch (Exception|MongoDBException $exception) {
+            throw new ModelManagerException(
+                sprintf('Failed to delete object: %s', $class),
+                $exception->getCode(),
+                $exception
+            );
+        }
     }
 
     public function getExportFields(string $class): array