vendor/theofidry/alice-data-fixtures/src/Loader/PurgerLoader.php line 45

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\Loader;
  16. use Fidry\AliceDataFixtures\LoaderInterface;
  17. use Fidry\AliceDataFixtures\Persistence\PersisterAwareInterface;
  18. use Fidry\AliceDataFixtures\Persistence\PersisterInterface;
  19. use Fidry\AliceDataFixtures\Persistence\PurgeMode;
  20. use Fidry\AliceDataFixtures\Persistence\PurgerFactoryInterface;
  21. use InvalidArgumentException;
  22. use Nelmio\Alice\IsAServiceTrait;
  23. use Psr\Log\LoggerInterface;
  24. use Psr\Log\NullLogger;
  26. /**
  27. * Loader decorating another loader to purge the database before loading.
  28. *
  29. * @final
  30. */
  31. /*final*/ class PurgerLoader implements LoaderInterface, PersisterAwareInterface
  32. {
  33. use IsAServiceTrait;
  35. public const PURGE_MODE_DELETE = 1;
  36. public const PURGE_MODE_TRUNCATE = 2;
  38. private static array $PURGE_MAPPING;
  40. private LoaderInterface $loader;
  41. private PurgerFactoryInterface $purgerFactory;
  42. private PurgeMode $defaultPurgeMode;
  43. private LoggerInterface $logger;
  45. public function __construct(
  46. LoaderInterface $decoratedLoader,
  47. PurgerFactoryInterface $purgerFactory,
  48. string $defaultPurgeMode,
  49. LoggerInterface $logger = null
  50. ) {
  51. if (!isset(self::$PURGE_MAPPING)) {
  52. self::$PURGE_MAPPING = [
  53. 'delete' => PurgeMode::createDeleteMode(),
  54. 'truncate' => PurgeMode::createTruncateMode(),
  55. 'no_purge' => PurgeMode::createNoPurgeMode(),
  56. ];
  57. }
  59. $this->loader = $decoratedLoader;
  60. $this->purgerFactory = $purgerFactory;
  62. if (false === in_array($defaultPurgeMode, ['delete', 'truncate', 'no_purge'], true)) {
  63. throw new InvalidArgumentException(
  64. sprintf(
  65. 'Unknown purge mode "%s". Use "delete", "truncate" or "no_purge".',
  66. $defaultPurgeMode
  67. )
  68. );
  69. }
  71. $this->defaultPurgeMode = self::$PURGE_MAPPING[$defaultPurgeMode];
  72. $this->logger = $logger ?? new NullLogger();
  73. }
  75. /**
  76. * @inheritdoc
  77. */
  78. public function withPersister(PersisterInterface $persister): self
  79. {
  80. $loader = $this->loader;
  82. if ($loader instanceof PersisterAwareInterface) {
  83. $loader = $loader->withPersister($persister);
  84. }
  86. foreach (self::$PURGE_MAPPING as $string => $mode) {
  87. if ($mode == $this->defaultPurgeMode) {
  88. $defaultPurgeMode = $string;
  89. break;
  90. }
  91. }
  93. return new self($loader, $this->purgerFactory, $defaultPurgeMode, $this->logger);
  94. }
  96. /**
  97. * Pre-process, persist and post process each object loaded.
  98. *
  99. * {@inheritdoc}
  100. */
  101. public function load(array $fixturesFiles, array $parameters = [], array $objects = [], PurgeMode $purgeMode = null): array
  102. {
  103. if (null === $purgeMode) {
  104. $purgeMode = $this->defaultPurgeMode;
  105. }
  107. if ($purgeMode != PurgeMode::createNoPurgeMode()) {
  108. $this->logger->info(
  109. sprintf(
  110. 'Purging database with purge mode "%s".',
  111. (string) $purgeMode
  112. )
  113. );
  115. $purger = $this->purgerFactory->create($purgeMode);
  116. $purger->purge();
  117. }
  119. return $this->loader->load($fixturesFiles, $parameters, $objects, $purgeMode);
  120. }
  121. }