EdmondsCommerce\DoctrineStaticMeta\Entity\Savers\BulkEntitySaver

Complexity

Total Complexity

19

Size/Duplication

Total Lines	115
Duplicated Lines	0 %

Test Coverage

Coverage

0%

Importance

Changes

0

10 Methods

Rating	Name	Duplication	Size	Complexity
A	freeResources()	0	9	2
A	startBulkProcess()	0	7	1
A	endBulkProcess()	0	11	3
A	setChunkSize()	0	5	1
A	bulkSaveIfChunkBigEnough()	0	7	2
A	addEntityToSave()	0	7	2
A	getChunkSize()	0	3	1
A	doSave()	0	8	2
A	__construct()	0	4	1
A	__destruct()	0	7	4

<?php declare(strict_types=1);

namespace EdmondsCommerce\DoctrineStaticMeta\Entity\Savers;

use Doctrine\ORM\EntityManagerInterface;
use EdmondsCommerce\DoctrineStaticMeta\Entity\Interfaces\EntityInterface;

class BulkEntitySaver
{
    /**
     * @var EntityManagerInterface
     */
    protected $entityManager;

    private $entitiesToSave = [];

    private $chunkSize = 1000;

    /**
     * @var bool
     */
    private $gcWasEnabled;

    private $started = false;
    private $ended   = false;

    public function __construct(EntityManagerInterface $entityManager)
    {
        $this->entityManager = $entityManager;
        $this->gcWasEnabled  = gc_enabled();
    }

    public function __destruct()
    {
        if (true === $this->started && false === $this->ended) {
            if (!$this->entityManager->isOpen()) {
                throw new \RuntimeException('Error in ' . __METHOD__ . ': Entity Manager has been closed');
            }
            $this->endBulkProcess();
        }
    }

    public function endBulkProcess(): void
    {
        $this->started = false;
        $this->ended   = true;
        if ([] !== $this->entitiesToSave) {
            $this->doSave();
        }
        if (false === $this->gcWasEnabled) {
            return;
        }
        gc_enable();
    }

    private function doSave()
    {
        foreach ($this->entitiesToSave as $entity) {
            $this->entityManager->persist($entity);
        }

        $this->entityManager->flush();
        $this->freeResources();
    }

    private function freeResources()
    {
        gc_enable();
        foreach ($this->entitiesToSave as $entity) {
            $this->entityManager->detach($entity);
        }
        $this->entitiesToSave = [];
        gc_collect_cycles();
        gc_disable();
    }

    public function addEntityToSave(EntityInterface $entity)
    {
        if (false === $this->started) {
            $this->startBulkProcess();
        }
        $this->entitiesToSave[] = $entity;
        $this->bulkSaveIfChunkBigEnough();
    }

    public function startBulkProcess(): self
    {
        gc_disable();
        $this->started = true;
        $this->ended   = false;

        return $this;
    }

    private function bulkSaveIfChunkBigEnough()
    {
        end($this->entitiesToSave);
        $key  = key($this->entitiesToSave);
        $size = $key + 1;
        if (($size % $this->chunkSize) === 0) {
            $this->doSave();
        }
    }

    /**
     * @return int
     */
    public function getChunkSize(): int
    {
        return $this->chunkSize;
    }

    /**
     * @param int $chunkSize
     *
     * @return BulkEntitySaver
     */
    public function setChunkSize(int $chunkSize): BulkEntitySaver
    {
        $this->chunkSize = $chunkSize;

        return $this;
    }
}