@skroz/profile-api 1.0.27 → 1.0.28

package/dist/index.d.ts

@@ -9,3 +9,4 @@ export * from './resolvers/AuthResolver';
 export * from './resolvers/ProfileResolver';
 export * from './resolvers/createOauthResolver';
 export * from './oauth';
+export { default as createEntityDataLoader } from './utils/createEntityDataLoader';

package/dist/index.js

@@ -13,7 +13,11 @@ var __createBinding = (this && this.__createBinding) || (Object.create ? (functi
 var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.createEntityDataLoader = void 0;
 __exportStar(require("./types"), exports);
 __exportStar(require("./adapters/TypeOrmProfileAdapter"), exports);
 __exportStar(require("./entities/TypeOrmBaseEntity"), exports);
@@ -25,3 +29,5 @@ __exportStar(require("./resolvers/AuthResolver"), exports);
 __exportStar(require("./resolvers/ProfileResolver"), exports);
 __exportStar(require("./resolvers/createOauthResolver"), exports);
 __exportStar(require("./oauth"), exports);
+var createEntityDataLoader_1 = require("./utils/createEntityDataLoader");
+Object.defineProperty(exports, "createEntityDataLoader", { enumerable: true, get: function () { return __importDefault(createEntityDataLoader_1).default; } });

package/dist/utils/createEntityDataLoader.d.ts

@@ -0,0 +1,5 @@
+import DataLoader from 'dataloader';
+import { Connection, ObjectType } from 'typeorm';
+import { TypeOrmBaseEntity as BaseEntity } from '../entities/TypeOrmBaseEntity';
+declare const createEntityDataLoader: <T extends BaseEntity>(conn: Connection, target: ObjectType<T>) => DataLoader<number, T, number>;
+export default createEntityDataLoader;

package/dist/utils/createEntityDataLoader.js

@@ -0,0 +1,25 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const dataloader_1 = __importDefault(require("dataloader"));
+const batch = (conn, target) => (ids) => __awaiter(void 0, void 0, void 0, function* () {
+    const entities = yield conn.getRepository(target).findByIds(ids);
+    const entitiesMap = {};
+    entities.forEach((item) => {
+        entitiesMap[item.id] = item;
+    });
+    return ids.map((id) => entitiesMap[id] || null);
+});
+const createEntityDataLoader = (conn, target) => new dataloader_1.default(batch(conn, target));
+exports.default = createEntityDataLoader;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skroz/profile-api",
-  "version": "1.0.27",
+  "version": "1.0.28",
   "license": "MIT",
   "repository": "git@gitlab.com:skroz/libs/utils.git",
   "main": "dist/index.js",
@@ -26,6 +26,7 @@
     "@os-team/session": "1.0.32",
     "argon2": "0.30.2",
     "class-validator": "0.13.2",
+    "dataloader": "2.1.0",
     "isomorphic-unfetch": "4.0.2",
     "nanoid": "3.3.4",
     "pug": "3.0.2"
@@ -44,5 +45,5 @@
     "type-graphql": "^1.1.1",
     "typeorm": "^0.2.45"
   },
-  "gitHead": "fe4ee603e441d8e65f1c77eebdd2359b2b99a789"
+  "gitHead": "47d38c667bd718863616823d46460c74ad2fc5eb"
 }

package/src/index.ts

@@ -9,3 +9,4 @@ export * from './resolvers/AuthResolver';
 export * from './resolvers/ProfileResolver';
 export * from './resolvers/createOauthResolver';
 export * from './oauth';
+export { default as createEntityDataLoader } from './utils/createEntityDataLoader';

package/src/utils/createEntityDataLoader.ts

@@ -0,0 +1,23 @@
+import DataLoader from 'dataloader';
+import { Connection, ObjectType } from 'typeorm';
+import { TypeOrmBaseEntity as BaseEntity } from '../entities/TypeOrmBaseEntity';
+
+const batch =
+  <T extends BaseEntity>(conn: Connection, target: ObjectType<T>) =>
+  async (ids: readonly number[]) => {
+    const entities = await conn.getRepository(target).findByIds(ids as number[]);
+
+    const entitiesMap: { [key: number]: T } = {};
+    entities.forEach((item) => {
+      entitiesMap[item.id] = item;
+    });
+
+    return ids.map((id) => entitiesMap[id] || null);
+  };
+
+const createEntityDataLoader = <T extends BaseEntity>(
+  conn: Connection,
+  target: ObjectType<T>
+) => new DataLoader<number, T>(batch(conn, target));
+
+export default createEntityDataLoader;