Cache : handle edge case when entity key is a MongoDB ObjectId

analogueorm · Mar 15, 2018 · e02226c · e02226c
1 parent 3c19f07
commit e02226c
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/System/Aggregate.php b/src/System/Aggregate.php
@@ -11,6 +11,7 @@
 use Illuminate\Support\Collection;
 use ProxyManager\Proxy\LazyLoadingInterface;
 use ProxyManager\Proxy\ProxyInterface;
+use MongoDB\BSON\ObjectId;
 
 /**
  * This class is aimed to facilitate the handling of
@@ -375,7 +376,13 @@ public function getEntityKeyName(): string
      */
     public function getEntityKeyValue()
     {
-        return $this->wrappedEntity->getEntityKeyValue();
+        $keyValue = $this->wrappedEntity->getEntityKeyValue();
+
+        if($keyValue instanceof ObjectId) {
+            $keyValue = (string) $keyValue;
+        }
+
+        return $keyValue;
     }
 
     /**