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

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\Core\Framework\Log\Package;
  13. use Shopware\Elasticsearch\Framework\DataAbstractionLayer\Event\ElasticsearchEntityAggregatorSearchEvent;
  14. use Shopware\Elasticsearch\Framework\ElasticsearchHelper;
  15. use Symfony\Contracts\EventDispatcher\EventDispatcherInterface;
  17. #[Package('core')]
  18. class ElasticsearchEntityAggregator implements EntityAggregatorInterface
  19. {
  20.     public const RESULT_STATE = 'loaded-by-elastic';
  22.     private ElasticsearchHelper $helper;
  24.     private Client $client;
  26.     private EntityAggregatorInterface $decorated;
  28.     private AbstractElasticsearchAggregationHydrator $hydrator;
  30.     private EventDispatcherInterface $eventDispatcher;
  32.     /**
  33.      * @internal
  34.      */
  35.     public function __construct(
  36.         ElasticsearchHelper $helper,
  37.         Client $client,
  38.         EntityAggregatorInterface $decorated,
  39.         AbstractElasticsearchAggregationHydrator $hydrator,
  40.         EventDispatcherInterface $eventDispatcher
  41.     ) {
  42.         $this->helper = $helper;
  43.         $this->client = $client;
  44.         $this->decorated = $decorated;
  45.         $this->hydrator = $hydrator;
  46.         $this->eventDispatcher = $eventDispatcher;
  47.     }
  49.     public function aggregate(EntityDefinition $definition, Criteria $criteria, Context $context): AggregationResultCollection
  50.     {
  51.         if (!$this->helper->allowSearch($definition, $context, $criteria)) {
  52.             return $this->decorated->aggregate($definition, $criteria, $context);
  53.         }
  55.         $search = $this->createSearch($definition, $criteria, $context);
  57.         $this->eventDispatcher->dispatch(
  58.             new ElasticsearchEntityAggregatorSearchEvent($search, $definition, $criteria, $context)
  59.         );
  61.         try {
  62.             $result = $this->client->search([
  63.                 'index' => $this->helper->getIndexName($definition, $context->getLanguageId()),
  64.                 'body' => $search->toArray(),
  65.             ]);
  66.         } catch (\Throwable $e) {
  67.             $this->helper->logAndThrowException($e);
  69.             return $this->decorated->aggregate($definition, $criteria, $context);
  70.         }
  72.         $result = $this->hydrator->hydrate($definition, $criteria, $context, $result);
  73.         $result->addState(self::RESULT_STATE);
  75.         return $result;
  76.     }
  78.     private function createSearch(EntityDefinition $definition, Criteria $criteria, Context $context): Search
  79.     {
  80.         $search = new Search();
  81.         $this->helper->addFilters($definition, $criteria, $search, $context);
  82.         $this->helper->addQueries($definition, $criteria, $search, $context);
  83.         $this->helper->addAggregations($definition, $criteria, $search, $context);
  84.         $this->helper->addTerm($criteria, $search, $context, $definition);
  85.         $this->helper->handleIds($definition, $criteria, $search, $context);
  86.         $search->setSize(0);
  88.         return $search;
  89.     }
  90. }