Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Doctrine: Counting an entity's items with a condition

How can I count an entity's items with a condition in Doctrine? For example, I realize that I can use:

$usersCount = $dm->getRepository('User')->count();

But that will only count all users. I would like to count only those that have type employee. I could do something like:

$users = $dm->getRepository('User')->findBy(array('type' => 'employee'));
$users = count($users);

That works but it's not optimal. Is there something like the following:?

$usersCount = $dm->getRepository('User')->count()->where('type', 'employee');
like image 949
luqita Avatar asked Sep 30 '13 20:09

luqita


2 Answers

Well, you could use the QueryBuilder to setup a COUNT query:

Presuming that $dm is your entity manager.

$qb = $dm->createQueryBuilder();

$qb->select($qb->expr()->count('u'))
   ->from('User', 'u')
   ->where('u.type = ?1')
   ->setParameter(1, 'employee');

$query = $qb->getQuery();

$usersCount = $query->getSingleScalarResult();

Or you could just write it in DQL:

$query = $dm->createQuery("SELECT COUNT(u) FROM User u WHERE u.type = ?1");
$query->setParameter(1, 'employee');

$usersCount = $query->getSingleScalarResult();

The counts might need to be on the id field, rather than the object, can't recall. If so just change the COUNT(u) or ->count('u') to COUNT(u.id) or ->count('u.id') or whatever your primary key field is called.

like image 68
Orbling Avatar answered Nov 06 '22 05:11

Orbling


This question is 3 years old but there is a way to keep the simplicity of the findBy() for count with criteria.

On your repository you can add this method :

    public function countBy(array $criteria)
    {
        $persister = $this->_em->getUnitOfWork()->getEntityPersister($this->_entityName);
        return $persister->count($criteria);
    }

So your code will looks like this :

$criteria = ['type' => 'employee'];
$users = $repository->findBy($criteria, ['name' => 'ASC'], 0, 20);
$nbUsers = $repository->countBy($criteria);
like image 16
Bacteries Avatar answered Nov 06 '22 06:11

Bacteries