vendor/doctrine/doctrine-bundle/Repository/ContainerRepositoryFactory.php line 42

Open in your IDE?
  1. <?php
  3. namespace Doctrine\Bundle\DoctrineBundle\Repository;
  5. use Doctrine\Bundle\DoctrineBundle\DependencyInjection\Compiler\ServiceRepositoryCompilerPass;
  6. use Doctrine\ORM\EntityManagerInterface;
  7. use Doctrine\ORM\Mapping\ClassMetadata;
  8. use Doctrine\ORM\Repository\RepositoryFactory;
  9. use Doctrine\Persistence\ObjectRepository;
  10. use InvalidArgumentException;
  11. use Psr\Container\ContainerInterface;
  12. use RuntimeException;
  14. /**
  15.  * Fetches repositories from the container or falls back to normal creation.
  16.  */
  17. final class ContainerRepositoryFactory implements RepositoryFactory
  18. {
  19.     /** @var ObjectRepository[] */
  20.     private $managedRepositories = [];
  22.     /** @var ContainerInterface|null */
  23.     private $container;
  25.     /**
  26.      * @param ContainerInterface $container A service locator containing the repositories
  27.      */
  28.     public function __construct(ContainerInterface $container = null)
  29.     {
  30.         // When DoctrineBundle requires Symfony 3.3+, this can be removed
  31.         // and the $container argument can become required.
  32.         if ($container === null) {
  33.             throw new InvalidArgumentException(sprintf('The first argument of %s::__construct() is required on Symfony 3.3 or higher.', self::class));
  34.         }
  36.         $this->container = $container;
  37.     }
  39.     /**
  40.      * {@inheritdoc}
  41.      */
  42.     public function getRepository(EntityManagerInterface $entityManager, $entityName)
  43.     {
  44.         $metadata            = $entityManager->getClassMetadata($entityName);
  45.         $repositoryServiceId = $metadata->customRepositoryClassName;
  47.         $customRepositoryName = $metadata->customRepositoryClassName;
  48.         if ($customRepositoryName !== null) {
  49.             // fetch from the container
  50.             if ($this->container && $this->container->has($customRepositoryName)) {
  51.                 $repository = $this->container->get($customRepositoryName);
  53.                 if (! $repository instanceof ObjectRepository) {
  54.                     throw new RuntimeException(sprintf('The service "%s" must implement ObjectRepository (or extend a base class, like ServiceEntityRepository).', $repositoryServiceId));
  55.                 }
  57.                 return $repository;
  58.             }
  60.             // if not in the container but the class/id implements the interface, throw an error
  61.             if (is_a($customRepositoryName, ServiceEntityRepositoryInterface::class, true)) {
  62.                 throw new RuntimeException(sprintf('The "%s" entity repository implements "%s", but its service could not be found. Make sure the service exists and is tagged with "%s".', $customRepositoryName, ServiceEntityRepositoryInterface::class, ServiceRepositoryCompilerPass::REPOSITORY_SERVICE_TAG));
  63.             }
  65.             if (! class_exists($customRepositoryName)) {
  66.                 throw new RuntimeException(sprintf('The "%s" entity has a repositoryClass set to "%s", but this is not a valid class. Check your class naming. If this is meant to be a service id, make sure this service exists and is tagged with "%s".', $metadata->name, $customRepositoryName, ServiceRepositoryCompilerPass::REPOSITORY_SERVICE_TAG));
  67.             }
  69.             // allow the repository to be created below
  70.         }
  72.         return $this->getOrCreateRepository($entityManager, $metadata);
  73.     }
  75.     private function getOrCreateRepository(EntityManagerInterface $entityManager, ClassMetadata $metadata)
  76.     {
  77.         $repositoryHash = $metadata->getName() . spl_object_hash($entityManager);
  78.         if (isset($this->managedRepositories[$repositoryHash])) {
  79.             return $this->managedRepositories[$repositoryHash];
  80.         }
  82.         $repositoryClassName = $metadata->customRepositoryClassName ?: $entityManager->getConfiguration()->getDefaultRepositoryClassName();
  84.         return $this->managedRepositories[$repositoryHash] = new $repositoryClassName($entityManager, $metadata);
  85.     }
  86. }