vendor/shopware/platform/src/Core/Content/Rule/DataAbstractionLayer/RuleIndexer.php line 90

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Shopware\Core\Content\Rule\DataAbstractionLayer;
  5. use Doctrine\DBAL\Connection;
  6. use Shopware\Core\Checkout\Cart\CartRuleLoader;
  7. use Shopware\Core\Content\Rule\Event\RuleIndexerEvent;
  8. use Shopware\Core\Content\Rule\RuleDefinition;
  9. use Shopware\Core\Content\Rule\RuleEvents;
  10. use Shopware\Core\Framework\DataAbstractionLayer\Dbal\Common\IteratorFactory;
  11. use Shopware\Core\Framework\DataAbstractionLayer\Doctrine\RetryableQuery;
  12. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepositoryInterface;
  13. use Shopware\Core\Framework\DataAbstractionLayer\Event\EntityWrittenContainerEvent;
  14. use Shopware\Core\Framework\DataAbstractionLayer\Event\EntityWrittenEvent;
  15. use Shopware\Core\Framework\DataAbstractionLayer\Indexing\EntityIndexer;
  16. use Shopware\Core\Framework\DataAbstractionLayer\Indexing\EntityIndexingMessage;
  17. use Shopware\Core\Framework\Plugin\Event\PluginPostActivateEvent;
  18. use Shopware\Core\Framework\Plugin\Event\PluginPostDeactivateEvent;
  19. use Shopware\Core\Framework\Plugin\Event\PluginPostInstallEvent;
  20. use Shopware\Core\Framework\Plugin\Event\PluginPostUninstallEvent;
  21. use Shopware\Core\Framework\Plugin\Event\PluginPostUpdateEvent;
  22. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  23. use Symfony\Contracts\EventDispatcher\EventDispatcherInterface;
  25. class RuleIndexer extends EntityIndexer implements EventSubscriberInterface
  26. {
  27.     /**
  28.      * @var IteratorFactory
  29.      */
  30.     private $iteratorFactory;
  32.     /**
  33.      * @var Connection
  34.      */
  35.     private $connection;
  37.     /**
  38.      * @var EntityRepositoryInterface
  39.      */
  40.     private $repository;
  42.     /**
  43.      * @var RulePayloadUpdater
  44.      */
  45.     private $payloadUpdater;
  47.     /**
  48.      * @var EventDispatcherInterface
  49.      */
  50.     private $eventDispatcher;
  52.     /**
  53.      * @var CartRuleLoader
  54.      */
  55.     private $cartRuleLoader;
  57.     public function __construct(
  58.         Connection $connection,
  59.         IteratorFactory $iteratorFactory,
  60.         EntityRepositoryInterface $repository,
  61.         RulePayloadUpdater $payloadUpdater,
  62.         CartRuleLoader $cartRuleLoader,
  63.         EventDispatcherInterface $eventDispatcher
  64.     ) {
  65.         $this->iteratorFactory = $iteratorFactory;
  66.         $this->repository = $repository;
  67.         $this->connection = $connection;
  68.         $this->payloadUpdater = $payloadUpdater;
  69.         $this->eventDispatcher = $eventDispatcher;
  70.         $this->cartRuleLoader = $cartRuleLoader;
  71.     }
  73.     public function getName(): string
  74.     {
  75.         return 'rule.indexer';
  76.     }
  78.     public static function getSubscribedEvents(): array
  79.     {
  80.         return [
  81.             PluginPostInstallEvent::class => 'refreshPlugin',
  82.             PluginPostActivateEvent::class => 'refreshPlugin',
  83.             PluginPostUpdateEvent::class => 'refreshPlugin',
  84.             PluginPostDeactivateEvent::class => 'refreshPlugin',
  85.             PluginPostUninstallEvent::class => 'refreshPlugin',
  86.             RuleEvents::RULE_WRITTEN_EVENT => 'onRuleWritten',
  87.         ];
  88.     }
  90.     public function refreshPlugin(): void
  91.     {
  92.         // Delete the payload and invalid flag of all rules
  93.         $update = new RetryableQuery(
  94.             $this->connection->prepare('UPDATE `rule` SET `payload` = null, `invalid` = 0')
  95.         );
  96.         $update->execute();
  97.     }
  99.     public function iterate($offset): ?EntityIndexingMessage
  100.     {
  101.         $iterator = $this->iteratorFactory->createIterator($this->repository->getDefinition(), $offset);
  103.         $ids = $iterator->fetch();
  105.         if (empty($ids)) {
  106.             return null;
  107.         }
  109.         return new RuleIndexingMessage(array_values($ids), $iterator->getOffset());
  110.     }
  112.     public function update(EntityWrittenContainerEvent $event): ?EntityIndexingMessage
  113.     {
  114.         $updates = $event->getPrimaryKeys(RuleDefinition::ENTITY_NAME);
  116.         if (empty($updates)) {
  117.             return null;
  118.         }
  120.         $this->handle(new RuleIndexingMessage(array_values($updates), null, $event->getContext()));
  122.         return null;
  123.     }
  125.     public function handle(EntityIndexingMessage $message): void
  126.     {
  127.         $ids = $message->getData();
  129.         $ids = array_unique(array_filter($ids));
  130.         if (empty($ids)) {
  131.             return;
  132.         }
  134.         $this->payloadUpdater->update($ids);
  136.         $this->eventDispatcher->dispatch(new RuleIndexerEvent($ids, $message->getContext()));
  137.     }
  139.     public function onRuleWritten(EntityWrittenEvent $event): void
  140.     {
  141.         $this->cartRuleLoader->reset();
  142.     }
  143. }