src/Repository/ProfileRecommendationRepository.php line 19

Open in your IDE?
  1. <?php
  3. namespace App\Repository;
  5. use App\Entity\Profile\Profile;
  6. use App\Specification\ElasticSearch\Spec;
  7. use FOS\ElasticaBundle\Finder\TransformedFinder;
  8. use FOS\ElasticaBundle\HybridResult;
  9. use Symfony\Component\DependencyInjection\ContainerInterface;
  11. class ProfileRecommendationRepository
  12. {
  13.     private ?array $lastQuery = null;
  14.     private array $scores = [];
  15.     private TransformedFinder $finder;
  17.     public function __construct(ContainerInterface $container)
  18.     {
  19.         $this->finder = $container->get('fos_elastica.finder.recommendations.profile');
  20.     }
  22.     /**
  23.      * @return mixed[]
  24.      */
  25.     public function getByRecommendationWeightsData(Spec $mainQuery, array $filtersQuery, ?array $customSpecQuery, $offset, $limit): array
  26.     {
  27.         $query = [
  28.             'query' => [
  29.                 'function_score' => [
  30.                     'query' => [
  31.                         //... required conditions
  32.                     ],
  33.                     'functions' => [
  34.                         //... weight filters
  35.                     ],
  36.                     'max_boost' => 99999,
  37.                     'score_mode' => 'sum',
  38.                     'boost_mode' => 'sum',
  39.                     'min_score' => 0,
  40.                 ],
  41.             ],
  42.             'from' => $offset,
  43.             'size' => $limit,
  44.         ];
  46.         $mainQuery = $mainQuery->toEsQueryObject();
  48.         if(key($mainQuery) == 'bool')
  49.             $query['query']['function_score']['query'] = $mainQuery;
  50.         else
  51.             $query['query']['function_score']['query']['bool']['must'] = $mainQuery;
  53.         $customQueryIsEmpty = isset($customSpecQuery['bool']) && (empty($customSpecQuery['bool']) || empty(current($customSpecQuery['bool'])));
  54.         if($customSpecQuery && false === $customQueryIsEmpty) {
  55.             $query['query']['function_score']['query']['bool']['must'][] = [$customSpecQuery];
  56.         }
  58.         $query['query']['function_score']['functions'] = $filtersQuery;
  60.         //print_r(json_encode($query, JSON_PRETTY_PRINT));die;
  61.         $this->lastQuery = $query;
  62.         //$this->debugLog->checkQuery($this->lastQuery);
  64.         $result = $this->finder->findHybrid($query);
  66.         $profiles = [];
  67.         foreach ($result as /** @var HybridResult $item */ $item) {
  68.             /** @var Profile $profile */
  69.             $profile = $item->getTransformed();
  70.             $profiles[] = $profile;
  71.             $this->scores[$profile->getId()] = $item->getResult()->getScore();
  72.         }
  74.         return $profiles;
  75.     }
  77.     public function getLastQuery(): ?array
  78.     {
  79.         return $this->lastQuery;
  80.     }
  82.     public function getScores(): array
  83.     {
  84.         return $this->scores;
  85.     }
  86. }