pure-orm 4.0.0-7 → 4.0.0-8

package/dist/src/core.js

@@ -58,15 +58,23 @@ const createCore = ({ entities: externalEntities }) => {
             getPkId
         };
     });
+    const tableNameToEntityMap = entities.reduce((map, entity) => {
+        map.set(entity.tableName, entity);
+        return map;
+    }, new Map());
     const getEntityByTableName = (tableName) => {
-        const entity = entities.find((data) => data.tableName == tableName);
+        const entity = tableNameToEntityMap.get(tableName);
         if (!entity) {
             throw new Error(`Could not find entity for table ${tableName}`);
         }
         return entity;
     };
+    const modelToEntityMap = entities.reduce((map, entity) => {
+        map.set(entity.Model, entity);
+        return map;
+    }, new Map());
     const getEntityByModel = (model) => {
-        const entity = entities.find((data) => data.Model == model.constructor);
+        const entity = modelToEntityMap.get(model.constructor);
         if (!entity) {
             throw new Error(`Could not find entity for class ${model.constructor}`);
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pure-orm",
-  "version": "4.0.0-7",
+  "version": "4.0.0-8",
   "main": "dist/src/index.js",
   "types": "dist/src/index.d.ts",
   "engines": {

package/src/core.ts

@@ -179,16 +179,38 @@ export const createCore = ({
     }
   );
 
+  const tableNameToEntityMap = entities.reduce(
+    (
+      map: Map<string, IEntityInternal<IModel>>,
+      entity: IEntityInternal<IModel>
+    ) => {
+      map.set(entity.tableName, entity);
+      return map;
+    },
+    new Map()
+  );
+
   const getEntityByTableName = (tableName: string): IEntityInternal<IModel> => {
-    const entity = entities.find((data) => data.tableName == tableName);
+    const entity = tableNameToEntityMap.get(tableName);
     if (!entity) {
       throw new Error(`Could not find entity for table ${tableName}`);
     }
     return entity;
   };
 
+  const modelToEntityMap = entities.reduce(
+    (
+      map: Map<IModel, IEntityInternal<IModel>>,
+      entity: IEntityInternal<IModel>
+    ) => {
+      map.set(entity.Model, entity);
+      return map;
+    },
+    new Map()
+  );
+
   const getEntityByModel = (model: IModel): IEntityInternal<IModel> => {
-    const entity = entities.find((data) => data.Model == model.constructor);
+    const entity = modelToEntityMap.get(model.constructor);
     if (!entity) {
       throw new Error(`Could not find entity for class ${model.constructor}`);
     }