custom/plugins/SwagEnterpriseSearchPlatform/src/Indexing/SynonymUpdateSubscriber.php line 55

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Swag\EnterpriseSearch\Indexing;
  5. use Elasticsearch\Client;
  6. use Shopware\Core\Framework\Context;
  7. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepositoryInterface;
  8. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  9. use Shopware\Core\Framework\DataAbstractionLayer\Search\Filter\EqualsFilter;
  10. use Shopware\Core\Framework\DataAbstractionLayer\Search\Filter\NotFilter;
  11. use Shopware\Core\Framework\DataAbstractionLayer\Search\Sorting\FieldSorting;
  12. use Shopware\Elasticsearch\Framework\ElasticsearchHelper;
  13. use Shopware\Elasticsearch\Framework\ElasticsearchRegistry;
  14. use Swag\EnterpriseSearch\Synonym\SynonymProvider;
  15. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  17. class SynonymUpdateSubscriber implements EventSubscriberInterface
  18. {
  19.     private $languageRepository;
  21.     private $synonymProvider;
  23.     private $helper;
  25.     private $client;
  27.     private $registry;
  29.     /**
  30.      * @var array<int, string>
  31.      */
  32.     private $languages;
  34.     public function __construct(
  35.         ElasticsearchRegistry $registry,
  36.         ElasticsearchHelper $helper,
  37.         EntityRepositoryInterface $languageRepository,
  38.         Client $client,
  39.         SynonymProvider $synonymProvider
  40.     ) {
  41.         $this->registry = $registry;
  42.         $this->helper = $helper;
  43.         $this->languageRepository = $languageRepository;
  44.         $this->client = $client;
  45.         $this->synonymProvider = $synonymProvider;
  46.     }
  48.     public static function getSubscribedEvents(): array
  49.     {
  50.         return [
  51.             'synonym.written' => 'onSynonymWritten',
  52.         ];
  53.     }
  55.     public function onSynonymWritten(): void
  56.     {
  57.         if (empty($this->languages)) {
  58.             $this->languages = $this->getLanguages();
  59.         }
  61.         $indices = $this->client->indices();
  63.         $definitions = $this->registry->getDefinitions();
  64.         foreach ($this->languages as $languageId) {
  65.             foreach ($definitions as $definition) {
  66.                 $alias = $this->helper->getIndexName($definition->getEntityDefinition(), $languageId);
  67.                 $config = $this->getConfigTemplate($languageId);
  68.                 $indices->close(['index' => $alias]);
  69.                 $indices->putSettings(['index' => $alias, 'body' => $config]);
  70.                 $indices->open(['index' => $alias]);
  71.             }
  72.         }
  73.     }
  75.     private function getConfigTemplate(string $languageId): array
  76.     {
  77.         return [
  78.             'settings' => [
  79.                 'analysis' => [
  80.                     'filter' => [
  81.                         'synonym' => [
  82.                             'synonyms' => $this->synonymProvider->getSynonyms($languageId),
  83.                         ],
  84.                     ],
  85.                 ],
  86.             ],
  87.         ];
  88.     }
  90.     private function getLanguages(): array
  91.     {
  92.         $criteria = new Criteria();
  93.         $context = Context::createDefaultContext();
  95.         $criteria->addFilter(new NotFilter(NotFilter::CONNECTION_AND, [
  96.             new EqualsFilter('salesChannelDomains.id', null),
  97.         ]));
  99.         $criteria->addSorting(new FieldSorting('id'));
  101.         return $this->languageRepository->searchIds($criteria, $context)->getIds();
  102.     }
  103. }