Update BaseListener.php

AppsDevTeam · Mar 25, 2024 · 050a6b6 · 050a6b6
1 parent 74ec2e2
commit 050a6b6
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/BaseListener.php b/src/BaseListener.php
@@ -20,7 +20,7 @@ abstract class BaseListener implements EventSubscriber
 	protected EntityManagerDecorator $em;
 
 	/** @var callable[] */
-	protected array $postFlush = [];
+	private array $postFlush = [];
 
 	/** @var callable[] */
 	protected array $entitiesToCompute = [];
@@ -170,4 +170,13 @@ protected function recalculateEntities(): void
 			$this->em->getUnitOfWork()->recomputeSingleEntityChangeSet($this->em->getClassMetadata($_entity::class), $_entity);
 		}
 	}
-}
+
+	final protected function addPostFlushCallback(callable $callback): void
+	{
+		if (!isset($this->getSubscribedEvents()['postFlush'])) {
+			throw new \Exception ('Missing postFlush subsribed event.');
+		}
+
+		$this->postFlush[] = $callback;
+	}
+}