vendor/doctrine/mongodb-odm-bundle/Repository/ContainerRepositoryFactory.php line 50

Open in your IDE?
  1. <?php
  3. declare(strict_types=1);
  5. namespace Doctrine\Bundle\MongoDBBundle\Repository;
  7. use Doctrine\Bundle\MongoDBBundle\DependencyInjection\Compiler\ServiceRepositoryCompilerPass;
  8. use Doctrine\ODM\MongoDB\DocumentManager;
  9. use Doctrine\ODM\MongoDB\Mapping\ClassMetadata;
  10. use Doctrine\ODM\MongoDB\Repository\DocumentRepository;
  11. use Doctrine\ODM\MongoDB\Repository\GridFSRepository;
  12. use Doctrine\ODM\MongoDB\Repository\RepositoryFactory;
  13. use Doctrine\Persistence\ObjectRepository;
  14. use Psr\Container\ContainerInterface;
  15. use RuntimeException;
  17. use function class_exists;
  18. use function is_a;
  19. use function spl_object_hash;
  20. use function sprintf;
  22. /**
  23.  * Fetches repositories from the container or falls back to normal creation.
  24.  */
  25. final class ContainerRepositoryFactory implements RepositoryFactory
  26. {
  27.     /** @var array<string, ObjectRepository> */
  28.     private $managedRepositories = [];
  30.     /** @var ContainerInterface|null */
  31.     private $container;
  33.     /**
  34.      * @param ContainerInterface $container A service locator containing the repositories
  35.      */
  36.     public function __construct(ContainerInterface $container)
  37.     {
  38.         $this->container = $container;
  39.     }
  41.     /**
  42.      * @psalm-param class-string<T> $documentName
  43.      *
  44.      * @psalm-return ObjectRepository<T>
  45.      *
  46.      * @template T of object
  47.      */
  48.     public function getRepository(DocumentManager $documentManager, string $documentName): ObjectRepository
  49.     {
  50.         $metadata             = $documentManager->getClassMetadata($documentName);
  51.         $customRepositoryName = $metadata->customRepositoryClassName;
  53.         if ($customRepositoryName !== null) {
  54.             // fetch from the container
  55.             if ($this->container && $this->container->has($customRepositoryName)) {
  56.                 /** @var ObjectRepository<T> $repository */
  57.                 $repository = $this->container->get($customRepositoryName);
  59.                 if (! $repository instanceof DocumentRepository) {
  60.                     throw new RuntimeException(sprintf('The service "%s" must extend DocumentRepository (or a base class, like ServiceDocumentRepository).', $customRepositoryName));
  61.                 }
  63.                 return $repository;
  64.             }
  66.             // if not in the container but the class/id implements the interface, throw an error
  67.             if (is_a($customRepositoryName, ServiceDocumentRepositoryInterface::class, true)) {
  68.                 throw new RuntimeException(sprintf('The "%s" document repository implements "%s", but its service could not be found. Make sure the service exists and is tagged with "%s".', $customRepositoryName, ServiceDocumentRepositoryInterface::class, ServiceRepositoryCompilerPass::REPOSITORY_SERVICE_TAG));
  69.             }
  71.             if (! class_exists($customRepositoryName)) {
  72.                 throw new RuntimeException(sprintf('The "%s" document 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));
  73.             }
  75.             // allow the repository to be created below
  76.         }
  78.         return $this->getOrCreateRepository($documentManager, $metadata);
  79.     }
  81.     /**
  82.      * @psalm-param ClassMetadata<T> $metadata
  83.      *
  84.      * @psalm-return ObjectRepository<T>
  85.      *
  86.      * @template T of object
  87.      */
  88.     private function getOrCreateRepository(DocumentManager $documentManager, ClassMetadata $metadata): ObjectRepository
  89.     {
  90.         $repositoryHash = $metadata->getName() . spl_object_hash($documentManager);
  91.         if (isset($this->managedRepositories[$repositoryHash])) {
  92.             /** @psalm-var ObjectRepository<T> */
  93.             return $this->managedRepositories[$repositoryHash];
  94.         }
  96.         if ($metadata->customRepositoryClassName) {
  97.             $repositoryClassName = $metadata->customRepositoryClassName;
  98.         } elseif ($metadata->isFile) {
  99.             /** @psalm-var class-string<GridFSRepository<T>> $repositoryClassName */
  100.             $repositoryClassName = $documentManager->getConfiguration()->getDefaultGridFSRepositoryClassName();
  101.         } else {
  102.             /** @psalm-var class-string<ObjectRepository<T>> $repositoryClassName */
  103.             $repositoryClassName = $documentManager->getConfiguration()->getDefaultDocumentRepositoryClassName();
  104.         }
  106.         return $this->managedRepositories[$repositoryHash] = new $repositoryClassName($documentManager, $documentManager->getUnitOfWork(), $metadata);
  107.     }
  108. }