更改主键值的 getResult 数组键

php doctrine-orm

2022-08-30 18:54:41

是否可以在 Doctrine2 中更改 getResult（）的数组键值？

例：

$qb->select('t.id, t.name')->from('Table', 't');

当我打印这个时，我得到，这不是我想要的：

print_r($qb->getQuery()->getResult()); //Print result: Array ( [0] => Array ( [id] => 20 [name] => Name1 ) [1] => Array ( [id] => 21 [percentagem] => Name2 ) )

我想要的是：

Array ( [20] => Array ( [id] => 20 [name] => Name1 ) [21] => Array ( [id] => 21 [percentagem] => Name2 ) )

建议，提示将不胜感激。

答案 1

我实际上对这个东西有多酷感到非常高兴：

$query = $this->getEntityManager()->createQuery('
            SELECT user FROM UserBundle:User user
            INDEX BY user.id
            WHERE user.id = 1
            '
        );

INDEX BY 构造不会直接转换为 SQL，但会影响对象和数组水合。在每个 FROM 和 JOIN 子句之后，指定应在结果中为此类编制索引的字段。默认情况下，结果按从 0 开始的数字键递增。但是，使用 INDEX BY，您可以将任何其他列指定为结果的键，但实际上它仅对主字段或唯一字段有意义。

资料来源：Doctrine ORM 2 Documentation Using INDEX BY

请在 WHERE 之前使用 INDEX BY

答案 2

但是，为了完整起见，您可以对查询生成器执行相同的操作，如下所示：

$queryBuilder = $this->getEntityManager()->createQueryBuilder();

$queryBuilder
    ->select('user')
    ->from('UserBundle:User', 'user', 'user.id')
    ->where('user.id = :userId')
    ->setParameter('userId', $userId)
;

var_dump(
    $queryBuilder->getQuery()->getArrayResult()
);

如您所见，索引依据选项可用作查询生成器方法的第三个参数：from

/**
 * Creates and adds a query root corresponding to the entity identified by the given alias,
 * forming a cartesian product with any existing query roots.
 *
 * <code>
 *     $qb = $em->createQueryBuilder()
 *         ->select('u')
 *         ->from('User', 'u')
 * </code>
 *
 * @param string $from    The class name.
 * @param string $alias   The alias of the class.
 * @param string $indexBy The index for the from.
 *
 * @return QueryBuilder This QueryBuilder instance.
 */
public function from($from, $alias, $indexBy = null)
{
    return $this->add('from', new Expr\From($from, $alias, $indexBy), true);
}