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

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 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. }