Malef\Associate\DoctrineOrm\Loader\UninitializedProxiesLoader

Complexity

Total Complexity

7

Size/Duplication

Total Lines	39
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	3	1
A	load()	0	27	6

<?php

namespace Malef\Associate\DoctrineOrm\Loader;

use Doctrine\Common\Collections\ArrayCollection;
use Doctrine\ORM\Proxy\Proxy;
use Malef\Associate\DoctrineOrm\Loader\ChunkingStrategy\ChunkingStrategy;
use Malef\Associate\DoctrineOrm\Metadata\ClassMetadataAdapter;

class UninitializedProxiesLoader
{
    /**
     * @var ChunkingStrategy
     */
    protected $chunkingStrategy;

    public function __construct(ChunkingStrategy $chunkingStrategy)
    {
        $this->chunkingStrategy = $chunkingStrategy;
    }

    public function load(
        ArrayCollection $entities,
        ClassMetadataAdapter $classMetadataAdapter
    ) {
        $uninitializedProxies = new ArrayCollection();
        foreach ($entities->getValues() as $entity) {
            if ($entity instanceof Proxy && !$entity->__isInitialized()) {
                $uninitializedProxies->add($entity);
            }
        }

        if ($uninitializedProxies->isEmpty()) {
            return;
        }

        foreach ($this->chunkingStrategy->chunk($uninitializedProxies) as $uninitializedProxiesChunk) {
            $queryBuilder = $classMetadataAdapter->createQueryBuilder('e');

            $queryBuilder
                ->andWhere(
                    $queryBuilder->expr()->in(
                        'e',
                        $classMetadataAdapter->getIdentifierValueForMany($uninitializedProxiesChunk)->getValues()
                    )
                )
                ->getQuery()
                ->execute();
        }
    }
}