Количество строк в Doctrine QueryBuilder
Я использую QueryBuilder от Doctrine для создания запроса и хочу получить общее количество результатов запроса.
$repository = $em->getRepository('FooBundle:Foo');
$qb = $repository->createQueryBuilder('n')
->where('n.bar = :bar')
->setParameter('bar', $bar);
$query = $qb->getQuery();
//this doesn't work
$totalrows = $query->getResult()->count();
Я просто хочу выполнить подсчет этого запроса, чтобы получить общее количество строк, но не вернуть фактические результаты. (После этого запроса количества я собираюсь дополнительно изменить запрос с maxResults для разбивки на страницы.)
11 ответов
Что-то вроде:
$qb = $entityManager->createQueryBuilder();
$qb->select('count(account.id)');
$qb->from('ZaysoCoreBundle:Account','account');
$count = $qb->getQuery()->getSingleScalarResult();
РЕДАКТИРОВАТЬ
Некоторые люди считают, что выражения как-то лучше, чем просто использование DQL. Один даже зашел так далеко, что отредактировал четырехлетний ответ. Я откатил его правку обратно. Пойди разберись.
Вот еще один способ отформатировать запрос:
return $repository->createQueryBuilder('u')
->select('count(u.id)')
->getQuery()
->getSingleScalarResult();
Лучше перенести всю логику работы с базой данных на репозитории.
Так в контроллере пишешь
/* you can also inject "FooRepository $repository" using autowire */
$repository = $this->getDoctrine()->getRepository(Foo::class);
$count = $repository->count();
И в Repository/FooRepository.php
public function count()
{
$qb = $repository->createQueryBuilder('t');
return $qb
->select('count(t.id)')
->getQuery()
->getSingleScalarResult();
}
Лучше двигаться $qb = ... разделить строку в случае, если вы хотите сделать сложные выражения, такие как
public function count()
{
$qb = $repository->createQueryBuilder('t');
return $qb
->select('count(t.id)')
->where($qb->expr()->isNotNull('t.fieldName'))
->andWhere($qb->expr()->orX(
$qb->expr()->in('t.fieldName2', 0),
$qb->expr()->isNull('t.fieldName2')
))
->getQuery()
->getSingleScalarResult();
}
Также подумайте о кэшировании результатов вашего запроса - http://symfony.com/doc/current/reference/configuration/doctrine.html
public function count()
{
$qb = $repository->createQueryBuilder('t');
return $qb
->select('count(t.id)')
->getQuery()
->useQueryCache(true)
->useResultCache(true, 3600)
->getSingleScalarResult();
}
В некоторых простых случаях, используя EXTRA_LAZY отношения сущности это хорошо
http://doctrine-orm.readthedocs.org/projects/doctrine-orm/en/latest/tutorials/extra-lazy-associations.html
Если вам нужно посчитать более сложный запрос, с groupBy, having и т.д... Вы можете взять у Doctrine\ORM\Tools\Pagination\Paginator:
$paginator = new \Doctrine\ORM\Tools\Pagination\Paginator($query);
$totalRows = count($paginator);
Поскольку Doctrine 2.6 можно использовать count() метод непосредственно из EntityRepository, Подробности смотрите по ссылке.
Пример работы с группировкой, объединением и прочим.
Проблема:
$qb = $em->createQueryBuilder()
->select('m.id', 'rm.id')
->from('Model', 'm')
->join('m.relatedModels', 'rm')
->groupBy('m.id');
Для того, чтобы это работало, возможное решение состоит в том, чтобы использовать собственный гидратор и эту странную вещь, называемую "ПОДСКАЗКА ВЫХОДНОГО ВЫХОДА":
class CountHydrator extends AbstractHydrator
{
const NAME = 'count_hydrator';
const FIELD = 'count';
/**
* {@inheritDoc}
*/
protected function hydrateAllData()
{
return (int)$this->_stmt->fetchColumn(0);
}
}
class CountSqlWalker extends SqlWalker
{
/**
* {@inheritDoc}
*/
public function walkSelectStatement(AST\SelectStatement $AST)
{
return sprintf("SELECT COUNT(*) AS %s FROM (%s) AS t", CountHydrator::FIELD, parent::walkSelectStatement($AST));
}
}
$doctrineConfig->addCustomHydrationMode(CountHydrator::NAME, CountHydrator::class);
// $qb from example above
$countQuery = clone $qb->getQuery();
// Doctrine bug ? Doesn't make a deep copy... (as of "doctrine/orm": "2.4.6")
$countQuery->setParameters($this->getQuery()->getParameters());
// set custom 'hint' stuff
$countQuery->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, CountSqlWalker::class);
$count = $countQuery->getResult(CountHydrator::NAME);
Для людей, которые используют только Doctrine DBAL, а не Doctrine ORM, они не смогут получить доступ к getQuery() метод, потому что он не существует. Им нужно сделать что-то вроде следующего.
$qb = new QueryBuilder($conn);
$count = $qb->select("count(id)")->from($tableName)->execute()->fetchColumn(0);
Для подсчета элементов после некоторого количества элементов (смещение) в этом случае нельзя применить $qb->setFirstResults(), так как он работает не как условие запроса, а как смещение результата запроса для диапазона выбранных элементов (то есть setFirstResult вообще не может использоваться вместе с COUNT). Поэтому для подсчета оставшихся предметов я просто сделал следующее:
//in repository class:
$count = $qb->select('count(p.id)')
->from('Products', 'p')
->getQuery()
->getSingleScalarResult();
return $count;
//in controller class:
$count = $this->em->getRepository('RepositoryBundle')->...
return $count-$offset;
Кто-нибудь знает более чистый способ сделать это?
Вы также можете получить количество данных, используя функцию подсчета.
$query = $this->dm->createQueryBuilder('AppBundle:Items')
->field('isDeleted')->equals(false)
->getQuery()->count();
Добавление следующего метода в ваш репозиторий позволит вам вызвать $repo->getCourseCount() от вашего контроллера.
/**
* @return array
*/
public function getCourseCount()
{
$qb = $this->getEntityManager()->createQueryBuilder();
$qb
->select('count(course.id)')
->from('CRMPicco\Component\Course\Model\Course', 'course')
;
$query = $qb->getQuery();
return $query->getSingleScalarResult();
}