-
Notifications
You must be signed in to change notification settings - Fork 284
/
TimestampableEventSubscriber.php
61 lines (50 loc) · 1.74 KB
/
TimestampableEventSubscriber.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
<?php
declare(strict_types=1);
namespace Knp\DoctrineBehaviors\EventSubscriber;
use Doctrine\Bundle\DoctrineBundle\EventSubscriber\EventSubscriberInterface;
use Doctrine\ORM\Event\LoadClassMetadataEventArgs;
use Doctrine\ORM\Events;
use Knp\DoctrineBehaviors\Contract\Entity\TimestampableInterface;
final class TimestampableEventSubscriber implements EventSubscriberInterface
{
/**
* @var string
*/
private $dateFieldType;
public function __construct(string $timestampableDateFieldType)
{
$this->dateFieldType = $timestampableDateFieldType;
}
public function loadClassMetadata(LoadClassMetadataEventArgs $loadClassMetadataEventArgs): void
{
$classMetadata = $loadClassMetadataEventArgs->getClassMetadata();
if ($classMetadata->reflClass === null) {
// Class has not yet been fully built, ignore this event
return;
}
if (! is_a($classMetadata->reflClass->getName(), TimestampableInterface::class, true)) {
return;
}
if ($classMetadata->isMappedSuperclass) {
return;
}
$classMetadata->addLifecycleCallback('updateTimestamps', Events::prePersist);
$classMetadata->addLifecycleCallback('updateTimestamps', Events::preUpdate);
foreach (['createdAt', 'updatedAt'] as $field) {
if (! $classMetadata->hasField($field)) {
$classMetadata->mapField([
'fieldName' => $field,
'type' => $this->dateFieldType,
'nullable' => true,
]);
}
}
}
/**
* @return string[]
*/
public function getSubscribedEvents(): array
{
return [Events::loadClassMetadata];
}
}