Doctrine ColumnHydrator

ColumnHydrator.php

<?php

declare(strict_types=1);

namespace App\ORM\Hydration;

use Doctrine\ORM\Internal\Hydration\ArrayHydrator;

/**
 * Returns one-dimensional scalar array from query: mixed[][] => mixed[]
 *
 * Example:
 * ArrayHydrator: [['id' => 1], ['id' => 2]]
 * ColumnHydrator: [1, 2]
 *
 * @see https://stackoverflow.com/questions/11657835/how-to-get-a-one-dimensional-scalar-array-as-a-doctrine-dql-query-result
 * @author Gabriel Ostrolucký
 * @license MIT
 */
class ColumnHydrator extends ArrayHydrator
{
    /**
     * @return mixed[]
     */
    protected function hydrateAllData(): array
    {
        if (!isset($this->_rsm->indexByMap['scalars'])) {
            return $this->_stmt->fetchAll(\PDO::FETCH_COLUMN);
        }

        if (!$result = parent::hydrateAllData()) {
            return $result;
        }

        $indexColumn = $this->_rsm->scalarMappings[$this->_rsm->indexByMap['scalars']];
        $keys = array_keys(reset($result));

        return array_column($result, isset($keys[1]) && $keys[0] === $indexColumn ? $keys[1] : $keys[0], $indexColumn);
    }
}

As a formality: can you specify a license (I want to use it in a closed source project).

Thanks!

For anyone wondering how to use it - you have to create entry in Doctrine config:

doctrine:
    orm:
        hydrators:
            ColumnHydrator: \App\ORM\Hydration\ColumnHydrator

Later on you can just use it as $query->getResult('ColumnHydrator').

@kiler129 thanks, added the license

Thanks for this! I have two questions:

1. fetchAll() is deprecated, and I don't know of an alternative, see doctrine/orm#8322
2. What is if (!$result = parent::hydrateAllData()) checking for - ArrayHydrator::hydrateAllData() is always returning an array.

2. It's checking if result is empty

Is there a way to create a PR for this gist? If no, I'd suggest to change it to (to make phpstan happy :-)

$result = parent::hydrateAllData();
if ([] === $result) {
    return [];
}

You should use this instead doctrine/orm#8919