Shlinkio\Shlink\Core\Util\DoctrineBatchHelper::wrapIterable()

Complexity

Conditions	3
Paths	3

Size

Total Lines	20
Code Lines	12

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

<?php
    public function wrapIterable(iterable $resultSet, int $batchSize): iterable
    {
        $iteration = 0;

        $this->em->beginTransaction();

        try {
            foreach ($resultSet as $key => $value) {
                $iteration++;
                yield $key => $value;
                $this->flushAndClearBatch($iteration, $batchSize);
            }
        } catch (Throwable $e) {
            $this->em->rollback();

            throw $e;
        }

        $this->flushAndClearEntityManager();
        $this->em->commit();
    }