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

Open in your IDE?
  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 rtrim;
  12. use function spl_object_hash;
  13. use function strpos;
  15. /**
  16. * The DriverChain allows you to add multiple other mapping drivers for
  17. * certain namespaces.
  18. */
  19. class MappingDriverChain implements MappingDriver
  20. {
  21. /**
  22. * The default driver.
  23. *
  24. * @var MappingDriver|null
  25. */
  26. private $defaultDriver;
  28. /** @var array<string, MappingDriver> */
  29. private $drivers = [];
  31. /**
  32. * Gets the default driver.
  33. *
  34. * @return MappingDriver|null
  35. */
  36. public function getDefaultDriver()
  37. {
  38. return $this->defaultDriver;
  39. }
  41. /**
  42. * Set the default driver.
  43. *
  44. * @return void
  45. */
  46. public function setDefaultDriver(MappingDriver $driver)
  47. {
  48. $this->defaultDriver = $driver;
  49. }
  51. /**
  52. * Adds a nested driver.
  53. *
  54. * @return void
  55. */
  56. public function addDriver(MappingDriver $nestedDriver, string $namespace)
  57. {
  58. $this->drivers[$namespace] = $nestedDriver;
  59. }
  61. /**
  62. * Gets the array of nested drivers.
  63. *
  64. * @return array<string, MappingDriver> $drivers
  65. */
  66. public function getDrivers()
  67. {
  68. return $this->drivers;
  69. }
  71. /**
  72. * {@inheritDoc}
  73. */
  74. public function loadMetadataForClass(string $className, ClassMetadata $metadata)
  75. {
  76. foreach ($this->drivers as $namespace => $driver) {
  77. if ($this->isInNamespace($className, $namespace)) {
  78. $driver->loadMetadataForClass($className, $metadata);
  80. return;
  81. }
  82. }
  84. if ($this->defaultDriver !== null) {
  85. $this->defaultDriver->loadMetadataForClass($className, $metadata);
  87. return;
  88. }
  90. throw MappingException::classNotFoundInNamespaces($className, array_keys($this->drivers));
  91. }
  93. /**
  94. * {@inheritDoc}
  95. */
  96. public function getAllClassNames()
  97. {
  98. $classNames = [];
  99. $driverClasses = [];
  101. foreach ($this->drivers as $namespace => $driver) {
  102. $oid = spl_object_hash($driver);
  104. if (! isset($driverClasses[$oid])) {
  105. $driverClasses[$oid] = $driver->getAllClassNames();
  106. }
  108. foreach ($driverClasses[$oid] as $className) {
  109. if (! $this->isInNamespace($className, $namespace)) {
  110. continue;
  111. }
  113. $classNames[$className] = true;
  114. }
  115. }
  117. if ($this->defaultDriver !== null) {
  118. foreach ($this->defaultDriver->getAllClassNames() as $className) {
  119. $classNames[$className] = true;
  120. }
  121. }
  123. return array_keys($classNames);
  124. }
  126. /**
  127. * {@inheritDoc}
  128. */
  129. public function isTransient(string $className)
  130. {
  131. foreach ($this->drivers as $namespace => $driver) {
  132. if ($this->isInNamespace($className, $namespace)) {
  133. return $driver->isTransient($className);
  134. }
  135. }
  137. if ($this->defaultDriver !== null) {
  138. return $this->defaultDriver->isTransient($className);
  139. }
  141. return true;
  142. }
  144. private function isInNamespace(string $className, string $namespace): bool
  145. {
  146. $namespace = rtrim($namespace, '\\') . '\\';
  148. return strpos($className, $namespace) === 0;
  149. }
  150. }