vendor/shopware/elasticsearch/Framework/DataAbstractionLayer/ElasticsearchEntityAggregator.php line 50

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Shopware\Elasticsearch\Framework\DataAbstractionLayer;
  5. use Elasticsearch\Client;
  6. use ONGR\ElasticsearchDSL\Search;
  7. use Shopware\Core\Framework\Context;
  8. use Shopware\Core\Framework\DataAbstractionLayer\EntityDefinition;
  9. use Shopware\Core\Framework\DataAbstractionLayer\Search\AggregationResult\AggregationResultCollection;
  10. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  11. use Shopware\Core\Framework\DataAbstractionLayer\Search\EntityAggregatorInterface;
  12. use Shopware\Elasticsearch\Framework\DataAbstractionLayer\Event\ElasticsearchEntityAggregatorSearchEvent;
  13. use Shopware\Elasticsearch\Framework\ElasticsearchHelper;
  14. use Symfony\Contracts\EventDispatcher\EventDispatcherInterface;
  16. /**
  17.  * @package core
  18.  */
  19. class ElasticsearchEntityAggregator implements EntityAggregatorInterface
  20. {
  21.     public const RESULT_STATE = 'loaded-by-elastic';
  23.     private ElasticsearchHelper $helper;
  25.     private Client $client;
  27.     private EntityAggregatorInterface $decorated;
  29.     private AbstractElasticsearchAggregationHydrator $hydrator;
  31.     private EventDispatcherInterface $eventDispatcher;
  33.     /**
  34.      * @internal
  35.      */
  36.     public function __construct(
  37.         ElasticsearchHelper $helper,
  38.         Client $client,
  39.         EntityAggregatorInterface $decorated,
  40.         AbstractElasticsearchAggregationHydrator $hydrator,
  41.         EventDispatcherInterface $eventDispatcher
  42.     ) {
  43.         $this->helper = $helper;
  44.         $this->client = $client;
  45.         $this->decorated = $decorated;
  46.         $this->hydrator = $hydrator;
  47.         $this->eventDispatcher = $eventDispatcher;
  48.     }
  50.     public function aggregate(EntityDefinition $definition, Criteria $criteria, Context $context): AggregationResultCollection
  51.     {
  52.         if (!$this->helper->allowSearch($definition, $context, $criteria)) {
  53.             return $this->decorated->aggregate($definition, $criteria, $context);
  54.         }
  56.         $search = $this->createSearch($definition, $criteria, $context);
  58.         $this->eventDispatcher->dispatch(
  59.             new ElasticsearchEntityAggregatorSearchEvent($search, $definition, $criteria, $context)
  60.         );
  62.         try {
  63.             $result = $this->client->search([
  64.                 'index' => $this->helper->getIndexName($definition, $context->getLanguageId()),
  65.                 'body' => $search->toArray(),
  66.             ]);
  67.         } catch (\Throwable $e) {
  68.             $this->helper->logAndThrowException($e);
  70.             return $this->decorated->aggregate($definition, $criteria, $context);
  71.         }
  73.         $result = $this->hydrator->hydrate($definition, $criteria, $context, $result);
  74.         $result->addState(self::RESULT_STATE);
  76.         return $result;
  77.     }
  79.     private function createSearch(EntityDefinition $definition, Criteria $criteria, Context $context): Search
  80.     {
  81.         $search = new Search();
  82.         $this->helper->addFilters($definition, $criteria, $search, $context);
  83.         $this->helper->addQueries($definition, $criteria, $search, $context);
  84.         $this->helper->addAggregations($definition, $criteria, $search, $context);
  85.         $this->helper->addTerm($criteria, $search, $context, $definition);
  86.         $this->helper->handleIds($definition, $criteria, $search, $context);
  87.         $search->setSize(0);
  89.         return $search;
  90.     }
  91. }