allow to supress events

valantic · Mar 22, 2024 · dcfeba7 · dcfeba7
1 parent d834048
commit dcfeba7
Show file tree

Hide file tree

Showing 7 changed files with 77 additions and 8 deletions.
diff --git a/src/Messenger/Handler/RefreshElementHandler.php b/src/Messenger/Handler/RefreshElementHandler.php
@@ -23,6 +23,10 @@ public function __invoke(RefreshElement $message): void
     {
         $element = $this->resolveElement($message);
 
+        if (!$message->shouldTriggerEvents()) {
+            PropagateChanges::disableTriggerEvents();
+        }
+
         $this->propagateChanges->handle($element);
     }
 }
diff --git a/src/Messenger/Handler/RefreshElementInIndexHandler.php b/src/Messenger/Handler/RefreshElementInIndexHandler.php
@@ -28,6 +28,10 @@ public function __invoke(RefreshElementInIndex $message): void
         $index = $this->indexRepository->flattenedGet($message->index);
         $element = $this->resolveElement($message);
 
+        if (!$message->shouldTriggerEvents()) {
+            PropagateChanges::disableTriggerEvents();
+        }
+
         if ($index->usesBlueGreenIndices() && !$this->lockService->getIndexingLock($index)->acquire()) {
             $this->propagateChanges->handleIndex($element, $index, $index->getBlueGreenInactiveElasticaIndex());
         }

diff --git a/src/Messenger/Message/RefreshElement.php b/src/Messenger/Message/RefreshElement.php
@@ -8,8 +8,13 @@
 
 class RefreshElement extends AbstractRefresh
 {
-    public function __construct(ElementInterface $element)
+    public function __construct(ElementInterface $element, private readonly bool $shouldTriggerEvents = true)
     {
         $this->setElement($element);
     }
+
+    public function shouldTriggerEvents(): bool
+    {
+        return $this->shouldTriggerEvents;
+    }
 }
diff --git a/src/Messenger/Message/RefreshElementInIndex.php b/src/Messenger/Message/RefreshElementInIndex.php
@@ -11,7 +11,13 @@ class RefreshElementInIndex extends AbstractRefresh
     public function __construct(
         ElementInterface $element,
         public readonly string $index,
+        private readonly bool $shouldTriggerEvents = true,
     ) {
         $this->setElement($element);
     }
+
+    public function shouldTriggerEvents(): bool
+    {
+        return $this->shouldTriggerEvents;
+    }
 }
diff --git a/src/Model/Event/RefreshedElementEvent.php b/src/Model/Event/RefreshedElementEvent.php
@@ -12,10 +12,23 @@ class RefreshedElementEvent extends Event
 {
     /**
      * @param AbstractElement $element
-     * @param array<IndexInterface> $index
+     * @param array<IndexInterface> $indices
      */
     public function __construct(
-        public readonly AbstractElement $element,
-        public readonly array $index,
+        private readonly AbstractElement $element,
+        private readonly array $indices,
     ) {}
+
+    public function getElement(): AbstractElement
+    {
+        return $this->element;
+    }
+
+    /**
+     * @return array<IndexInterface>
+     */
+    public function getIndices(): array
+    {
+        return $this->indices;
+    }
 }
diff --git a/src/Model/Event/RefreshedElementInIndexEvent.php b/src/Model/Event/RefreshedElementInIndexEvent.php
@@ -18,4 +18,24 @@ public function __construct(
         public readonly Index $elasticaIndex,
         public readonly Operation $operation,
     ) {}
+
+    public function getElement(): AbstractElement
+    {
+        return $this->element;
+    }
+
+    public function getIndex(): IndexInterface
+    {
+        return $this->index;
+    }
+
+    public function getElasticaIndex(): Index
+    {
+        return $this->elasticaIndex;
+    }
+
+    public function getOperation(): Operation
+    {
+        return $this->operation;
+    }
 }
diff --git a/src/Service/PropagateChanges.php b/src/Service/PropagateChanges.php
@@ -21,6 +21,8 @@
 
 class PropagateChanges
 {
+    private static bool $triggerEvents = true;
+
     public function __construct(
         private readonly IndexRepository $indexRepository,
         private readonly DocumentHelper $documentHelper,
@@ -39,7 +41,10 @@ public function handle(AbstractElement $element): void
         $indices = $this->matchingIndicesForElement($this->indexRepository->flattenedAll(), $element);
 
         $event = new RefreshedElementEvent($element, $indices);
-        $this->eventDispatcher->dispatch($event, ElasticaBridgeEvents::PRE_REFRESH_ELEMENT);
+
+        if (self::$triggerEvents) {
+            $this->eventDispatcher->dispatch($event, ElasticaBridgeEvents::PRE_REFRESH_ELEMENT);
+        }
 
         if ($event->isPropagationStopped()) {
             return;
@@ -49,7 +54,9 @@ public function handle(AbstractElement $element): void
             $this->messageBus->dispatch(new RefreshElementInIndex($element, $index->getName()));
         }
 
-        $this->eventDispatcher->dispatch($event, ElasticaBridgeEvents::POST_REFRESH_ELEMENT);
+        if (self::$triggerEvents) {
+            $this->eventDispatcher->dispatch($event, ElasticaBridgeEvents::POST_REFRESH_ELEMENT);
+        }
     }
 
     public function handleIndex(
@@ -60,6 +67,11 @@ public function handleIndex(
         $this->doHandleIndex($element, $index, $elasticaIndex ?? $index->getElasticaIndex());
     }
 
+    public static function disableTriggerEvents(): void
+    {
+        self::$triggerEvents = false;
+    }
+
     private function doHandleIndex(
         AbstractElement $element,
         IndexInterface $index,
@@ -91,7 +103,10 @@ private function doHandleIndex(
             default => Operation::NOTHING,
         };
         $event = new RefreshedElementInIndexEvent($element, $index, $elasticaIndex, $operation);
-        $this->eventDispatcher->dispatch($event, ElasticaBridgeEvents::PRE_REFRESH_ELEMENT_IN_INDEX);
+
+        if (self::$triggerEvents) {
+            $this->eventDispatcher->dispatch($event, ElasticaBridgeEvents::PRE_REFRESH_ELEMENT_IN_INDEX);
+        }
 
         if ($event->isPropagationStopped()) {
             return;
@@ -104,7 +119,9 @@ private function doHandleIndex(
             Operation::UPDATE => $this->updateElementInIndex($element, $elasticaIndex, $document),
         };
 
-        $this->eventDispatcher->dispatch($event, ElasticaBridgeEvents::POST_REFRESH_ELEMENT_IN_INDEX);
+        if (self::$triggerEvents) {
+            $this->eventDispatcher->dispatch($event, ElasticaBridgeEvents::POST_REFRESH_ELEMENT_IN_INDEX);
+        }
 
         $this->documentHelper->resetTenantIfNeeded($document, $index);
     }