vendor/doctrine/doctrine-bundle/src/Repository/ServiceEntityRepositoryProxy.php line 72

  1. <?php
  3. declare(strict_types=1);
  5. namespace Doctrine\Bundle\DoctrineBundle\Repository;
  7. use Doctrine\Common\Collections\AbstractLazyCollection;
  8. use Doctrine\Common\Collections\Criteria;
  9. use Doctrine\Common\Collections\Selectable;
  10. use Doctrine\DBAL\LockMode;
  11. use Doctrine\ORM\EntityManagerInterface;
  12. use Doctrine\ORM\EntityRepository;
  13. use Doctrine\ORM\Mapping\ClassMetadata;
  14. use Doctrine\ORM\Query\ResultSetMappingBuilder;
  15. use Doctrine\ORM\QueryBuilder;
  16. use Doctrine\Persistence\ManagerRegistry;
  17. use LogicException;
  18. use Symfony\Component\VarExporter\LazyObjectInterface;
  20. use function sprintf;
  22. /**
  23.  * @internal Extend {@see ServiceEntityRepository} instead.
  24.  *
  25.  * @template T of object
  26.  * @template-extends EntityRepository<T>
  27.  */
  28. class ServiceEntityRepositoryProxy extends EntityRepository implements ServiceEntityRepositoryInterface
  29. {
  30.     private ?EntityRepository $repository = null;
  32.     /** @param class-string<T> $entityClass The class name of the entity this repository manages */
  33.     public function __construct(
  34.         private readonly ManagerRegistry $registry,
  35.         private readonly string $entityClass,
  36.     ) {
  37.         if (! $this instanceof LazyObjectInterface) {
  38.             return;
  39.         }
  41.         $this->repository = $this->resolveRepository();
  42.     }
  44.     public function createQueryBuilder(string $alias, ?string $indexBy = null): QueryBuilder
  45.     {
  46.         return ($this->repository ??= $this->resolveRepository())
  47.             ->createQueryBuilder($alias, $indexBy);
  48.     }
  50.     public function createResultSetMappingBuilder(string $alias): ResultSetMappingBuilder
  51.     {
  52.         return ($this->repository ??= $this->resolveRepository())
  53.             ->createResultSetMappingBuilder($alias);
  54.     }
  56.     public function find(mixed $id, LockMode|int|null $lockMode = null, int|null $lockVersion = null): object|null
  57.     {
  58.         /** @psalm-suppress InvalidReturnStatement This proxy is used only in combination with newer parent class */
  59.         return ($this->repository ??= $this->resolveRepository())
  60.             ->find($id, $lockMode, $lockVersion);
  61.     }
  63.     /**
  64.      * {@inheritDoc}
  65.      *
  66.      * @psalm-suppress InvalidReturnStatement This proxy is used only in combination with newer parent class
  67.      * @psalm-suppress InvalidReturnType This proxy is used only in combination with newer parent class
  68.      */
  69.     public function findBy(array $criteria, ?array $orderBy = null, ?int $limit = null, ?int $offset = null): array
  70.     {
  71.         return ($this->repository ??= $this->resolveRepository())
  72.             ->findBy($criteria, $orderBy, $limit, $offset);
  73.     }
  75.     /** {@inheritDoc} */
  76.     public function findOneBy(array $criteria, ?array $orderBy = null): object|null
  77.     {
  78.         /** @psalm-suppress InvalidReturnStatement This proxy is used only in combination with newer parent class */
  79.         return ($this->repository ??= $this->resolveRepository())
  80.             ->findOneBy($criteria, $orderBy);
  81.     }
  83.     /** {@inheritDoc} */
  84.     public function count(array $criteria = []): int
  85.     {
  86.         return ($this->repository ??= $this->resolveRepository())->count($criteria);
  87.     }
  89.     /**
  90.      * {@inheritDoc}
  91.      */
  92.     public function __call(string $method, array $arguments): mixed
  93.     {
  94.         return ($this->repository ??= $this->resolveRepository())->$method(...$arguments);
  95.     }
  97.     protected function getEntityName(): string
  98.     {
  99.         return ($this->repository ??= $this->resolveRepository())->getEntityName();
  100.     }
  102.     protected function getEntityManager(): EntityManagerInterface
  103.     {
  104.         return ($this->repository ??= $this->resolveRepository())->getEntityManager();
  105.     }
  107.     /** @psalm-suppress InvalidReturnType This proxy is used only in combination with newer parent class */
  108.     protected function getClassMetadata(): ClassMetadata
  109.     {
  110.         /** @psalm-suppress InvalidReturnStatement This proxy is used only in combination with newer parent class */
  111.         return ($this->repository ??= $this->resolveRepository())->getClassMetadata();
  112.     }
  114.     public function matching(Criteria $criteria): AbstractLazyCollection&Selectable
  115.     {
  116.         return ($this->repository ??= $this->resolveRepository())->matching($criteria);
  117.     }
  119.     private function resolveRepository(): EntityRepository
  120.     {
  121.         $manager = $this->registry->getManagerForClass($this->entityClass);
  123.         if ($manager === null) {
  124.             throw new LogicException(sprintf(
  125.                 'Could not find the entity manager for class "%s". Check your Doctrine configuration to make sure it is configured to load this entity’s metadata.',
  126.                 $this->entityClass,
  127.             ));
  128.         }
  130.         return new EntityRepository($manager, $manager->getClassMetadata($this->entityClass));
  131.     }
  132. }