src/Repository/ProfileRecommendationRepository.php line 18

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 array $explains = [];
  16.     private TransformedFinder $finder;
  18.     public function __construct(ContainerInterface $container)
  19.     {
  20.         $this->finder = $container->get('fos_elastica.finder.recommendations.profile');
  21.     }
  23.     /**
  24.      * @return mixed[]
  25.      */
  26.     public function getByRecommendationWeightsData(Spec $mainQuery, array $filtersQuery, ?array $customSpecQuery, $offset, $limit): array
  27.     {
  28.         $query = [
  29.             'query' => [
  30.                 'function_score' => [
  31.                     'query' => [
  32.                         //... required conditions
  33.                     ],
  34.                     'functions' => [
  35.                         //... weight filters
  36.                     ],
  37.                     'max_boost' => 99999,
  38.                     'score_mode' => 'sum',
  39.                     'boost_mode' => 'replace',
  40.                     'min_score' => 0,
  41.                 ],
  42.             ],
  43.             'from' => $offset,
  44.             'size' => $limit,
  45.             'explain' => true,
  46.         ];
  48.         $mainQuery = $mainQuery->toEsQueryObject();
  50.         if(key($mainQuery) == 'bool')
  51.             $query['query']['function_score']['query'] = $mainQuery;
  52.         else
  53.             $query['query']['function_score']['query']['bool']['must'] = $mainQuery;
  55.         $customQueryIsEmpty = isset($customSpecQuery['bool']) && (empty($customSpecQuery['bool']) || empty(current($customSpecQuery['bool'])));
  56.         if($customSpecQuery && false === $customQueryIsEmpty) {
  57.             $query['query']['function_score']['query']['bool']['must'][] = [$customSpecQuery];
  58.         }
  60.         $query['query']['function_score']['functions'] = $filtersQuery;
  62.         //print_r(json_encode($query, JSON_PRETTY_PRINT));die;
  63.         $this->lastQuery = $query;
  64.         //$this->debugLog->checkQuery($this->lastQuery);
  66.         $result = $this->finder->findHybrid($query);
  68.         $profiles = [];
  69.         foreach ($result as /** @var HybridResult $item */ $item) {
  70.             /** @var Profile $profile */
  71.             $profile = $item->getTransformed();
  72.             $profiles[] = $profile;
  73.             $this->scores[$profile->getId()] = $item->getResult()->getScore();
  74.             $this->explains[$profile->getId()] = json_encode($item->getResult()->getExplanation());
  75.         }
  76. dump($this->explains);
  77.         return $profiles;
  78.     }
  80.     public function getLastQuery(): ?array
  81.     {
  82.         return $this->lastQuery;
  83.     }
  85.     public function getScores(): array
  86.     {
  87.         return $this->scores;
  88.     }
  90.     public function getExplains(): array
  91.     {
  92.         return $this->explains;
  93.     }
  94. }