From 8b5f10443d86d95980715d613a152413e6ad3867 Mon Sep 17 00:00:00 2001
From: Nuno Amorim <ntma@users.noreply.github.com>
Date: Thu, 6 May 2021 15:39:44 +0100
Subject: [PATCH] fix(database): improve performance with entity lookup (#718)

---
 src/database/Database.ts | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/database/Database.ts b/src/database/Database.ts
index 3323e376..1a87e77b 100644
--- a/src/database/Database.ts
+++ b/src/database/Database.ts
@@ -99,7 +99,11 @@ export default class Database {
   model(model: string): typeof Model
   model(model: typeof Model | string): typeof Model | string {
     const name = typeof model === 'string' ? model : model.entity
-    const m = this.models()[name]
+    const entity = this.entities.find((entity) => {
+      return entity.name === name
+    })
+
+    const m = entity ? entity.model : null
 
     if (!m) {
       throw new Error(
@@ -118,7 +122,11 @@ export default class Database {
   baseModel(model: string): typeof Model
   baseModel(model: typeof Model | string): typeof Model | string {
     const name = typeof model === 'string' ? model : model.entity
-    const m = this.baseModels()[name]
+    const entity = this.entities.find((entity) => {
+      return entity.name === name
+    })
+
+    const m = entity ? this.model(entity.base) : null
 
     if (!m) {
       throw new Error(