custom/plugins/SwagEnterpriseSearchPlatform/src/Boosting/DataAbstractionLayer/BoostingDeletionSubscriber.php line 43

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Swag\EnterpriseSearch\Boosting\DataAbstractionLayer;
  5. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepositoryInterface;
  6. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  7. use Shopware\Core\Framework\DataAbstractionLayer\Write\Command\DeleteCommand;
  8. use Shopware\Core\Framework\DataAbstractionLayer\Write\Command\WriteCommand;
  9. use Shopware\Core\Framework\DataAbstractionLayer\Write\Validation\PreWriteValidationEvent;
  10. use Shopware\Core\Framework\Uuid\Uuid;
  11. use Swag\EnterpriseSearch\Boosting\BoostingDefinition;
  12. use Swag\EnterpriseSearch\Boosting\BoostingEntity;
  13. use Swag\EnterpriseSearch\Boosting\EntityStream\Service\EntityStreamResolver;
  14. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  16. class BoostingDeletionSubscriber implements EventSubscriberInterface
  17. {
  18.     /**
  19.      * @var EntityRepositoryInterface
  20.      */
  21.     private $entityStreamRepository;
  23.     /**
  24.      * @var EntityRepositoryInterface
  25.      */
  26.     private $boostingRepository;
  28.     public function __construct(
  29.         EntityRepositoryInterface $entityStreamRepository,
  30.         EntityRepositoryInterface $boostingRepository
  31.     ) {
  32.         $this->entityStreamRepository = $entityStreamRepository;
  33.         $this->boostingRepository = $boostingRepository;
  34.     }
  36.     public static function getSubscribedEvents(): array
  37.     {
  38.         return [
  39.             PreWriteValidationEvent::class => 'checkAndDeleteEntityStream',
  40.         ];
  41.     }
  43.     public function checkAndDeleteEntityStream(PreWriteValidationEvent $preWriteValidationEvent): void
  44.     {
  45.         $commands = $preWriteValidationEvent->getCommands();
  46.         $boostingIds = $this->getBoostingIdsFromDeleteCommands($commands);
  48.         if (!$boostingIds) {
  49.             return;
  50.         }
  52.         $boostingIds = Uuid::fromBytesToHexList($boostingIds);
  54.         $idsToDelete = $this->getEntityStreamIdsFromBoostingIds($boostingIds, $preWriteValidationEvent);
  56.         if (!$idsToDelete) {
  57.             return;
  58.         }
  60.         $this->entityStreamRepository->delete($idsToDelete, $preWriteValidationEvent->getContext());
  61.     }
  63.     /**
  64.      * @param WriteCommand[] $commands
  65.      */
  66.     private function getBoostingIdsFromDeleteCommands(array $commands): array
  67.     {
  68.         $boostingIds = [];
  70.         foreach ($commands as $command) {
  71.             if (!$command instanceof DeleteCommand) {
  72.                 continue;
  73.             }
  75.             if (!$command->getDefinition() instanceof BoostingDefinition) {
  76.                 continue;
  77.             }
  79.             $boostingIds = array_merge($command->getPrimaryKey(), $boostingIds);
  80.         }
  82.         return $boostingIds;
  83.     }
  85.     /**
  86.      * @param string[] $boostingIds
  87.      */
  88.     private function getEntityStreamIdsFromBoostingIds(
  89.         array $boostingIds,
  90.         PreWriteValidationEvent $preWriteValidationEvent
  91.     ): array {
  92.         $criteria = new Criteria($boostingIds);
  93.         $boostingCollection = $this->boostingRepository->search($criteria, $preWriteValidationEvent->getContext());
  94.         $idsToDelete = [];
  96.         /** @var BoostingEntity $boosting */
  97.         foreach ($boostingCollection->getElements() as $boosting) {
  98.             if ($boosting->getType() !== EntityStreamResolver::TYPE) {
  99.                 continue;
  100.             }
  102.             $idsToDelete[] = ['id' => $boosting->getTypeId()];
  103.         }
  105.         return $idsToDelete;
  106.     }
  107. }