forked from Sylius/Sylius
-
Notifications
You must be signed in to change notification settings - Fork 0
/
ORMPurgerListener.php
92 lines (78 loc) · 2.26 KB
/
ORMPurgerListener.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
<?php
/*
* This file is part of the Sylius package.
*
* (c) Paweł Jędrzejewski
*
* For the full copyright and license information, please view the LICENSE
* file that was distributed with this source code.
*/
declare(strict_types=1);
namespace Sylius\Bundle\FixturesBundle\Listener;
use Doctrine\Common\DataFixtures\Purger\ORMPurger;
use Doctrine\Common\Persistence\ManagerRegistry;
use Doctrine\ORM\EntityManagerInterface;
use Symfony\Component\Config\Definition\Builder\ArrayNodeDefinition;
final class ORMPurgerListener extends AbstractListener implements BeforeSuiteListenerInterface
{
/**
* @var ManagerRegistry
*/
private $managerRegistry;
/**
* @var array
*/
private static $purgeModes = [
'delete' => ORMPurger::PURGE_MODE_DELETE,
'truncate' => ORMPurger::PURGE_MODE_TRUNCATE,
];
/**
* @param ManagerRegistry $managerRegistry
*/
public function __construct(ManagerRegistry $managerRegistry)
{
$this->managerRegistry = $managerRegistry;
}
/**
* {@inheritdoc}
*/
public function beforeSuite(SuiteEvent $suiteEvent, array $options): void
{
foreach ($options['managers'] as $managerName) {
/** @var EntityManagerInterface $manager */
$manager = $this->managerRegistry->getManager($managerName);
$purger = new ORMPurger($manager, $options['exclude']);
$purger->setPurgeMode(static::$purgeModes[$options['mode']]);
$purger->purge();
}
}
/**
* {@inheritdoc}
*/
public function getName(): string
{
return 'orm_purger';
}
/**
* {@inheritdoc}
*/
protected function configureOptionsNode(ArrayNodeDefinition $optionsNode): void
{
$optionsNodeBuilder = $optionsNode->children();
$optionsNodeBuilder
->enumNode('mode')
->values(['delete', 'truncate'])
->defaultValue('delete')
;
$optionsNodeBuilder
->arrayNode('managers')
->defaultValue([null])
->scalarPrototype()
;
$optionsNodeBuilder
->arrayNode('exclude')
->defaultValue([])
->scalarPrototype()
;
}
}