Fixed invokable storage mapper (see #146) (#148)

DamienHarper · Dec 15, 2022 · 1de4929 · 1de4929
1 parent 11d5d2f
commit 1de4929
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/Provider/Doctrine/Auditing/Transaction/AuditTrait.php b/src/Provider/Doctrine/Auditing/Transaction/AuditTrait.php
@@ -75,7 +75,8 @@ private function value(EntityManagerInterface $entityManager, Type $type, $value
         if (null === $value) {
             return null;
         }
-        if (interface_exists(UnitEnum::class) && $value instanceof UnitEnum && property_exists($value, 'value')) {
+
+        if (interface_exists(UnitEnum::class) && $value instanceof UnitEnum && property_exists($value, 'value')) { /** @phpstan-ignore-line */
             $value = $value->value;
         }
 

diff --git a/src/Provider/Doctrine/DoctrineProvider.php b/src/Provider/Doctrine/DoctrineProvider.php
@@ -89,6 +89,10 @@ public function getStorageServiceForEntity(string $entity): StorageServiceInterf
             return array_values($this->getStorageServices())[0];
         }
 
+        if (\is_string($storageMapper) && class_exists($storageMapper)) {
+            $storageMapper = new $storageMapper();
+        }
+
         \assert(\is_callable($storageMapper));   // helps PHPStan
 
         return $storageMapper($entity, $this->getStorageServices());