Zenstruck\Porpaginas\Doctrine\Batch\ORMIterableResultDecorator

Complexity

Total Complexity

9

Size/Duplication

Total Lines	38
Duplicated Lines	0 %

Test Coverage

Coverage

93.75%

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	3	1
A	getIterator()	0	4	2
A	normalizeResult()	0	17	6

<?php

namespace Zenstruck\Porpaginas\Doctrine\Batch;

use Doctrine\ORM\Internal\Hydration\IterableResult;

/**
 * Fixes https://github.com/doctrine/orm/issues/2821.
 *
 * @internal
 *
 * @author Kevin Bond <kevinbond@gmail.com>
 */
final class ORMIterableResultDecorator implements \IteratorAggregate
{
    private IterableResult $result;

    public function __construct(IterableResult $result)
    {
        $this->result = $result;
    }

    public function getIterator(): \Traversable
    {
        foreach ($this->result as $key => $value) {
            yield $key => self::normalizeResult($value);
        }
    }

    /**
     * @param mixed $result
     *
     * @return mixed
     */
    private static function normalizeResult($result)
    {
        if (!\is_array($result)) {
            return $result;
        }

        $firstKey = \array_key_first($result);

        if (null !== $firstKey && \is_object($result[$firstKey]) && $result === [$firstKey => $result[$firstKey]]) {
            return $result[$firstKey];
        }

        if (\count($result) > 1) {
            $result = [\array_merge(...$result)];
        }

        return $result[$firstKey];
    }
}