add integer identity trait (#382)

ro0NL · web-flow · commit 28f96469c45f · 2020-03-14T13:17:43.000+01:00
diff --git a/src/Domain/Infrastructure/Doctrine/DomainObjectMappings.php b/src/Domain/Infrastructure/Doctrine/DomainObjectMappings.php
@@ -4,6 +4,7 @@
 
 namespace MsgPhp\Domain\Infrastructure\Doctrine;
 
+use Doctrine\ORM\Mapping\ClassMetadata;
 use MsgPhp\Domain\Model;
 
 /**
@@ -15,6 +16,13 @@ final class DomainObjectMappings implements ObjectMappingProvider
 {
     public static function provideObjectMappings(MappingConfig $config): iterable
     {
+        yield Model\IntIdentity::class => [
+            'id' => [
+                'id' => true,
+                'id_generator' => ClassMetadata::GENERATOR_TYPE_IDENTITY,
+                'type' => 'integer',
+            ],
+        ];
         yield Model\CanBeConfirmed::class => [
             'confirmationToken' => [
                 'type' => 'string',
diff --git a/src/Domain/Infrastructure/Doctrine/Event/ObjectMappingListener.php b/src/Domain/Infrastructure/Doctrine/Event/ObjectMappingListener.php
@@ -5,6 +5,7 @@
 namespace MsgPhp\Domain\Infrastructure\Doctrine\Event;
 
 use Doctrine\ORM\Event\LoadClassMetadataEventArgs;
+use Doctrine\ORM\Id\AbstractIdGenerator;
 use Doctrine\ORM\Mapping\ClassMetadataFactory;
 use Doctrine\ORM\Mapping\ClassMetadataInfo;
 use MsgPhp\Domain\Infrastructure\Doctrine\MappingConfig;
@@ -89,6 +90,15 @@ private function processFieldMapping(ClassMetadataInfo $metadata, array $mapping
                 continue;
             }
 
+            if (null !== $idGenerator = $info['id_generator'] ?? null) {
+                if ($idGenerator instanceof AbstractIdGenerator) {
+                    $metadata->setIdGenerator($idGenerator);
+                } else {
+                    $metadata->setIdGeneratorType($idGenerator);
+                }
+                unset($info['id_generator']);
+            }
+
             $info = ['fieldName' => $field] + $info;
 
             if (isset($info['type']) && method_exists($metadata, $method = 'map'.ucfirst($info['type']))) {
diff --git a/src/Domain/Model/IntIdentity.php b/src/Domain/Model/IntIdentity.php
@@ -0,0 +1,14 @@
+<?php
+
+declare(strict_types=1);
+
+namespace MsgPhp\Domain\Model;
+
+/**
+ * @author Roland Franssen <franssen.roland@gmail.com>
+ */
+trait IntIdentity
+{
+    /** @var null|int */
+    private $id;
+}
