src/Elasticsearch/Framework/DataAbstractionLayer/ElasticsearchEntityAggregator.php line 32

  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(private readonly ElasticsearchHelper $helper, private readonly Client $client, private readonly EntityAggregatorInterface $decorated, private readonly AbstractElasticsearchAggregationHydrator $hydrator, private readonly EventDispatcherInterface $eventDispatcher)
  26.     {
  27.     }
  29.     public function aggregate(EntityDefinition $definition, Criteria $criteria, Context $context): AggregationResultCollection
  30.     {
  31.         if (!$this->helper->allowSearch($definition, $context, $criteria)) {
  32.             return $this->decorated->aggregate($definition, $criteria, $context);
  33.         }
  35.         $search = $this->createSearch($definition, $criteria, $context);
  37.         $this->eventDispatcher->dispatch(
  38.             new ElasticsearchEntityAggregatorSearchEvent($search, $definition, $criteria, $context)
  39.         );
  41.         try {
  42.             $result = $this->client->search([
  43.                 'index' => $this->helper->getIndexName($definition, $context->getLanguageId()),
  44.                 'body' => $search->toArray(),
  45.             ]);
  46.         } catch (\Throwable $e) {
  47.             $this->helper->logAndThrowException($e);
  49.             return $this->decorated->aggregate($definition, $criteria, $context);
  50.         }
  52.         $result = $this->hydrator->hydrate($definition, $criteria, $context, $result);
  53.         $result->addState(self::RESULT_STATE);
  55.         return $result;
  56.     }
  58.     private function createSearch(EntityDefinition $definition, Criteria $criteria, Context $context): Search
  59.     {
  60.         $search = new Search();
  61.         $this->helper->addFilters($definition, $criteria, $search, $context);
  62.         $this->helper->addQueries($definition, $criteria, $search, $context);
  63.         $this->helper->addAggregations($definition, $criteria, $search, $context);
  64.         $this->helper->addTerm($criteria, $search, $context, $definition);
  65.         $this->helper->handleIds($definition, $criteria, $search, $context);
  66.         $search->setSize(0);
  68.         return $search;
  69.     }
  70. }