61 lines
1.9 KiB
PHP
61 lines
1.9 KiB
PHP
<?php
|
|
|
|
namespace FOS\ElasticaBundle\Doctrine\ORM;
|
|
|
|
use Doctrine\ORM\QueryBuilder;
|
|
use FOS\ElasticaBundle\Doctrine\AbstractProvider;
|
|
use FOS\ElasticaBundle\Exception\InvalidArgumentTypeException;
|
|
|
|
class Provider extends AbstractProvider
|
|
{
|
|
/**
|
|
* @see FOS\ElasticaBundle\Doctrine\AbstractProvider::countObjects()
|
|
*/
|
|
protected function countObjects($queryBuilder)
|
|
{
|
|
if (!$queryBuilder instanceof QueryBuilder) {
|
|
throw new InvalidArgumentTypeException($queryBuilder, 'Doctrine\ORM\QueryBuilder');
|
|
}
|
|
|
|
/* Clone the query builder before altering its field selection and DQL,
|
|
* lest we leave the query builder in a bad state for fetchSlice().
|
|
*/
|
|
$qb = clone $queryBuilder;
|
|
|
|
return $qb
|
|
->select($qb->expr()->count($queryBuilder->getRootAlias()))
|
|
// Remove ordering for efficiency; it doesn't affect the count
|
|
->resetDQLPart('orderBy')
|
|
->getQuery()
|
|
->getSingleScalarResult();
|
|
}
|
|
|
|
/**
|
|
* @see FOS\ElasticaBundle\Doctrine\AbstractProvider::fetchSlice()
|
|
*/
|
|
protected function fetchSlice($queryBuilder, $limit, $offset)
|
|
{
|
|
if (!$queryBuilder instanceof QueryBuilder) {
|
|
throw new InvalidArgumentTypeException($queryBuilder, 'Doctrine\ORM\QueryBuilder');
|
|
}
|
|
|
|
return $queryBuilder
|
|
->setFirstResult($offset)
|
|
->setMaxResults($limit)
|
|
->getQuery()
|
|
->getResult();
|
|
}
|
|
|
|
/**
|
|
* @see FOS\ElasticaBundle\Doctrine\AbstractProvider::createQueryBuilder()
|
|
*/
|
|
protected function createQueryBuilder()
|
|
{
|
|
return $this->managerRegistry
|
|
->getManagerForClass($this->objectClass)
|
|
->getRepository($this->objectClass)
|
|
// ORM query builders require an alias argument
|
|
->{$this->options['query_builder_method']}('a');
|
|
}
|
|
}
|