Added some type checks

DamienHarper · Apr 12, 2019 · 02469a0 · 02469a0
1 parent dcdb0ce
commit 02469a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/DoctrineAuditBundle/Reader/AuditReader.php b/src/DoctrineAuditBundle/Reader/AuditReader.php
@@ -123,7 +123,7 @@ public function getAudits($entity, $id = null, ?int $page = null, ?int $pageSize
         }
 
         $connection = $this->entityManager->getConnection();
-        $schema = $this->entityManager->getClassMetadata($entity)->getSchemaName();
+        $schema = $this->entityManager->getClassMetadata(\is_string($entity) ? $entity : \get_class($entity))->getSchemaName();
 
         $auditTable = implode('', [
             null === $schema ? '' : $schema.'.',
@@ -175,7 +175,7 @@ public function getAudits($entity, $id = null, ?int $page = null, ?int $pageSize
     public function getAudit($entity, $id)
     {
         $connection = $this->entityManager->getConnection();
-        $schema = $this->entityManager->getClassMetadata($entity)->getSchemaName();
+        $schema = $this->entityManager->getClassMetadata(\is_string($entity) ? $entity : \get_class($entity))->getSchemaName();
 
         $auditTable = implode('', [
             null === $schema ? '' : $schema.'.',