vendor/pagerfanta/pagerfanta/lib/Adapter/Elastica/ElasticaAdapter.php line 98

Open in your IDE?
  1. <?php
  3. namespace Pagerfanta\Elastica;
  5. use Elastica\Query;
  6. use Elastica\ResultSet;
  7. use Elastica\SearchableInterface;
  8. use Pagerfanta\Adapter\AdapterInterface;
  10. /**
  11.  * Adapter which calculates pagination from a Elastica Query.
  12.  */
  13. class ElasticaAdapter implements AdapterInterface
  14. {
  15.     /**
  16.      * @var SearchableInterface
  17.      */
  18.     private $searchable;
  20.     /**
  21.      * @var Query
  22.      */
  23.     private $query;
  25.     /**
  26.      * @var array
  27.      */
  28.     private $options;
  30.     /**
  31.      * Used to limit the number of totalHits returned by ElasticSearch.
  32.      * For more information, see: https://github.com/whiteoctober/Pagerfanta/pull/213#issue-87631892.
  33.      *
  34.      * @var int|null
  35.      */
  36.     private $maxResults;
  38.     /**
  39.      * @var ResultSet|null
  40.      */
  41.     private $resultSet;
  43.     /**
  44.      * @param int|null $maxResults
  45.      */
  46.     public function __construct(SearchableInterface $searchable, Query $query, array $options = [], $maxResults = null)
  47.     {
  48.         $this->searchable = $searchable;
  49.         $this->query = $query;
  50.         $this->options = $options;
  51.         $this->maxResults = $maxResults;
  52.     }
  54.     /**
  55.      * Returns the Elastica ResultSet.
  56.      *
  57.      * Will return null if getSlice has not yet been called.
  58.      *
  59.      * @return ResultSet|null
  60.      */
  61.     public function getResultSet()
  62.     {
  63.         return $this->resultSet;
  64.     }
  66.     /**
  67.      * @return int
  68.      */
  69.     public function getNbResults()
  70.     {
  71.         if (!$this->resultSet) {
  72.             $totalHits = $this->searchable->count($this->query);
  73.         } else {
  74.             $totalHits = $this->resultSet->getTotalHits();
  75.         }
  77.         if (null === $this->maxResults) {
  78.             return $totalHits;
  79.         }
  81.         return min($totalHits, $this->maxResults);
  82.     }
  84.     /**
  85.      * @param int $offset
  86.      * @param int $length
  87.      *
  88.      * @return iterable
  89.      */
  90.     public function getSlice($offset, $length)
  91.     {
  92.         return $this->resultSet = $this->searchable->search(
  93.             $this->query,
  94.             array_merge(
  95.                 $this->options,
  96.                 [
  97.                     'from' => $offset,
  98.                     'size' => $length,
  99.                 ]
  100.             )
  101.         );
  102.     }
  103. }