vendor/theofidry/alice-data-fixtures/src/Bridge/Doctrine/Purger/Purger.php line 47

Open in your IDE?
  1. <?php
  3. /*
  4. * This file is part of the Fidry\AliceDataFixtures package.
  5. *
  6. * (c) Théo FIDRY <theo.fidry@gmail.com>
  7. *
  8. * For the full copyright and license information, please view the LICENSE
  9. * file that was distributed with this source code.
  10. */
  12. declare(strict_types=1);
  14. namespace Fidry\AliceDataFixtures\Bridge\Doctrine\Purger;
  16. use Doctrine\Common\DataFixtures\Purger\MongoDBPurger as DoctrineMongoDBPurger;
  17. use Doctrine\Common\DataFixtures\Purger\ORMPurger as DoctrineOrmPurger;
  18. use Doctrine\Common\DataFixtures\Purger\PHPCRPurger as DoctrinePhpCrPurger;
  19. use Doctrine\Common\DataFixtures\Purger\PurgerInterface as DoctrinePurgerInterface;
  20. use Doctrine\DBAL\Platforms\MariaDBPlatform;
  21. use Doctrine\DBAL\Platforms\MySqlPlatform;
  22. use Doctrine\ODM\MongoDB\DocumentManager as DoctrineMongoDocumentManager;
  23. use Doctrine\ODM\PHPCR\DocumentManager as DoctrinePhpCrDocumentManager;
  24. use Doctrine\ORM\EntityManagerInterface;
  25. use Doctrine\Persistence\ObjectManager;
  26. use Fidry\AliceDataFixtures\Persistence\PurgeMode;
  27. use Fidry\AliceDataFixtures\Persistence\PurgerFactoryInterface;
  28. use Fidry\AliceDataFixtures\Persistence\PurgerInterface;
  29. use InvalidArgumentException;
  30. use Nelmio\Alice\IsAServiceTrait;
  32. /**
  33. * Bridge for Doctrine purger.
  34. *
  35. * @author Vincent CHALAMON <vincentchalamon@gmail.com>
  36. *
  37. * @final
  38. */
  39. /* final */ class Purger implements PurgerInterface, PurgerFactoryInterface
  40. {
  41. use IsAServiceTrait;
  43. private ObjectManager $manager;
  44. private ?PurgeMode $purgeMode;
  45. private DoctrinePurgerInterface $purger;
  47. public function __construct(ObjectManager $manager, PurgeMode $purgeMode = null)
  48. {
  49. $this->manager = $manager;
  50. $this->purgeMode = $purgeMode;
  52. $this->purger = static::createPurger($manager, $purgeMode);
  53. }
  55. public function create(PurgeMode $mode, PurgerInterface $purger = null): PurgerInterface
  56. {
  57. if (null === $purger) {
  58. return new self($this->manager, $mode);
  59. }
  61. if ($purger instanceof DoctrinePurgerInterface) {
  62. $manager = $purger->getObjectManager();
  63. } elseif ($purger instanceof self) {
  64. $manager = $purger->manager;
  65. } else {
  66. throw new InvalidArgumentException(
  67. sprintf(
  68. 'Expected purger to be either and instance of "%s" or "%s". Got "%s".',
  69. DoctrinePurgerInterface::class,
  70. __CLASS__,
  71. get_class($purger)
  72. )
  73. );
  74. }
  76. if (null === $manager) {
  77. throw new InvalidArgumentException(
  78. sprintf(
  79. 'Expected purger "%s" to have an object manager, got "null" instead.',
  80. get_class($purger)
  81. )
  82. );
  83. }
  85. return new self($manager, $mode);
  86. }
  88. public function purge(): void
  89. {
  90. // Because MySQL rocks, you got to disable foreign key checks when doing a TRUNCATE/DELETE unlike in for example
  91. // PostgreSQL. This ideally should be done in the Purger of doctrine/data-fixtures but meanwhile we are doing
  92. // it here.
  93. // See the progress in https://github.com/doctrine/data-fixtures/pull/272
  94. $disableFkChecks = (
  95. $this->purger instanceof DoctrineOrmPurger
  96. && in_array($this->purgeMode->getValue(), [PurgeMode::createDeleteMode()->getValue(), PurgeMode::createTruncateMode()->getValue()])
  97. && $this->doesDatabaseSupportTruncate()
  98. );
  100. if ($disableFkChecks) {
  101. $connection = $this->purger->getObjectManager()->getConnection();
  103. $connection->executeStatement('SET FOREIGN_KEY_CHECKS = 0;');
  104. }
  106. $this->purger->purge();
  108. if ($disableFkChecks && isset($connection)) {
  109. $connection->executeStatement('SET FOREIGN_KEY_CHECKS = 1;');
  110. }
  111. }
  113. private static function createPurger(ObjectManager $manager, ?PurgeMode $purgeMode): DoctrinePurgerInterface
  114. {
  115. if ($manager instanceof EntityManagerInterface) {
  116. $purger = new DoctrineOrmPurger($manager);
  118. if (null !== $purgeMode) {
  119. $purger->setPurgeMode($purgeMode->getValue());
  120. }
  122. return $purger;
  123. }
  125. if ($manager instanceof DoctrinePhpCrDocumentManager) {
  126. return new DoctrinePhpCrPurger($manager);
  127. }
  129. if ($manager instanceof DoctrineMongoDocumentManager) {
  130. return new DoctrineMongoDBPurger($manager);
  131. }
  133. throw new InvalidArgumentException(
  134. sprintf(
  135. 'Cannot create a purger for ObjectManager of class %s',
  136. get_class($manager)
  137. )
  138. );
  139. }
  141. private function doesDatabaseSupportTruncate(): bool
  142. {
  143. $platform = $this->purger->getObjectManager()->getConnection()->getDatabasePlatform();
  145. return $platform instanceof MySqlPlatform
  146. || $platform instanceof MariaDBPlatform;
  147. }
  148. }