vendor/theofidry/alice-data-fixtures/src/Loader/PersisterLoader.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\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\ProcessorInterface;
  21. use JetBrains\PhpStorm\Pure;
  22. use Nelmio\Alice\IsAServiceTrait;
  23. use Psr\Log\LoggerInterface;
  24. use Psr\Log\NullLogger;
  26. /**
  27. * Loader decorating another loader to add a persistence layer.
  28. *
  29. * @author Jordi Boggiano <j.boggiano@seld.be>
  30. * @author Théo FIDRY <theo.fidry@gmail.com>
  31. *
  32. * @final
  33. */
  34. /*final*/ class PersisterLoader implements LoaderInterface, PersisterAwareInterface
  35. {
  36. use IsAServiceTrait;
  38. private LoaderInterface $loader;
  39. private PersisterInterface $persister;
  40. private LoggerInterface $logger;
  41. private array $processors;
  43. /**
  44. * @param ProcessorInterface[] $processors
  45. */
  46. #[Pure]
  47. public function __construct(
  48. LoaderInterface $decoratedLoader,
  49. PersisterInterface $persister,
  50. LoggerInterface $logger = null,
  51. array $processors = []
  52. ) {
  53. $this->loader = $decoratedLoader;
  54. $this->persister = $persister;
  55. $this->logger = $logger ?? new NullLogger();
  56. $this->processors = (static function (ProcessorInterface ...$processors) {
  57. return $processors;
  58. })(...$processors);
  59. }
  61. public function withPersister(PersisterInterface $persister): self
  62. {
  63. $loader = $this->loader;
  65. if ($loader instanceof PersisterAwareInterface) {
  66. $loader = $loader->withPersister($persister);
  67. }
  69. return new self($loader, $persister, $this->logger, $this->processors);
  70. }
  72. /**
  73. * Pre-process, persist and post process each object loaded.
  74. *
  75. * {@inheritdoc}
  76. */
  77. public function load(array $fixturesFiles, array $parameters = [], array $objects = [], PurgeMode $purgeMode = null): array
  78. {
  79. $objects = $this->loader->load($fixturesFiles, $parameters, $objects, $purgeMode);
  81. $this->logger->info('Pre-processing objects.');
  83. foreach ($objects as $id => $object) {
  84. foreach ($this->processors as $processor) {
  85. $processor->preProcess($id, $object);
  86. }
  88. $this->persister->persist($object);
  89. }
  91. $this->logger->info('Flushing objects.');
  93. $this->persister->flush();
  95. $this->logger->info('Post-processing objects.');
  97. foreach ($objects as $id => $object) {
  98. foreach ($this->processors as $processor) {
  99. $processor->postProcess($id, $object);
  100. }
  101. }
  103. $this->logger->info('Done.');
  105. return $objects;
  106. }
  107. }