vendor/store.shopware.com/swagenterprisesearchplatform/src/Boosting/BoostingQueryBuilder.php line 45

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Swag\EnterpriseSearch\Boosting;
  5. use ONGR\ElasticsearchDSL\BuilderInterface;
  6. use ONGR\ElasticsearchDSL\Query\Compound\BoolQuery;
  7. use ONGR\ElasticsearchDSL\Query\Compound\ConstantScoreQuery;
  8. use ONGR\ElasticsearchDSL\Search;
  9. use Shopware\Core\Framework\Context;
  10. use Shopware\Core\Framework\DataAbstractionLayer\EntityDefinition;
  11. use Shopware\Core\Framework\DataAbstractionLayer\Exception\SearchRequestException;
  12. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  13. use Shopware\Core\Framework\DataAbstractionLayer\Search\Parser\QueryStringParser;
  14. use Shopware\Elasticsearch\Framework\DataAbstractionLayer\CriteriaParser;
  15. use Swag\EnterpriseSearch\Common\SesExtension;
  17. /**
  18.  * Handles a given boosting struct and returns the corresponding ES query object for the `boostingRule`
  19.  */
  20. class BoostingQueryBuilder
  21. {
  22.     public const FILTER = 'filter';
  23.     public const BOOSTING = 'boosting';
  25.     /**
  26.      * @var BoostingProvider
  27.      */
  28.     private $boostingProvider;
  30.     /**
  31.      * @var CriteriaParser
  32.      */
  33.     private $criteriaParser;
  35.     public function __construct(
  36.         BoostingProvider $boostingProvider,
  37.         CriteriaParser $criteriaParser
  38.     ) {
  39.         $this->boostingProvider = $boostingProvider;
  40.         $this->criteriaParser = $criteriaParser;
  41.     }
  43.     public function addBoostingQuery(EntityDefinition $definition, Criteria $criteria, Search $search, Context $context): void
  44.     {
  45.         $boostingCombinations = $this->boostingProvider->getActiveBoostings($definition, $context);
  47.         foreach ($boostingCombinations as $boostingCombination) {
  48.             $filters = $boostingCombination[self::FILTER];
  49.             /** @var BoostingEntity $boostingEntity */
  50.             $boostingEntity = $boostingCombination[self::BOOSTING];
  52.             if (empty($filters)) {
  53.                 continue;
  54.             }
  56.             if (!$boostingEntity->isActive() || !$boostingEntity->isValid()) {
  57.                 continue;
  58.             }
  60.             $search->addQuery($this->getBoostingQuery($boostingEntity, $definition, $context, $filters), BoolQuery::SHOULD);
  61.         }
  62.     }
  64.     protected function getBoostingQuery(BoostingEntity $boosting, EntityDefinition $definition, Context $context, array $filters): BuilderInterface
  65.     {
  66.         $filter = QueryStringParser::fromArray($definition, $filters, new SearchRequestException());
  68.         $query = new ConstantScoreQuery($this->criteriaParser->parseFilter($filter, $definition, $definition->getEntityName(), $context));
  70.         $query->addParameter('boost', $boosting->getBoost());
  72.         if (SesExtension::isExplanation($context)) {
  73.             $boostingValue = (string) $boosting->getBoost();
  75.             $query->addParameter('_name', 'Boosting ' . $boosting->getTranslation('name') . ' ' . $boostingValue);
  76.         }
  78.         return $query;
  79.     }
  80. }