From 58547c980282b465ff565a68fc354b895730000f Mon Sep 17 00:00:00 2001
From: Danny van Wijk <dannyvanwijk@gmail.com>
Date: Tue, 27 Dec 2022 15:27:31 +0100
Subject: [PATCH] chore(doctrine): use getObjectManager instead of deprecated
 getEntityManager

---
 src/Doctrine/EventListener/PublishMercureUpdatesListener.php | 2 +-
 src/Doctrine/EventListener/PurgeHttpCacheListener.php        | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/Doctrine/EventListener/PublishMercureUpdatesListener.php b/src/Doctrine/EventListener/PublishMercureUpdatesListener.php
index de3164fda32..0f04d00fd34 100644
--- a/src/Doctrine/EventListener/PublishMercureUpdatesListener.php
+++ b/src/Doctrine/EventListener/PublishMercureUpdatesListener.php
@@ -92,7 +92,7 @@ public function __construct(ResourceClassResolverInterface $resourceClassResolve
     public function onFlush(EventArgs $eventArgs): void
     {
         if ($eventArgs instanceof OrmOnFlushEventArgs) {
-            $uow = $eventArgs->getEntityManager()->getUnitOfWork();
+            $uow = method_exists($eventArgs, 'getObjectManager') ? $eventArgs->getObjectManager()->getUnitOfWork() : $eventArgs->getEntityManager()->getUnitOfWork();
         } elseif ($eventArgs instanceof MongoDbOdmOnFlushEventArgs) {
             $uow = $eventArgs->getDocumentManager()->getUnitOfWork();
         } else {
diff --git a/src/Doctrine/EventListener/PurgeHttpCacheListener.php b/src/Doctrine/EventListener/PurgeHttpCacheListener.php
index f6302ce72c1..39ad2208fd5 100644
--- a/src/Doctrine/EventListener/PurgeHttpCacheListener.php
+++ b/src/Doctrine/EventListener/PurgeHttpCacheListener.php
@@ -55,7 +55,8 @@ public function preUpdate(PreUpdateEventArgs $eventArgs): void
         $this->gatherResourceAndItemTags($object, true);
 
         $changeSet = $eventArgs->getEntityChangeSet();
-        $associationMappings = $eventArgs->getEntityManager()->getClassMetadata(ClassUtils::getClass($eventArgs->getObject()))->getAssociationMappings();
+        $objectManager = method_exists($eventArgs, 'getObjectManager') ? $eventArgs->getObjectManager() : $eventArgs->getEntityManager();
+        $associationMappings = $objectManager->getClassMetadata(ClassUtils::getClass($eventArgs->getObject()))->getAssociationMappings();
 
         foreach ($changeSet as $key => $value) {
             if (!isset($associationMappings[$key])) {
@@ -72,7 +73,7 @@ public function preUpdate(PreUpdateEventArgs $eventArgs): void
      */
     public function onFlush(OnFlushEventArgs $eventArgs): void
     {
-        $em = $eventArgs->getEntityManager();
+        $em = method_exists($eventArgs, 'getObjectManager') ? $eventArgs->getObjectManager() : $eventArgs->getEntityManager();
         $uow = $em->getUnitOfWork();
 
         foreach ($uow->getScheduledEntityInsertions() as $entity) {