Zenstruck\Porpaginas\Doctrine\Batch\ORMCountableBatchProcessor

Complexity

Total Complexity

4

Size/Duplication

Total Lines	26
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	8	2
A	count()	0	3	1
A	getIterator()	0	3	1

<?php

namespace Zenstruck\Porpaginas\Doctrine\Batch;

use Doctrine\ORM\EntityManagerInterface;

/**
 * @author Kevin Bond <kevinbond@gmail.com>
 */
final class ORMCountableBatchProcessor implements \IteratorAggregate, \Countable
{
    private ORMBatchProcessor $batchProcessor;
    private iterable $countable;

    /**
     * @param iterable $items Must be countable
     */
    public function __construct(iterable $items, EntityManagerInterface $em, int $chunkSize = 100)
    {
        if (!\is_countable($items)) {
            throw new \InvalidArgumentException('$items must be countable.');
        }

        $this->batchProcessor = new ORMBatchProcessor($items, $em, $chunkSize);
        $this->countable = &$items;
    }

    public function getIterator(): \Traversable
    {
        yield from $this->batchProcessor;
    }

    public function count(): int
    {
        return \count($this->countable);
    }
}