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

  1. <?php declare(strict_types=1);
  3. namespace Shopware\Elasticsearch\Framework\DataAbstractionLayer;
  5. use OpenSearch\Client;
  6. use OpenSearchDSL\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.     final public const RESULT_STATE = 'loaded-by-elastic';
  22.     /**
  23.      * @internal
  24.      */
  25.     public function __construct(
  26.         private readonly ElasticsearchHelper $helper,
  27.         private readonly Client $client,
  28.         private readonly EntityAggregatorInterface $decorated,
  29.         private readonly AbstractElasticsearchAggregationHydrator $hydrator,
  30.         private readonly EventDispatcherInterface $eventDispatcher
  31.     ) {
  32.     }
  34.     public function aggregate(EntityDefinition $definition, Criteria $criteria, Context $context): AggregationResultCollection
  35.     {
  36.         if (!$this->helper->allowSearch($definition, $context, $criteria)) {
  37.             return $this->decorated->aggregate($definition, $criteria, $context);
  38.         }
  40.         $search = $this->createSearch($definition, $criteria, $context);
  42.         $this->eventDispatcher->dispatch(
  43.             new ElasticsearchEntityAggregatorSearchEvent($search, $definition, $criteria, $context)
  44.         );
  46.         try {
  47.             $result = $this->client->search([
  48.                 'index' => $this->helper->getIndexName($definition, $context->getLanguageId()),
  49.                 'body' => $search->toArray(),
  50.             ]);
  51.         } catch (\Throwable $e) {
  52.             $this->helper->logAndThrowException($e);
  54.             return $this->decorated->aggregate($definition, $criteria, $context);
  55.         }
  57.         $result = $this->hydrator->hydrate($definition, $criteria, $context, $result);
  58.         $result->addState(self::RESULT_STATE);
  60.         return $result;
  61.     }
  63.     private function createSearch(EntityDefinition $definition, Criteria $criteria, Context $context): Search
  64.     {
  65.         $search = new Search();
  66.         $this->helper->addFilters($definition, $criteria, $search, $context);
  67.         $this->helper->addQueries($definition, $criteria, $search, $context);
  68.         $this->helper->addAggregations($definition, $criteria, $search, $context);
  69.         $this->helper->addTerm($criteria, $search, $context, $definition);
  70.         $this->helper->handleIds($definition, $criteria, $search, $context);
  71.         $search->setSize(0);
  73.         return $search;
  74.     }
  75. }