File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -65,7 +65,7 @@ public function onFlush(EventArgs $eventArgs): void
65
65
return ;
66
66
}
67
67
68
- $ em = $ eventArgs ->getEntityManager ();
68
+ $ em = method_exists ( $ eventArgs , ' getObjectManager ' ) ? $ eventArgs -> getObjectManager () : $ eventArgs ->getEntityManager ();
69
69
$ uow = $ em ->getUnitOfWork ();
70
70
foreach ($ uow ->getScheduledEntityInsertions () as $ entity ) {
71
71
$ this ->storeEntitiesToPublish ($ em , $ entity , 'createdEntities ' );
@@ -89,7 +89,7 @@ public function postFlush(EventArgs $eventArgs): void
89
89
return ;
90
90
}
91
91
92
- $ em = $ eventArgs ->getEntityManager ();
92
+ $ em = method_exists ( $ eventArgs , ' getObjectManager ' ) ? $ eventArgs -> getObjectManager () : $ eventArgs ->getEntityManager ();
93
93
94
94
try {
95
95
foreach ($ this ->createdEntities as $ entity ) {
You can’t perform that action at this time.
0 commit comments