Skip to content

Commit 57c2aac

Browse files
authored
chore(doctrine): use getObjectManager instead of deprecated getEntityManager (#5291)
1 parent af98b64 commit 57c2aac

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

src/Doctrine/EventListener/PublishMercureUpdatesListener.php

+1-1
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ public function __construct(ResourceClassResolverInterface $resourceClassResolve
9292
public function onFlush(EventArgs $eventArgs): void
9393
{
9494
if ($eventArgs instanceof OrmOnFlushEventArgs) {
95-
$uow = $eventArgs->getEntityManager()->getUnitOfWork();
95+
$uow = method_exists($eventArgs, 'getObjectManager') ? $eventArgs->getObjectManager()->getUnitOfWork() : $eventArgs->getEntityManager()->getUnitOfWork();
9696
} elseif ($eventArgs instanceof MongoDbOdmOnFlushEventArgs) {
9797
$uow = $eventArgs->getDocumentManager()->getUnitOfWork();
9898
} else {

src/Doctrine/EventListener/PurgeHttpCacheListener.php

+3-2
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,8 @@ public function preUpdate(PreUpdateEventArgs $eventArgs): void
5555
$this->gatherResourceAndItemTags($object, true);
5656

5757
$changeSet = $eventArgs->getEntityChangeSet();
58-
$associationMappings = $eventArgs->getEntityManager()->getClassMetadata(ClassUtils::getClass($eventArgs->getObject()))->getAssociationMappings();
58+
$objectManager = method_exists($eventArgs, 'getObjectManager') ? $eventArgs->getObjectManager() : $eventArgs->getEntityManager();
59+
$associationMappings = $objectManager->getClassMetadata(ClassUtils::getClass($eventArgs->getObject()))->getAssociationMappings();
5960

6061
foreach ($changeSet as $key => $value) {
6162
if (!isset($associationMappings[$key])) {
@@ -72,7 +73,7 @@ public function preUpdate(PreUpdateEventArgs $eventArgs): void
7273
*/
7374
public function onFlush(OnFlushEventArgs $eventArgs): void
7475
{
75-
$em = $eventArgs->getEntityManager();
76+
$em = method_exists($eventArgs, 'getObjectManager') ? $eventArgs->getObjectManager() : $eventArgs->getEntityManager();
7677
$uow = $em->getUnitOfWork();
7778

7879
foreach ($uow->getScheduledEntityInsertions() as $entity) {

0 commit comments

Comments
 (0)