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

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 array $managedRepositories = [];
  30.     private ContainerInterface $container;
  32.     /** @param ContainerInterface $container A service locator containing the repositories */
  33.     public function __construct(ContainerInterface $container)
  34.     {
  35.         $this->container = $container;
  36.     }
  38.     /**
  39.      * @psalm-param class-string<T> $documentName
  40.      *
  41.      * @psalm-return ObjectRepository<T>
  42.      *
  43.      * @template T of object
  44.      */
  45.     public function getRepository(DocumentManager $documentManager, string $documentName): ObjectRepository
  46.     {
  47.         $metadata             = $documentManager->getClassMetadata($documentName);
  48.         $customRepositoryName = $metadata->customRepositoryClassName;
  50.         if ($customRepositoryName !== null) {
  51.             // fetch from the container
  52.             if ($this->container->has($customRepositoryName)) {
  53.                 /** @var ObjectRepository<T> $repository */
  54.                 $repository = $this->container->get($customRepositoryName);
  56.                 if (! $repository instanceof DocumentRepository) {
  57.                     throw new RuntimeException(sprintf('The service "%s" must extend DocumentRepository (or a base class, like ServiceDocumentRepository).', $customRepositoryName));
  58.                 }
  60.                 return $repository;
  61.             }
  63.             // if not in the container but the class/id implements the interface, throw an error
  64.             if (is_a($customRepositoryName, ServiceDocumentRepositoryInterface::class, true)) {
  65.                 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));
  66.             }
  68.             if (! class_exists($customRepositoryName)) {
  69.                 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));
  70.             }
  72.             // allow the repository to be created below
  73.         }
  75.         return $this->getOrCreateRepository($documentManager, $metadata);
  76.     }
  78.     /**
  79.      * @psalm-param ClassMetadata<T> $metadata
  80.      *
  81.      * @psalm-return ObjectRepository<T>
  82.      *
  83.      * @template T of object
  84.      */
  85.     private function getOrCreateRepository(DocumentManager $documentManager, ClassMetadata $metadata): ObjectRepository
  86.     {
  87.         $repositoryHash = $metadata->getName() . spl_object_hash($documentManager);
  88.         if (isset($this->managedRepositories[$repositoryHash])) {
  89.             /** @psalm-var ObjectRepository<T> */
  90.             return $this->managedRepositories[$repositoryHash];
  91.         }
  93.         if ($metadata->customRepositoryClassName) {
  94.             $repositoryClassName = $metadata->customRepositoryClassName;
  95.         } elseif ($metadata->isFile) {
  96.             /** @psalm-var class-string<GridFSRepository<T>> $repositoryClassName */
  97.             $repositoryClassName = $documentManager->getConfiguration()->getDefaultGridFSRepositoryClassName();
  98.         } else {
  99.             /** @psalm-var class-string<ObjectRepository<T>> $repositoryClassName */
  100.             $repositoryClassName = $documentManager->getConfiguration()->getDefaultDocumentRepositoryClassName();
  101.         }
  103.         return $this->managedRepositories[$repositoryHash] = new $repositoryClassName($documentManager, $documentManager->getUnitOfWork(), $metadata);
  104.     }
  105. }