fix(database): improve performance with entity lookup (#718)

vuex-orm · May 6, 2021 · 8b5f104 · 8b5f104
1 parent 7573d2f
commit 8b5f104
Showing 1 changed file with 10 additions and 2 deletions.
diff --git 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(