Xsolve\Associate\AssociationCollecting\DoctrineOrmAssociationCollectingStrategy

Complexity

Total Complexity

5

Size/Duplication

Total Lines	59
Duplicated Lines	0 %

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	collect()	0	5	1
A	loadObjectsIfPossible()	0	17	3
A	__construct()	0	8	1

<?php

namespace Xsolve\Associate\AssociationCollecting;

use Xsolve\Associate\Loader\DoctrineOrmEntityLoader;
use Xsolve\Associate\Metadata\AssociationMetadataWrapper;
use Xsolve\Associate\Metadata\ClassMetadataWrapper;
use Xsolve\Associate\Metadata\MetadataWrapperProvider;

class DoctrineOrmAssociationCollectingStrategy extends BasicAssociationCollectingStrategy
{
    /**
     * @var MetadataWrapperProvider
     */
    protected $metadataWrapperProvider;

    /**
     * @var DoctrineOrmEntityLoader
     */
    protected $doctrineOrmEntityLoader;

    /**
     * @param MetadataWrapperProvider $metadataWrapperProvider
     * @param DoctrineOrmEntityLoader $doctrineOrmEntityLoader
     */
    public function __construct(
        MetadataWrapperProvider $metadataWrapperProvider,
        DoctrineOrmEntityLoader $doctrineOrmEntityLoader
    ) {
        parent::__construct();

        $this->metadataWrapperProvider = $metadataWrapperProvider;
        $this->doctrineOrmEntityLoader = $doctrineOrmEntityLoader;
    }

    /**
     * @param array  $objects
     * @param string $associationName
     *
     * @return array
     */
    public function collect(array $objects, string $associationName): array
    {
        $this->loadObjectsIfPossible($objects, $associationName);

        return parent::collect($objects, $associationName);
    }

    /**
     * {@inheritdoc}
     */
    protected function loadObjectsIfPossible(array $objects, string $associationName)
    {
        $classMetadataWrapper = $this->metadataWrapperProvider->getClassMetadataWrapperByObjects($objects);
        if (!$classMetadataWrapper instanceof ClassMetadataWrapper) {
            return;
        }

        $this->doctrineOrmEntityLoader->loadUninitializedProxies($objects, $classMetadataWrapper);

        $associationMetadataWrapper = $classMetadataWrapper->getAssociationMetadataWrapper($associationName);
        if (!$associationMetadataWrapper instanceof AssociationMetadataWrapper) {
            return;
        }

        $this->doctrineOrmEntityLoader->loadAssociatedUninitializedCollectionsAndProxies(
            $objects,
            $associationMetadataWrapper
        );
    }
}