原则:有条件地计算实体的项目


Doctrine: Counting an entity's items with a condition

如何在教义中有条件的情况下计算实体的项目?例如,我意识到我可以使用:

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

但这只会计算所有用户。我只想计算那些有员工类型的人。我可以做这样的事情:

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

这有效,但不是最佳的。有没有像下面这样的东西:?

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

好吧,您可以使用QueryBuilder来设置COUNT查询:

假设$dm是您的实体经理。

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

或者你可以用DQL写它:

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

计数可能需要在 id 字段上,而不是对象上,无法回忆。 如果是这样,只需将COUNT(u)->count('u')更改为COUNT(u.id)->count('u.id')或任何主键字段的名称。

这个问题已经有 3 年的历史了,但有一种方法可以保持 findBy() 的简单性,以便使用标准进行计数。

在您的存储库中,您可以添加此方法:

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

所以你的代码将看起来像这样:

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

从原则 2.6 开始,当您不需要数据时,您可以只计算所提供条件的结果:

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

升级到 2.6

刚刚进入一个旧项目,还需要在 Where 条件已经设置时获取 QueryBuilder 的计数。

我在服务中添加了以下函数以获得结果:

public function getTotalRecordsFromQueryBuilder(QueryBuilder $queryBuilder): int
{
    $alias = $queryBuilder->getRootAliases()[0];
    $clone = clone $queryBuilder;
    $clone->select(sprintf('COUNT(%s)', $alias));
    return $clone->getQuery()->getSingleScalarResult();
}

我克隆了查询构建器,以免为将来的添加而更改它。

然后我在添加我的 Where 条件后调用该函数

$queryBuilder = $manager->createQueryBuilder()
            ->select('import')
            ->from('App:Import', 'import');
$dataTableService->addWhereDQLFromRequest($request, $queryBuilder, $sortColumns);
$dataTableService->addOrderDQLFromRequest($request, $queryBuilder, $sortColumns);
$totalRecords = $dataTableService->getTotalRecordsFromQueryBuilder($queryBuilder);
$dataTableService->addLimitDQLFromRequest($request, $queryBuilder);