From 02469a07e190b17786f890bb1ce9a6449cb53ebe Mon Sep 17 00:00:00 2001
From: Damien Harper <damien.harper@gmail.com>
Date: Fri, 12 Apr 2019 17:37:35 +0200
Subject: [PATCH] Added some type checks

---
 src/DoctrineAuditBundle/Reader/AuditReader.php | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/DoctrineAuditBundle/Reader/AuditReader.php b/src/DoctrineAuditBundle/Reader/AuditReader.php
index 0e3e5cff..4059ca70 100644
--- 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.'.',