web-dev-qa-db-fra.com

Doctrine: compter les objets d'une entité avec une condition

Comment puis-je compter les éléments d'une entité avec une condition dans Doctrine? Par exemple, je me rends compte que je peux utiliser:

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

Mais cela ne comptera que tous les utilisateurs. Je voudrais ne compter que ceux qui ont un type d'employé. Je pourrais faire quelque chose comme:

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

Cela fonctionne mais ce n'est pas optimal. Y a-t-il quelque chose comme ceci:?

$usersCount = $dm->getRepository('User')->count()->where('type', 'employee');
22
luqita

Eh bien, vous pouvez utiliser QueryBuilder pour configurer une requête COUNT:

En supposant que $dm Est votre gestionnaire d'entité.

$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();

Ou vous pouvez simplement l'écrire dans DQL :

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

$usersCount = $query->getSingleScalarResult();

Les comptes peuvent devoir être sur le champ id, plutôt que sur l'objet, je ne me souviens pas. Si c'est le cas, changez simplement la COUNT(u) ou ->count('u') en COUNT(u.id) ou ->count('u.id') ou quel que soit votre champ de clé primaire.

22
Orbling

Cette question a 3 ans mais il existe un moyen de conserver la simplicité du findBy () pour compter avec des critères.

Sur votre référentiel, vous pouvez ajouter cette méthode:

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

Ainsi, votre code ressemblera à ceci:

$criteria = ['type' => 'employee'];
$users = $repository->findBy($criteria, ['name' => 'ASC'], 0, 20);
$nbUsers = $repository->countBy($criteria);
14
Bacteries