File tree 2 files changed +4
-3
lines changed
src/Doctrine/EventListener
2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -92,7 +92,7 @@ public function __construct(ResourceClassResolverInterface $resourceClassResolve
92
92
public function onFlush (EventArgs $ eventArgs ): void
93
93
{
94
94
if ($ eventArgs instanceof OrmOnFlushEventArgs) {
95
- $ uow = $ eventArgs ->getEntityManager ()->getUnitOfWork ();
95
+ $ uow = method_exists ( $ eventArgs , ' getObjectManager ' ) ? $ eventArgs -> getObjectManager ()-> getUnitOfWork () : $ eventArgs ->getEntityManager ()->getUnitOfWork ();
96
96
} elseif ($ eventArgs instanceof MongoDbOdmOnFlushEventArgs) {
97
97
$ uow = $ eventArgs ->getDocumentManager ()->getUnitOfWork ();
98
98
} else {
Original file line number Diff line number Diff line change @@ -55,7 +55,8 @@ public function preUpdate(PreUpdateEventArgs $eventArgs): void
55
55
$ this ->gatherResourceAndItemTags ($ object , true );
56
56
57
57
$ 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 ();
59
60
60
61
foreach ($ changeSet as $ key => $ value ) {
61
62
if (!isset ($ associationMappings [$ key ])) {
@@ -72,7 +73,7 @@ public function preUpdate(PreUpdateEventArgs $eventArgs): void
72
73
*/
73
74
public function onFlush (OnFlushEventArgs $ eventArgs ): void
74
75
{
75
- $ em = $ eventArgs ->getEntityManager ();
76
+ $ em = method_exists ( $ eventArgs , ' getObjectManager ' ) ? $ eventArgs -> getObjectManager () : $ eventArgs ->getEntityManager ();
76
77
$ uow = $ em ->getUnitOfWork ();
77
78
78
79
foreach ($ uow ->getScheduledEntityInsertions () as $ entity ) {
You can’t perform that action at this time.
0 commit comments