vendor/doctrine/persistence/src/Persistence/Mapping/Driver/MappingDriverChain.php line 73

  1. <?php
  3. declare(strict_types=1);
  5. namespace Doctrine\Persistence\Mapping\Driver;
  7. use Doctrine\Persistence\Mapping\ClassMetadata;
  8. use Doctrine\Persistence\Mapping\MappingException;
  10. use function array_keys;
  11. use function spl_object_hash;
  12. use function strpos;
  14. /**
  15.  * The DriverChain allows you to add multiple other mapping drivers for
  16.  * certain namespaces.
  17.  */
  18. class MappingDriverChain implements MappingDriver
  19. {
  20.     /**
  21.      * The default driver.
  22.      *
  23.      * @var MappingDriver|null
  24.      */
  25.     private $defaultDriver;
  27.     /** @var array<string, MappingDriver> */
  28.     private $drivers = [];
  30.     /**
  31.      * Gets the default driver.
  32.      *
  33.      * @return MappingDriver|null
  34.      */
  35.     public function getDefaultDriver()
  36.     {
  37.         return $this->defaultDriver;
  38.     }
  40.     /**
  41.      * Set the default driver.
  42.      *
  43.      * @return void
  44.      */
  45.     public function setDefaultDriver(MappingDriver $driver)
  46.     {
  47.         $this->defaultDriver = $driver;
  48.     }
  50.     /**
  51.      * Adds a nested driver.
  52.      *
  53.      * @return void
  54.      */
  55.     public function addDriver(MappingDriver $nestedDriver, string $namespace)
  56.     {
  57.         $this->drivers[$namespace] = $nestedDriver;
  58.     }
  60.     /**
  61.      * Gets the array of nested drivers.
  62.      *
  63.      * @return array<string, MappingDriver> $drivers
  64.      */
  65.     public function getDrivers()
  66.     {
  67.         return $this->drivers;
  68.     }
  70.     /**
  71.      * {@inheritDoc}
  72.      */
  73.     public function loadMetadataForClass(string $className, ClassMetadata $metadata)
  74.     {
  75.         foreach ($this->drivers as $namespace => $driver) {
  76.             if (strpos($className, $namespace) === 0) {
  77.                 $driver->loadMetadataForClass($className, $metadata);
  79.                 return;
  80.             }
  81.         }
  83.         if ($this->defaultDriver !== null) {
  84.             $this->defaultDriver->loadMetadataForClass($className, $metadata);
  86.             return;
  87.         }
  89.         throw MappingException::classNotFoundInNamespaces($className, array_keys($this->drivers));
  90.     }
  92.     /**
  93.      * {@inheritDoc}
  94.      */
  95.     public function getAllClassNames()
  96.     {
  97.         $classNames    = [];
  98.         $driverClasses = [];
  100.         foreach ($this->drivers as $namespace => $driver) {
  101.             $oid = spl_object_hash($driver);
  103.             if (! isset($driverClasses[$oid])) {
  104.                 $driverClasses[$oid] = $driver->getAllClassNames();
  105.             }
  107.             foreach ($driverClasses[$oid] as $className) {
  108.                 if (strpos($className, $namespace) !== 0) {
  109.                     continue;
  110.                 }
  112.                 $classNames[$className] = true;
  113.             }
  114.         }
  116.         if ($this->defaultDriver !== null) {
  117.             foreach ($this->defaultDriver->getAllClassNames() as $className) {
  118.                 $classNames[$className] = true;
  119.             }
  120.         }
  122.         return array_keys($classNames);
  123.     }
  125.     /**
  126.      * {@inheritDoc}
  127.      */
  128.     public function isTransient(string $className)
  129.     {
  130.         foreach ($this->drivers as $namespace => $driver) {
  131.             if (strpos($className, $namespace) === 0) {
  132.                 return $driver->isTransient($className);
  133.             }
  134.         }
  136.         if ($this->defaultDriver !== null) {
  137.             return $this->defaultDriver->isTransient($className);
  138.         }
  140.         return true;
  141.     }
  142. }